use of io.aeron.cluster.client.AeronCluster in project aeron by real-logic.
the class ConsensusModuleHarness method makeRecordingLog.
public static long makeRecordingLog(final int numMessages, final int maxMessageLength, final Random random, final ConsensusModule.Context context) {
try (ConsensusModuleHarness harness = new ConsensusModuleHarness(context, new StubClusteredService(), null, true, false)) {
harness.awaitServiceOnStart();
final AeronCluster aeronCluster = AeronCluster.connect(new AeronCluster.Context().lock(new NoOpLock()));
final SessionDecorator sessionDecorator = new SessionDecorator(aeronCluster.clusterSessionId());
final Publication publication = aeronCluster.ingressPublication();
final ExpandableArrayBuffer msgBuffer = new ExpandableArrayBuffer(maxMessageLength);
for (int i = 0; i < numMessages; i++) {
final long messageCorrelationId = aeronCluster.context().aeron().nextCorrelationId();
final int length = (null == random) ? maxMessageLength : random.nextInt(maxMessageLength);
msgBuffer.putInt(0, i);
while (true) {
final long result = sessionDecorator.offer(publication, messageCorrelationId, msgBuffer, 0, length);
if (result > 0) {
break;
}
checkOfferResult(result);
TestUtil.checkInterruptedStatus();
Thread.yield();
}
}
harness.awaitServiceOnMessageCounter(numMessages);
return publication.position();
}
}
use of io.aeron.cluster.client.AeronCluster in project Aeron by real-logic.
the class ClusterNodeTest method shouldEchoMessageViaServiceUsingTryClaim.
@Test
@InterruptAfter(10)
public void shouldEchoMessageViaServiceUsingTryClaim() {
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) -> {
assertEquals(msg, buffer.getStringWithoutLengthAscii(offset, length));
messageCount.value += 1;
};
container = launchEchoService();
aeronCluster = connectToCluster(listener);
final BufferClaim bufferClaim = new BufferClaim();
long publicationResult;
do {
publicationResult = aeronCluster.tryClaim(msg.length(), bufferClaim);
if (publicationResult > 0) {
final int offset = bufferClaim.offset() + AeronCluster.SESSION_HEADER_LENGTH;
bufferClaim.buffer().putBytes(offset, msgBuffer, 0, msg.length());
bufferClaim.commit();
} else {
Tests.yield();
}
} while (publicationResult < 0);
offerMessage(msgBuffer, msg);
awaitResponse(messageCount);
ClusterTests.failOnClusterError();
}
use of io.aeron.cluster.client.AeronCluster in project Aeron by real-logic.
the class ClusterNodeTest method shouldEchoMessageViaServiceUsingDirectOffer.
@Test
@InterruptAfter(10)
public void shouldEchoMessageViaServiceUsingDirectOffer() {
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) -> {
assertEquals(msg, buffer.getStringWithoutLengthAscii(offset, length));
messageCount.value += 1;
};
container = launchEchoService();
aeronCluster = connectToCluster(listener);
offerMessage(msgBuffer, msg);
awaitResponse(messageCount);
ClusterTests.failOnClusterError();
}
use of io.aeron.cluster.client.AeronCluster in project Aeron by real-logic.
the class ClusterBackupTest method shouldBackupClusterNoSnapshotsAndNonEmptyLogWithFailure.
@Test
@InterruptAfter(60)
public void shouldBackupClusterNoSnapshotsAndNonEmptyLogWithFailure() {
final TestCluster cluster = aCluster().withStaticNodes(3).start();
systemTestWatcher.cluster(cluster);
final TestNode leaderOne = cluster.awaitLeader();
final int messageCount = 10;
final AeronCluster aeronCluster = cluster.connectClient();
cluster.sendMessages(messageCount);
cluster.awaitResponseMessageCount(messageCount);
cluster.awaitServicesMessageCount(messageCount);
final long logPosition = leaderOne.service().cluster().logPosition();
cluster.startClusterBackupNode(true);
aeronCluster.sendKeepAlive();
cluster.awaitBackupState(ClusterBackup.State.BACKING_UP);
aeronCluster.sendKeepAlive();
cluster.awaitBackupLiveLogPosition(logPosition);
cluster.stopNode(leaderOne);
final TestNode leaderTwo = cluster.awaitLeader();
cluster.awaitNewLeadershipEvent(1);
cluster.sendMessages(5);
cluster.awaitResponseMessageCount(messageCount + 5);
final long nextLogPosition = leaderTwo.service().cluster().logPosition();
cluster.awaitBackupState(ClusterBackup.State.BACKING_UP);
cluster.awaitBackupLiveLogPosition(nextLogPosition);
cluster.stopAllNodes();
final TestNode node = cluster.startStaticNodeFromBackup();
cluster.awaitLeader();
assertEquals(messageCount + 5, node.service().messageCount());
assertFalse(node.service().wasSnapshotLoaded());
}
use of io.aeron.cluster.client.AeronCluster in project Aeron by real-logic.
the class ClusterNetworkTopologyTest method pollUntilConnected.
private AeronCluster pollUntilConnected(final AeronCluster.AsyncConnect asyncConnect, final Selector selector) {
AeronCluster aeronCluster;
while (null == (aeronCluster = asyncConnect.poll())) {
pollSelector(selector);
Tests.sleep(1);
}
return aeronCluster;
}
Aggregations