Examples with ReadIndexClosure org.apache.ignite.raft.jraft.closure.ReadIndexClosure used on opensource projects

Example 1 with ReadIndexClosure

use of org.apache.ignite.raft.jraft.closure.ReadIndexClosure in project ignite-3 by apache.

the class ItNodeTest method assertReadIndex.

@SuppressWarnings({ "unused", "SameParameterValue" })
private boolean assertReadIndex(Node node, int index) throws InterruptedException {
    CountDownLatch latch = new CountDownLatch(1);
    byte[] requestContext = TestUtils.getRandomBytes();
    AtomicBoolean success = new AtomicBoolean(false);
    node.readIndex(requestContext, new ReadIndexClosure() {

        @Override
        public void run(Status status, long theIndex, byte[] reqCtx) {
            if (status.isOk()) {
                assertEquals(index, theIndex);
                assertArrayEquals(requestContext, reqCtx);
                success.set(true);
            } else {
                assertTrue(status.getErrorMsg().contains("RPC exception:Check connection["), status.getErrorMsg());
                assertTrue(status.getErrorMsg().contains("] fail and try to create new one"), status.getErrorMsg());
            }
            latch.countDown();
        }
    });
    latch.await();
    return success.get();
}

Example 2 with ReadIndexClosure

use of org.apache.ignite.raft.jraft.closure.ReadIndexClosure in project ignite-3 by apache.

the class ReadOnlyServiceImpl method notifySuccess.

private void notifySuccess(final ReadIndexStatus status) {
    final long nowMs = Utils.monotonicMs();
    final List<ReadIndexState> states = status.getStates();
    final int taskCount = states.size();
    for (int i = 0; i < taskCount; i++) {
        final ReadIndexState task = states.get(i);
        // stack copy
        final ReadIndexClosure done = task.getDone();
        if (done != null) {
            this.nodeMetrics.recordLatency("read-index", nowMs - task.getStartTimeMs());
            done.setResult(task.getIndex(), task.getRequestContext().get());
            done.run(Status.OK());
        }
    }
}

Example 3 with ReadIndexClosure

use of org.apache.ignite.raft.jraft.closure.ReadIndexClosure in project ignite-3 by apache.

the class ReadOnlyServiceImpl method addRequest.

@Override
public void addRequest(final byte[] reqCtx, final ReadIndexClosure closure) {
    if (this.shutdownLatch != null) {
        Utils.runClosureInThread(this.node.getOptions().getCommonExecutor(), closure, new Status(RaftError.EHOSTDOWN, "Was stopped"));
        throw new IllegalStateException("Service already shutdown.");
    }
    try {
        EventTranslator<ReadIndexEvent> translator = (event, sequence) -> {
            event.groupId = this.groupId;
            event.done = closure;
            event.requestContext = new Bytes(reqCtx);
            event.startTime = Utils.monotonicMs();
        };
        int retryTimes = 0;
        while (true) {
            if (this.readIndexQueue.tryPublishEvent(translator)) {
                break;
            } else {
                retryTimes++;
                if (retryTimes > MAX_ADD_REQUEST_RETRY_TIMES) {
                    Utils.runClosureInThread(this.node.getOptions().getCommonExecutor(), closure, new Status(RaftError.EBUSY, "Node is busy, has too many read-only requests."));
                    this.nodeMetrics.recordTimes("read-index-overload-times", 1);
                    LOG.warn("Node {} ReadOnlyServiceImpl readIndexQueue is overload.", this.node.getNodeId());
                    return;
                }
                ThreadHelper.onSpinWait();
            }
        }
    } catch (final Exception e) {
        Utils.runClosureInThread(this.node.getOptions().getCommonExecutor(), closure, new Status(RaftError.EPERM, "Node is down."));
    }
}

Example 4 with ReadIndexClosure

use of org.apache.ignite.raft.jraft.closure.ReadIndexClosure in project ignite-3 by apache.

the class ReadOnlyServiceImpl method reportError.

private void reportError(final ReadIndexStatus status, final Status st) {
    final long nowMs = Utils.monotonicMs();
    final List<ReadIndexState> states = status.getStates();
    final int taskCount = states.size();
    for (int i = 0; i < taskCount; i++) {
        final ReadIndexState task = states.get(i);
        final ReadIndexClosure done = task.getDone();
        if (done != null) {
            this.nodeMetrics.recordLatency("read-index", nowMs - task.getStartTimeMs());
            done.run(st);
        }
    }
}

Example 5 with ReadIndexClosure

use of org.apache.ignite.raft.jraft.closure.ReadIndexClosure in project ignite-3 by apache.

the class ReadOnlyServiceTest method testOnApplied.

@Test
public void testOnApplied() throws Exception {
    final ArrayList<ReadIndexState> states = new ArrayList<>();
    final byte[] reqContext = TestUtils.getRandomBytes();
    final CountDownLatch latch = new CountDownLatch(1);
    final ReadIndexState state = new ReadIndexState(new Bytes(reqContext), new ReadIndexClosure() {

        @Override
        public void run(final Status status, final long index, final byte[] reqCtx) {
            assertTrue(status.isOk());
            assertEquals(index, 1);
            assertArrayEquals(reqCtx, reqContext);
            latch.countDown();
        }
    }, Utils.monotonicMs());
    state.setIndex(1);
    states.add(state);
    final ReadIndexStatus readIndexStatus = new ReadIndexStatus(states, null, 1);
    this.readOnlyServiceImpl.getPendingNotifyStatus().put(1L, Arrays.asList(readIndexStatus));
    this.readOnlyServiceImpl.onApplied(2);
    latch.await();
    assertTrue(this.readOnlyServiceImpl.getPendingNotifyStatus().isEmpty());
}