Examples with DEFAULT_COUNT_NAME org.apache.nifi.processors.standard.WaitNotifyProtocol.DEFAULT_COUNT_NAME used on opensource projects

Example 1 with DEFAULT_COUNT_NAME

use of org.apache.nifi.processors.standard.WaitNotifyProtocol.DEFAULT_COUNT_NAME in project nifi by apache.

the class TestWaitNotifyProtocol method testReleaseCandidate.

@Test
public void testReleaseCandidate() throws Exception {
    final List<Integer> candidates = IntStream.range(0, 10).boxed().collect(Collectors.toList());
    final Signal signal = new Signal();
    final List<Integer> released = new ArrayList<>();
    final List<Integer> waiting = new ArrayList<>();
    // Test default name.
    final String counterName = DEFAULT_COUNT_NAME;
    final BiConsumer<Long, Integer> releaseCandidate = (requiredCountForPass, releasableCandidatePerPass) -> {
        released.clear();
        waiting.clear();
        signal.releaseCandidates(counterName, requiredCountForPass, releasableCandidatePerPass, candidates, r -> released.addAll(r), w -> waiting.addAll(w));
    };
    final Field releasableCount = Signal.class.getDeclaredField("releasableCount");
    releasableCount.setAccessible(true);
    // No counter, should wait.
    releaseCandidate.accept(3L, 1);
    assertEquals(0, released.size());
    assertEquals(10, waiting.size());
    assertEquals(0, signal.getCount(DEFAULT_COUNT_NAME));
    assertEquals(0, releasableCount.getInt(signal));
    // Counter is not enough yet.
    signal.getCounts().put(DEFAULT_COUNT_NAME, 1L);
    releaseCandidate.accept(3L, 1);
    assertEquals(0, released.size());
    assertEquals(10, waiting.size());
    // Counter incremented, but not enough
    assertEquals(1, signal.getCount(DEFAULT_COUNT_NAME));
    assertEquals(0, releasableCount.getInt(signal));
    // Counter reached the target.
    signal.getCounts().put(DEFAULT_COUNT_NAME, 3L);
    releaseCandidate.accept(3L, 1);
    assertEquals(1, released.size());
    assertEquals(9, waiting.size());
    // Counter 3 was converted into 1 release
    assertEquals(0, signal.getCount(DEFAULT_COUNT_NAME));
    assertEquals(0, releasableCount.getInt(signal));
    // Counter reached the target for two candidates.
    signal.getCounts().put(DEFAULT_COUNT_NAME, 6L);
    releaseCandidate.accept(3L, 1);
    assertEquals(2, released.size());
    assertEquals(8, waiting.size());
    // Counter 3 was converted into 1 release
    assertEquals(0, signal.getCount(DEFAULT_COUNT_NAME));
    assertEquals(0, releasableCount.getInt(signal));
    // Counter reached the target for two candidates, and reminder is 2.
    signal.getCounts().put(DEFAULT_COUNT_NAME, 11L);
    releaseCandidate.accept(3L, 1);
    // 11 / 3 = 3
    assertEquals(3, released.size());
    assertEquals(7, waiting.size());
    // 11 % 3 = 2
    assertEquals(2, signal.getCount(DEFAULT_COUNT_NAME));
    assertEquals(0, releasableCount.getInt(signal));
    // Counter reached the target for two pass count and each pass can release 2 candidates.
    signal.getCounts().put(DEFAULT_COUNT_NAME, 6L);
    releaseCandidate.accept(3L, 2);
    // (6 / 3) * 2 = 4
    assertEquals(4, released.size());
    assertEquals(6, waiting.size());
    // 6 % 3 = 0
    assertEquals(0, signal.getCount(DEFAULT_COUNT_NAME));
    assertEquals(0, releasableCount.getInt(signal));
    // If there are counts more than enough to release current candidates, unused releasableCount should remain.
    signal.getCounts().put(DEFAULT_COUNT_NAME, 50L);
    releaseCandidate.accept(3L, 2);
    // (50 / 3) * 2 = 32. Used 10.
    assertEquals(10, released.size());
    assertEquals(0, waiting.size());
    // 50 % 3 = 2.
    assertEquals(2, signal.getCount(DEFAULT_COUNT_NAME));
    // 32 - 10 = 22.
    assertEquals(22, releasableCount.getInt(signal));
}