Examples with HeartbeatModule org.opennms.core.ipc.sink.camel.heartbeat.HeartbeatModule used on opensource projects

Example 1 with HeartbeatModule

use of org.opennms.core.ipc.sink.camel.heartbeat.HeartbeatModule in project opennms by OpenNMS.

the class HeartbeatSinkPerfIT method setUp.

@Before
public void setUp() throws Exception {
    super.setUp();
    final HeartbeatModule parallelHeartbeatModule = new HeartbeatModule() {

        @Override
        public int getNumConsumerThreads() {
            return NUM_CONSUMER_THREADS;
        }
    };
    final HeartbeatConsumer consumer = new HeartbeatConsumer(parallelHeartbeatModule, receivedMeter);
    consumerManager.registerConsumer(consumer);
    final MessageDispatcherFactory remoteMessageDispatcherFactory = context.getRegistry().lookupByNameAndType("camelRemoteMessageDispatcherFactory", MessageDispatcherFactory.class);
    final SyncDispatcher<Heartbeat> dispatcher = remoteMessageDispatcherFactory.createSyncDispatcher(HeartbeatModule.INSTANCE);
    // Fire up the generators
    generators = new ArrayList<>(NUM_GENERATORS);
    for (int k = 0; k < NUM_GENERATORS; k++) {
        final HeartbeatGenerator generator = new HeartbeatGenerator(dispatcher, RATE_PER_GENERATOR, sentMeter, sendTimer);
        generators.add(generator);
        generator.start();
    }
}

Example 2 with HeartbeatModule

use of org.opennms.core.ipc.sink.camel.heartbeat.HeartbeatModule in project opennms by OpenNMS.

the class HeartbeatSinkBlueprintIT method canConsumeMessagesInParallel.

@Test(timeout = 60000)
public void canConsumeMessagesInParallel() throws Exception {
    final int NUM_CONSUMER_THREADS = 7;
    final HeartbeatModule parallelHeartbeatModule = new HeartbeatModule() {

        @Override
        public int getNumConsumerThreads() {
            return NUM_CONSUMER_THREADS;
        }
    };
    ThreadLockingMessageConsumer<Heartbeat, Heartbeat> consumer = new ThreadLockingMessageConsumer<>(parallelHeartbeatModule);
    CompletableFuture<Integer> future = consumer.waitForThreads(NUM_CONSUMER_THREADS);
    consumerManager.registerConsumer(consumer);
    MessageDispatcherFactory remoteMessageDispatcherFactory = context.getRegistry().lookupByNameAndType("camelRemoteMessageDispatcherFactory", MessageDispatcherFactory.class);
    SyncDispatcher<Heartbeat> dispatcher = remoteMessageDispatcherFactory.createSyncDispatcher(HeartbeatModule.INSTANCE);
    HeartbeatGenerator generator = new HeartbeatGenerator(dispatcher, 100.0);
    generator.start();
    // Wait until we have NUM_CONSUMER_THREADS locked
    future.get();
    // Take a snooze
    Thread.sleep(TimeUnit.SECONDS.toMillis(5));
    // Verify that there aren't more than NUM_CONSUMER_THREADS waiting
    assertEquals(0, consumer.getNumExtraThreadsWaiting());
    generator.stop();
// This doesn't work because the consumer seems to leave NUM_CONSUMER_THREADS
// messages in-flight if it's unregistered before the context shuts down
// consumerManager.unregisterConsumer(consumer);
}

Example 3 with HeartbeatModule

use of org.opennms.core.ipc.sink.camel.heartbeat.HeartbeatModule in project opennms by OpenNMS.

the class HeartbeatSinkBlueprintIT method canProduceAndConsumerMessages.

@Test(timeout = 60000)
public void canProduceAndConsumerMessages() throws Exception {
    HeartbeatModule module = new HeartbeatModule();
    AtomicInteger heartbeatCount = new AtomicInteger();
    MessageConsumer<Heartbeat, Heartbeat> consumer = new MessageConsumer<Heartbeat, Heartbeat>() {

        @Override
        public SinkModule<Heartbeat, Heartbeat> getModule() {
            return module;
        }

        @Override
        public void handleMessage(Heartbeat heartbeat) {
            heartbeatCount.incrementAndGet();
        }
    };
    consumerManager.registerConsumer(consumer);
    SyncDispatcher<Heartbeat> localDispatcher = localMessageDispatcherFactory.createSyncDispatcher(module);
    localDispatcher.send(new Heartbeat());
    await().atMost(1, MINUTES).until(() -> heartbeatCount.get(), equalTo(1));
    MessageDispatcherFactory remoteMessageDispatcherFactory = context.getRegistry().lookupByNameAndType("camelRemoteMessageDispatcherFactory", MessageDispatcherFactory.class);
    SyncDispatcher<Heartbeat> remoteDispatcher = remoteMessageDispatcherFactory.createSyncDispatcher(module);
    remoteDispatcher.send(new Heartbeat());
    await().atMost(1, MINUTES).until(() -> heartbeatCount.get(), equalTo(2));
    consumerManager.unregisterConsumer(consumer);
}

Example 4 with HeartbeatModule

use of org.opennms.core.ipc.sink.camel.heartbeat.HeartbeatModule in project opennms by OpenNMS.

the class SinkBlueprintMessageFailureIT method doesntLogMessageBody.

@Test(timeout = 60000)
public void doesntLogMessageBody() throws Exception {
    HeartbeatModule module = new HeartbeatModule();
    final CountDownLatch consumed = new CountDownLatch(1);
    // Create a consumer that just throws exceptions during the exchange
    MessageConsumer<Heartbeat, Heartbeat> consumer = new MessageConsumer<Heartbeat, Heartbeat>() {

        @Override
        public SinkModule<Heartbeat, Heartbeat> getModule() {
            return module;
        }

        @Override
        public void handleMessage(Heartbeat heartbeat) {
            consumed.countDown();
            throw new IllegalStateException();
        }
    };
    consumerManager.registerConsumer(consumer);
    // Fetch the remote dispatcher from the blueprint context
    MessageDispatcherFactory remoteMessageDispatcherFactory = context.getRegistry().lookupByNameAndType("camelRemoteMessageDispatcherFactory", MessageDispatcherFactory.class);
    SyncDispatcher<Heartbeat> dispatcher = remoteMessageDispatcherFactory.createSyncDispatcher(HeartbeatModule.INSTANCE);
    dispatcher.send(new Heartbeat());
    consumed.await();
    // Sleep slightly longer to allow the body to be logged on the sink consumer listener thread
    Thread.sleep(2000);
    // Verify that the exchange error was logged
    MockLogAppender.assertLogMatched(Level.ERROR, "Message History");
    MockLogAppender.assertLogMatched(Level.ERROR, "queuingservice://OpenNMS.Sink.Heartbeat");
    // Verify that the message body was suppressed
    MockLogAppender.assertNoLogMatched(Level.ERROR, "<heartbeat/>");
    consumerManager.unregisterConsumer(consumer);
}