Examples with ClientSession io.aeron.cluster.service.ClientSession used on opensource projects

Example 1 with ClientSession

use of io.aeron.cluster.service.ClientSession in project aeron by real-logic.

the class AuthenticationTest method launchService.

private void launchService(final MutableLong sessionId, final MutableReference<byte[]> encodedPrincipal, final AtomicLong msgCounter) {
    final ClusteredService service = new StubClusteredService() {

        private int counterValue = 0;

        public void onSessionOpen(final ClientSession session, final long timestampMs) {
            sessionId.value = session.id();
            encodedPrincipal.set(session.encodedPrincipal());
        }

        public void onSessionMessage(final long clusterSessionId, final long correlationId, final long timestampMs, final DirectBuffer buffer, final int offset, final int length, final Header header) {
            assertThat(buffer.getInt(offset), is(counterValue));
            msgCounter.getAndIncrement();
            counterValue++;
        }
    };
    container = null;
    container = ClusteredServiceContainer.launch(new ClusteredServiceContainer.Context().clusteredService(service).errorHandler(Throwable::printStackTrace).deleteDirOnStart(true));
}

Example 2 with ClientSession

use of io.aeron.cluster.service.ClientSession in project aeron by real-logic.

the class ClusterNodeTest method launchEchoService.

private ClusteredServiceContainer launchEchoService() {
    final ClusteredService echoService = new StubClusteredService() {

        public void onSessionMessage(final long clusterSessionId, final long correlationId, final long timestampMs, final DirectBuffer buffer, final int offset, final int length, final Header header) {
            final ClientSession session = cluster.getClientSession(clusterSessionId);
            while (session.offer(correlationId, buffer, offset, length) < 0) {
                TestUtil.checkInterruptedStatus();
                Thread.yield();
            }
        }
    };
    return ClusteredServiceContainer.launch(new ClusteredServiceContainer.Context().clusteredService(echoService).errorHandler(Throwable::printStackTrace).deleteDirOnStart(true));
}

Example 3 with ClientSession

use of io.aeron.cluster.service.ClientSession in project Aeron by real-logic.

the class ClusterNodeRestartTest method launchReschedulingService.

private void launchReschedulingService(final AtomicLong triggeredTimersCounter) {
    final ClusteredService service = new StubClusteredService() {

        public void onSessionMessage(final ClientSession session, final long timestamp, final DirectBuffer buffer, final int offset, final int length, final Header header) {
            scheduleNext(serviceCorrelationId(7), timestamp + 200);
        }

        public void onTimerEvent(final long correlationId, final long timestamp) {
            triggeredTimersCounter.getAndIncrement();
            scheduleNext(correlationId, timestamp + 200);
        }

        public void onStart(final Cluster cluster, final Image snapshotImage) {
            super.onStart(cluster, snapshotImage);
            if (null != snapshotImage) {
                final FragmentHandler fragmentHandler = (buffer, offset, length, header) -> triggeredTimersCounter.set(buffer.getLong(offset));
                while (true) {
                    final int fragments = snapshotImage.poll(fragmentHandler, 1);
                    if (fragments == 1 || snapshotImage.isEndOfStream()) {
                        break;
                    }
                    idleStrategy.idle();
                }
            }
        }

        public void onTakeSnapshot(final ExclusivePublication snapshotPublication) {
            final ExpandableArrayBuffer buffer = new ExpandableArrayBuffer();
            buffer.putLong(0, triggeredTimersCounter.get());
            while (snapshotPublication.offer(buffer, 0, SIZE_OF_INT) < 0) {
                idleStrategy.idle();
            }
        }

        private void scheduleNext(final long correlationId, final long deadline) {
            idleStrategy.reset();
            while (!cluster.scheduleTimer(correlationId, deadline)) {
                idleStrategy.idle();
            }
        }
    };
    container = ClusteredServiceContainer.launch(new ClusteredServiceContainer.Context().clusteredService(service).terminationHook(ClusterTests.NOOP_TERMINATION_HOOK).errorHandler(ClusterTests.errorHandler(0)));
}

Example 4 with ClientSession

use of io.aeron.cluster.service.ClientSession in project Aeron by real-logic.

the class ClusterTimerTest method shouldRescheduleTimerWhenSchedulingWithExistingCorrelationId.

@Test
@InterruptAfter(10)
void shouldRescheduleTimerWhenSchedulingWithExistingCorrelationId() {
    final AtomicLong timerCounter1 = new AtomicLong();
    final AtomicLong timerCounter2 = new AtomicLong();
    final ClusteredService service = new StubClusteredService() {

        public void onSessionOpen(final ClientSession session, final long timestamp) {
            // Too far in the future
            schedule(1, timestamp + 1_000_000);
            schedule(1, timestamp + 20);
            schedule(2, timestamp + 30);
        }

        public void onTimerEvent(final long correlationId, final long timestamp) {
            if (correlationId == 1) {
                timerCounter1.incrementAndGet();
            } else {
                timerCounter2.incrementAndGet();
            }
        }

        private void schedule(final long correlationId, final long deadlineMs) {
            idleStrategy.reset();
            while (!cluster.scheduleTimer(correlationId, deadlineMs)) {
                idleStrategy.idle();
            }
        }
    };
    container = ClusteredServiceContainer.launch(new ClusteredServiceContainer.Context().clusteredService(service).terminationHook(ClusterTests.NOOP_TERMINATION_HOOK).errorHandler(ClusterTests.errorHandler(0)));
    connectClient();
    Tests.awaitValue(timerCounter2, 1);
    assertEquals(1, timerCounter1.get());
    ClusterTests.failOnClusterError();
}

Example 5 with ClientSession

use of io.aeron.cluster.service.ClientSession in project aeron by real-logic.

the class ClusterNodeTest method launchTimedService.

private ClusteredServiceContainer launchTimedService() {
    final ClusteredService timedService = new StubClusteredService() {

        long clusterSessionId;

        long correlationId;

        String msg;

        public void onSessionMessage(final long clusterSessionId, final long correlationId, final long timestampMs, final DirectBuffer buffer, final int offset, final int length, final Header header) {
            this.clusterSessionId = clusterSessionId;
            this.correlationId = correlationId;
            this.msg = buffer.getStringWithoutLengthAscii(offset, length);
            cluster.scheduleTimer(correlationId, timestampMs + 100);
        }

        public void onTimerEvent(final long correlationId, final long timestampMs) {
            final String responseMsg = msg + "-scheduled";
            final ExpandableArrayBuffer buffer = new ExpandableArrayBuffer();
            buffer.putStringWithoutLengthAscii(0, responseMsg);
            final ClientSession clientSession = cluster.getClientSession(clusterSessionId);
            while (clientSession.offer(correlationId, buffer, 0, responseMsg.length()) < 0) {
                TestUtil.checkInterruptedStatus();
                Thread.yield();
            }
        }
    };
    return ClusteredServiceContainer.launch(new ClusteredServiceContainer.Context().clusteredService(timedService).errorHandler(Throwable::printStackTrace).deleteDirOnStart(true));
}