Examples with Header io.aeron.logbuffer.Header used on opensource projects

Example 51 with Header

use of io.aeron.logbuffer.Header 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)));
}

Example 52 with Header

use of io.aeron.logbuffer.Header 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)));
}

Example 53 with Header

use of io.aeron.logbuffer.Header 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 54 with Header

use of io.aeron.logbuffer.Header in project aeron by real-logic.

the class EgressAdapterTest method onFragmentShouldInvokeOnMessageCallbackIfSessionIdMatches.

@Test
void onFragmentShouldInvokeOnMessageCallbackIfSessionIdMatches() {
    final int offset = 4;
    final long sessionId = 2973438724L;
    final long timestamp = -46328746238764832L;
    sessionMessageHeaderEncoder.wrapAndApplyHeader(buffer, offset, messageHeaderEncoder).clusterSessionId(sessionId).timestamp(timestamp);
    final EgressListener egressListener = mock(EgressListener.class);
    final Header header = new Header(0, 0);
    final EgressAdapter adapter = new EgressAdapter(egressListener, sessionId, mock(Subscription.class), 3);
    adapter.onFragment(buffer, offset, sessionMessageHeaderEncoder.encodedLength(), header);
    verify(egressListener).onMessage(sessionId, timestamp, buffer, offset + SESSION_HEADER_LENGTH, sessionMessageHeaderEncoder.encodedLength() - SESSION_HEADER_LENGTH, header);
    verifyNoMoreInteractions(egressListener);
}

Example 55 with Header

use of io.aeron.logbuffer.Header in project aeron by real-logic.

the class EgressAdapterTest method onFragmentShouldInvokeOnNewLeaderCallbackIfSessionIdMatches.

@Test
void onFragmentShouldInvokeOnNewLeaderCallbackIfSessionIdMatches() {
    final int offset = 0;
    final long clusterSessionId = 0;
    final long leadershipTermId = 6;
    final int leaderMemberId = 9999;
    final String ingressEndpoints = "ingress endpoints ...";
    newLeaderEventEncoder.wrapAndApplyHeader(buffer, offset, messageHeaderEncoder).leadershipTermId(leadershipTermId).clusterSessionId(clusterSessionId).leaderMemberId(leaderMemberId).ingressEndpoints(ingressEndpoints);
    final EgressListener egressListener = mock(EgressListener.class);
    final Header header = new Header(1, 3);
    final EgressAdapter adapter = new EgressAdapter(egressListener, clusterSessionId, mock(Subscription.class), 10);
    adapter.onFragment(buffer, offset, newLeaderEventEncoder.encodedLength(), header);
    verify(egressListener).onNewLeader(clusterSessionId, leadershipTermId, leaderMemberId, ingressEndpoints);
    verifyNoMoreInteractions(egressListener);
}