io.vertx.core.eventbus.MessageConsumer#completionHandler

Source File: VertxEventChannel.java From jstarcraft-core with Apache License 2.0

6 votes

@Override
public void registerMonitor(Set<Class> types, EventMonitor monitor) {
    try {
        for (Class type : types) {
            EventManager manager = managers.get(type);
            if (manager == null) {
                manager = new EventManager();
                managers.put(type, manager);
                // TODO 需要防止路径冲突
                EventHandler handler = new EventHandler(type, manager);
                MessageConsumer<byte[]> consumer = bus.consumer(name + StringUtility.DOT + type.getName(), handler);
                CountDownLatch latch = new CountDownLatch(1);
                consumer.completionHandler((register) -> {
                    latch.countDown();
                });
                latch.await();
                consumers.put(type, consumer);
            }
            manager.attachMonitor(monitor);
        }
    } catch (Exception exception) {
        throw new RuntimeException(exception);
    }
}

Source File: VertxExtension.java From weld-vertx with Apache License 2.0

5 votes

public void registerConsumers(Vertx vertx, Event<Object> event) {
    CountDownLatch latch = new CountDownLatch(consumerAddresses.size());
    for (String address : consumerAddresses) {
        MessageConsumer<?> consumer = vertx.eventBus().consumer(address, VertxHandler.from(vertx, event, address));
        consumer.completionHandler(ar -> {
            if (ar.succeeded()) {
                LOGGER.debug("Sucessfully registered event consumer for {0}", address);
                latch.countDown();
            } else {
                LOGGER.error("Cannot register event consumer for {0}", ar.cause(), address);
            }
        });
    }
    Context context = this.context;
    if (context == null && vertx != null) {
        context = vertx.getOrCreateContext();
    }
    long timeout = context != null ? context.config().getLong(CONSUMER_REGISTRATION_TIMEOUT_KEY, DEFAULT_CONSUMER_REGISTRATION_TIMEOUT)
            : DEFAULT_CONSUMER_REGISTRATION_TIMEOUT;
    try {
        if (!latch.await(timeout, TimeUnit.MILLISECONDS)) {
            throw new IllegalStateException(String.format("Message consumers not registered within %s ms [registered: %s, total: %s]", timeout,
                    latch.getCount(), consumerAddresses.size()));
        }
    } catch (InterruptedException e) {
        Thread.currentThread().interrupt();
        throw new RuntimeException(e);
    }
}

Source File: ReportingTest.java From vertx-unit with Apache License 2.0

5 votes

@org.junit.Test
public void testReportToEventBus() {
  MessageConsumer<JsonObject> consumer = vertx.eventBus().<JsonObject>consumer("the_address");
  consumer.handler(msg -> {
    if (msg.body().getString("type").equals(EventBusCollector.EVENT_TEST_SUITE_END)) {
      consumer.unregister();
      testComplete();
    }
  });
  consumer.completionHandler(ar -> {
    assertTrue(ar.succeeded());
    suite.run(vertx, new TestOptions().addReporter(new ReportOptions().setTo("bus:the_address")));
  });
  await();
}