|
@@ -49,26 +49,11 @@ public class PeerTaskInstancePriorityQueueTest {
|
|
|
@Test
|
|
|
public void poll() throws Exception {
|
|
|
PeerTaskInstancePriorityQueue queue = getPeerTaskInstancePriorityQueue();
|
|
|
- int peekBeforeLength = queue.size();
|
|
|
- queue.poll(1000, TimeUnit.MILLISECONDS);
|
|
|
- queue.poll(1000, TimeUnit.MILLISECONDS);
|
|
|
- Assert.assertEquals(0, queue.size());
|
|
|
- Thread producer = new Thread(() -> {
|
|
|
- System.out.println(String.format("Ready to producing...,now time is %s ", System.currentTimeMillis()));
|
|
|
- try {
|
|
|
- Thread.sleep(100);
|
|
|
- TaskInstance task = createTaskInstance("low_task", Priority.LOW);
|
|
|
- queue.put(task);
|
|
|
- } catch (Exception e) {
|
|
|
- e.printStackTrace();
|
|
|
- }
|
|
|
- System.out.println(String.format("End to produce %s at time %s",
|
|
|
- queue.peek() != null ? queue.peek().getName() : null, System.currentTimeMillis()));
|
|
|
- });
|
|
|
- producer.start();
|
|
|
- System.out.println("Begin to consume at " + System.currentTimeMillis());
|
|
|
- queue.poll(1000, TimeUnit.MILLISECONDS);
|
|
|
- System.out.println("End to consume at " + System.currentTimeMillis());
|
|
|
+ try {
|
|
|
+ queue.poll(1000, TimeUnit.MILLISECONDS);
|
|
|
+ } catch (TaskPriorityQueueException e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
@Test
|