this._pendingWrites = new ShellBoltMessageQueue(((Number) maxPending).intValue());
@Test public void testPollWhileThereAreNoDataAvailable() throws InterruptedException { ShellBoltMessageQueue queue = new ShellBoltMessageQueue(); long start = System.currentTimeMillis(); Object msg = queue.poll(1, TimeUnit.SECONDS); long finish = System.currentTimeMillis(); long waitDuration = finish - start; assertNull(msg); assertTrue("wait duration should be equal or greater than 1000, current: " + waitDuration, waitDuration >= 1000); }
@Test public void testPollShouldReturnASAPWhenDataAvailable() throws InterruptedException { final ShellBoltMessageQueue queue = new ShellBoltMessageQueue(); final List<Integer> taskIds = Lists.newArrayList(1, 2, 3); Thread t = new Thread(new Runnable() { @Override public void run() { try { Thread.sleep(1000); } catch (InterruptedException e) { // NOOP } queue.putTaskIds(taskIds); } }); t.start(); long start = System.currentTimeMillis(); Object msg = queue.poll(10, TimeUnit.SECONDS); long finish = System.currentTimeMillis(); assertEquals(msg, taskIds); assertTrue(finish - start < (10 * 1000)); } }
@Test public void testPollTaskIdsFirst() throws InterruptedException { ShellBoltMessageQueue queue = new ShellBoltMessageQueue(); // put bolt message first, then put task ids queue.putBoltMsg(new BoltMsg()); ArrayList<Integer> taskIds = Lists.newArrayList(1, 2, 3); queue.putTaskIds(taskIds); Object msg = queue.poll(10, TimeUnit.SECONDS); // task ids should be pulled first assertTrue(msg instanceof List<?>); assertEquals(msg, taskIds); }
Object maxPending = stormConf.get(Config.TOPOLOGY_SHELLBOLT_MAX_PENDING); if (maxPending != null) { this._pendingWrites = new ShellBoltMessageQueue(((Number)maxPending).intValue());