Search in sources :

Example 16 with ExpandableArrayBuffer

use of org.agrona.ExpandableArrayBuffer 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));
}
Also used : DirectBuffer(org.agrona.DirectBuffer) Header(io.aeron.logbuffer.Header) ClientSession(io.aeron.cluster.service.ClientSession) ClusteredService(io.aeron.cluster.service.ClusteredService) ExpandableArrayBuffer(org.agrona.ExpandableArrayBuffer)

Example 17 with ExpandableArrayBuffer

use of org.agrona.ExpandableArrayBuffer in project aeron by real-logic.

the class ClusterTest method shouldEchoMessagesViaService.

@Test(timeout = 10_000)
public void shouldEchoMessagesViaService() throws InterruptedException {
    final Aeron aeron = client.context().aeron();
    final SessionDecorator sessionDecorator = new SessionDecorator(client.clusterSessionId());
    final Publication publication = client.ingressPublication();
    final ExpandableArrayBuffer msgBuffer = new ExpandableArrayBuffer();
    final long msgCorrelationId = aeron.nextCorrelationId();
    msgBuffer.putStringWithoutLengthAscii(0, MSG);
    final EchoConsumer consumer = new EchoConsumer(client.egressSubscription());
    final Thread thread = new Thread(consumer);
    thread.setName("consumer");
    thread.setDaemon(true);
    thread.start();
    for (int i = 0; i < MESSAGE_COUNT; i++) {
        while (sessionDecorator.offer(publication, msgCorrelationId, msgBuffer, 0, MSG.length()) < 0) {
            TestUtil.checkInterruptedStatus();
            Thread.yield();
        }
    }
    latch.await();
    for (final EchoService service : echoServices) {
        assertThat(service.messageCount(), is(MESSAGE_COUNT));
    }
}
Also used : Publication(io.aeron.Publication) SessionDecorator(io.aeron.cluster.client.SessionDecorator) Aeron(io.aeron.Aeron) ExpandableArrayBuffer(org.agrona.ExpandableArrayBuffer)

Example 18 with ExpandableArrayBuffer

use of org.agrona.ExpandableArrayBuffer in project Aeron by real-logic.

the class ClusterNodeRestartTest method launchService.

private void launchService(final AtomicLong msgCounter) {
    final ClusteredService service = new StubClusteredService() {

        private int nextCorrelationId = 0;

        private int counterValue = 0;

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

        public void onSessionMessage(final ClientSession session, final long timestamp, final DirectBuffer buffer, final int offset, final int length, final Header header) {
            final int sentValue = buffer.getInt(offset + MESSAGE_VALUE_OFFSET);
            assertEquals(counterValue, sentValue);
            counterValue++;
            serviceState.set(Integer.toString(counterValue));
            msgCounter.getAndIncrement();
            if (TIMER_MESSAGE_LENGTH == length) {
                final long correlationId = serviceCorrelationId(nextCorrelationId++);
                final long deadlineMs = timestamp + buffer.getLong(offset + TIMER_MESSAGE_DELAY_OFFSET);
                while (!cluster.scheduleTimer(correlationId, deadlineMs)) {
                    idleStrategy.idle();
                }
            }
        }

        public void onTakeSnapshot(final ExclusivePublication snapshotPublication) {
            final ExpandableArrayBuffer buffer = new ExpandableArrayBuffer();
            int length = 0;
            buffer.putInt(length, nextCorrelationId);
            length += SIZE_OF_INT;
            buffer.putInt(length, counterValue);
            length += SIZE_OF_INT;
            length += buffer.putStringAscii(length, Integer.toString(counterValue));
            snapshotPublication.offer(buffer, 0, length);
        }
    };
    container = ClusteredServiceContainer.launch(new ClusteredServiceContainer.Context().clusteredService(service).terminationHook(ClusterTests.NOOP_TERMINATION_HOOK).errorHandler(ClusterTests.errorHandler(0)));
}
Also used : ClusterTests(io.aeron.test.cluster.ClusterTests) AeronCluster(io.aeron.cluster.client.AeronCluster) BeforeEach(org.junit.jupiter.api.BeforeEach) StubClusteredService(io.aeron.test.cluster.StubClusteredService) AtomicCounter(org.agrona.concurrent.status.AtomicCounter) ClusteredServiceContainer(io.aeron.cluster.service.ClusteredServiceContainer) io.aeron.test(io.aeron.test) CountersReader(org.agrona.concurrent.status.CountersReader) CLUSTER_MEMBERS(io.aeron.cluster.ClusterTestConstants.CLUSTER_MEMBERS) ClusteredService(io.aeron.cluster.service.ClusteredService) SIZE_OF_INT(org.agrona.BitUtil.SIZE_OF_INT) AtomicReference(java.util.concurrent.atomic.AtomicReference) ExtendWith(org.junit.jupiter.api.extension.ExtendWith) SIZE_OF_LONG(org.agrona.BitUtil.SIZE_OF_LONG) Publication(io.aeron.Publication) ArchiveThreadingMode(io.aeron.archive.ArchiveThreadingMode) CloseHelper(org.agrona.CloseHelper) ExclusivePublication(io.aeron.ExclusivePublication) PrintStream(java.io.PrintStream) MediaDriver(io.aeron.driver.MediaDriver) Image(io.aeron.Image) Archive(io.aeron.archive.Archive) ExpandableArrayBuffer(org.agrona.ExpandableArrayBuffer) File(java.io.File) Mockito.verify(org.mockito.Mockito.verify) Test(org.junit.jupiter.api.Test) TimeUnit(java.util.concurrent.TimeUnit) CountDownLatch(java.util.concurrent.CountDownLatch) AtomicLong(java.util.concurrent.atomic.AtomicLong) Cluster(io.aeron.cluster.service.Cluster) AfterEach(org.junit.jupiter.api.AfterEach) Header(io.aeron.logbuffer.Header) ClientSession(io.aeron.cluster.service.ClientSession) ThreadingMode(io.aeron.driver.ThreadingMode) Assertions(org.junit.jupiter.api.Assertions) INGRESS_ENDPOINTS(io.aeron.cluster.ClusterTestConstants.INGRESS_ENDPOINTS) FragmentHandler(io.aeron.logbuffer.FragmentHandler) DirectBuffer(org.agrona.DirectBuffer) Mockito.mock(org.mockito.Mockito.mock) AeronCluster(io.aeron.cluster.client.AeronCluster) Cluster(io.aeron.cluster.service.Cluster) StubClusteredService(io.aeron.test.cluster.StubClusteredService) ClusteredService(io.aeron.cluster.service.ClusteredService) ExclusivePublication(io.aeron.ExclusivePublication) Image(io.aeron.Image) ExpandableArrayBuffer(org.agrona.ExpandableArrayBuffer) DirectBuffer(org.agrona.DirectBuffer) Header(io.aeron.logbuffer.Header) FragmentHandler(io.aeron.logbuffer.FragmentHandler) ClientSession(io.aeron.cluster.service.ClientSession) StubClusteredService(io.aeron.test.cluster.StubClusteredService) ClusteredServiceContainer(io.aeron.cluster.service.ClusteredServiceContainer)

Example 19 with ExpandableArrayBuffer

use of org.agrona.ExpandableArrayBuffer in project Aeron by real-logic.

the class ClusterNodeTest method shouldScheduleEventInService.

@Test
@InterruptAfter(10)
public void shouldScheduleEventInService() {
    final ExpandableArrayBuffer msgBuffer = new ExpandableArrayBuffer();
    final String msg = "Hello World!";
    msgBuffer.putStringWithoutLengthAscii(0, msg);
    final MutableInteger messageCount = new MutableInteger();
    final EgressListener listener = (clusterSessionId, timestamp, buffer, offset, length, header) -> {
        final String expected = msg + "-scheduled";
        assertEquals(expected, buffer.getStringWithoutLengthAscii(offset, length));
        messageCount.value += 1;
    };
    container = launchTimedService();
    aeronCluster = connectToCluster(listener);
    offerMessage(msgBuffer, msg);
    awaitResponse(messageCount);
    ClusterTests.failOnClusterError();
}
Also used : ClusterTests(io.aeron.test.cluster.ClusterTests) AeronCluster(io.aeron.cluster.client.AeronCluster) Tests(io.aeron.test.Tests) BeforeEach(org.junit.jupiter.api.BeforeEach) StubClusteredService(io.aeron.test.cluster.StubClusteredService) ClusteredServiceContainer(io.aeron.cluster.service.ClusteredServiceContainer) CLUSTER_MEMBERS(io.aeron.cluster.ClusterTestConstants.CLUSTER_MEMBERS) ClusteredService(io.aeron.cluster.service.ClusteredService) ExtendWith(org.junit.jupiter.api.extension.ExtendWith) BufferClaim(io.aeron.logbuffer.BufferClaim) Assertions.assertEquals(org.junit.jupiter.api.Assertions.assertEquals) ArchiveThreadingMode(io.aeron.archive.ArchiveThreadingMode) MutableInteger(org.agrona.collections.MutableInteger) CloseHelper(org.agrona.CloseHelper) MediaDriver(io.aeron.driver.MediaDriver) InterruptingTestCallback(io.aeron.test.InterruptingTestCallback) Archive(io.aeron.archive.Archive) ExpandableArrayBuffer(org.agrona.ExpandableArrayBuffer) Test(org.junit.jupiter.api.Test) EgressListener(io.aeron.cluster.client.EgressListener) InterruptAfter(io.aeron.test.InterruptAfter) AfterEach(org.junit.jupiter.api.AfterEach) Header(io.aeron.logbuffer.Header) ClientSession(io.aeron.cluster.service.ClientSession) ThreadingMode(io.aeron.driver.ThreadingMode) Assertions.assertTrue(org.junit.jupiter.api.Assertions.assertTrue) INGRESS_ENDPOINTS(io.aeron.cluster.ClusterTestConstants.INGRESS_ENDPOINTS) DirectBuffer(org.agrona.DirectBuffer) MutableInteger(org.agrona.collections.MutableInteger) ExpandableArrayBuffer(org.agrona.ExpandableArrayBuffer) EgressListener(io.aeron.cluster.client.EgressListener) Test(org.junit.jupiter.api.Test) InterruptAfter(io.aeron.test.InterruptAfter)

Example 20 with ExpandableArrayBuffer

use of org.agrona.ExpandableArrayBuffer in project Aeron by real-logic.

the class ClusterNodeTest method launchTimedService.

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

        private final ExpandableArrayBuffer buffer = new ExpandableArrayBuffer();

        private long clusterSessionId;

        private int nextCorrelationId;

        private String msg;

        public void onSessionMessage(final ClientSession session, final long timestamp, final DirectBuffer buffer, final int offset, final int length, final Header header) {
            clusterSessionId = session.id();
            msg = buffer.getStringWithoutLengthAscii(offset, length);
            final long correlationId = serviceCorrelationId(nextCorrelationId++);
            idleStrategy.reset();
            while (!cluster.scheduleTimer(correlationId, timestamp + 100)) {
                idleStrategy.idle();
            }
        }

        public void onTimerEvent(final long correlationId, final long timestamp) {
            final String responseMsg = msg + "-scheduled";
            buffer.putStringWithoutLengthAscii(0, responseMsg);
            final ClientSession clientSession = cluster.getClientSession(clusterSessionId);
            idleStrategy.reset();
            while (clientSession.offer(buffer, 0, responseMsg.length()) < 0) {
                idleStrategy.idle();
            }
        }
    };
    return ClusteredServiceContainer.launch(new ClusteredServiceContainer.Context().clusteredService(clusteredService).terminationHook(ClusterTests.NOOP_TERMINATION_HOOK).errorHandler(ClusterTests.errorHandler(0)));
}
Also used : DirectBuffer(org.agrona.DirectBuffer) Header(io.aeron.logbuffer.Header) ClientSession(io.aeron.cluster.service.ClientSession) StubClusteredService(io.aeron.test.cluster.StubClusteredService) ClusteredService(io.aeron.cluster.service.ClusteredService) StubClusteredService(io.aeron.test.cluster.StubClusteredService) ExpandableArrayBuffer(org.agrona.ExpandableArrayBuffer)

Aggregations

ExpandableArrayBuffer (org.agrona.ExpandableArrayBuffer)21 Header (io.aeron.logbuffer.Header)11 DirectBuffer (org.agrona.DirectBuffer)11 ClusteredService (io.aeron.cluster.service.ClusteredService)10 AeronCluster (io.aeron.cluster.client.AeronCluster)9 ClientSession (io.aeron.cluster.service.ClientSession)9 StubClusteredService (io.aeron.test.cluster.StubClusteredService)8 Archive (io.aeron.archive.Archive)7 ArchiveThreadingMode (io.aeron.archive.ArchiveThreadingMode)7 CLUSTER_MEMBERS (io.aeron.cluster.ClusterTestConstants.CLUSTER_MEMBERS)7 INGRESS_ENDPOINTS (io.aeron.cluster.ClusterTestConstants.INGRESS_ENDPOINTS)7 ClusteredServiceContainer (io.aeron.cluster.service.ClusteredServiceContainer)7 MediaDriver (io.aeron.driver.MediaDriver)7 ThreadingMode (io.aeron.driver.ThreadingMode)7 ClusterTests (io.aeron.test.cluster.ClusterTests)7 CloseHelper (org.agrona.CloseHelper)7 AfterEach (org.junit.jupiter.api.AfterEach)7 BeforeEach (org.junit.jupiter.api.BeforeEach)7 Test (org.junit.jupiter.api.Test)7 ExtendWith (org.junit.jupiter.api.extension.ExtendWith)7