forked from GoodforGod/java-etherscan-api
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathQueueManager.java
72 lines (59 loc) · 2.02 KB
/
QueueManager.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package io.api.etherscan.manager.impl;
import io.api.etherscan.manager.IQueueManager;
import java.util.List;
import java.util.concurrent.*;
import java.util.logging.Logger;
import java.util.stream.Collectors;
import java.util.stream.IntStream;
/**
* Queue implementation
* With size and reset time as params
* @see IQueueManager
*
* @author GoodforGod
* @since 30.10.2018
*/
public class QueueManager implements IQueueManager {
private static final Logger logger = Logger.getLogger(QueueManager.class.getName());
private final int queueSize;
private final BlockingQueue<Integer> queue;
private final List<Integer> queueValues;
private final ScheduledExecutorService queueExecutor;
public QueueManager(int queueSize, int queueResetTimeInSec) {
this(queueSize, queueResetTimeInSec, 0);
}
public QueueManager(int queueSize, int queueResetTimeInSec, int delayInSec) {
this.queueSize = queueSize;
this.queueValues = generateList(queueSize);
this.queue = new ArrayBlockingQueue<>(queueSize);
this.queueExecutor = Executors.newSingleThreadScheduledExecutor();
this.queueExecutor.scheduleAtFixedRate(createTask(), delayInSec, queueResetTimeInSec, TimeUnit.SECONDS);
}
@Override
public boolean takeTurn() {
try {
queue.take();
return true;
} catch (InterruptedException e) {
logger.warning(e.getLocalizedMessage());
return false;
}
}
private Runnable createTask() {
return () -> {
try {
if(queue.size() == queueSize)
return;
queue.clear();
queue.addAll(queueValues);
} catch (Exception e) {
logger.warning(e.getLocalizedMessage());
}
};
}
private List<Integer> generateList(int size) {
return IntStream.range(0, size)
.boxed()
.collect(Collectors.toList());
}
}