com.github.edgar615.util.vertx.deployment.MainVerticle Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vertx-util Show documentation
There is a newer version: 1.0.14
package com.github.edgar615.util.vertx.deployment;

import io.vertx.core.AbstractVerticle;
import io.vertx.core.AsyncResult;
import io.vertx.core.Future;
import io.vertx.core.Handler;
import io.vertx.core.Vertx;
import io.vertx.core.eventbus.MessageCodec;
import io.vertx.core.json.JsonArray;
import io.vertx.core.json.JsonObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.concurrent.atomic.AtomicInteger;

/**
 * 用于启动Vert.x应用的组件，通过配置文件指定需要启动哪些Verticle，避免硬编码.
 * 
 * 
 *     "verticles": {
 * "MetricsVerticle": {
 * "class": "com.groupon.vertx.utils.MetricsVerticle",
 * "instances": 1,
 * "worker": true,
 * "config": { }
 * },
 * "ExampleVerticle": {
 * "class": "com.groupon.example.verticle.ExampleVerticle",
 * "instances": 1,
 * "worker": true,
 * "config": { },
 * "dependencies": [ "MetricsVerticle" ]
 * }
 * }
 * 
 *
 * @author Edgar  Date 2017/6/12
 */
public class MainVerticle extends AbstractVerticle {
  private static final Logger LOGGER = LoggerFactory.getLogger(MainVerticle.class);

  private final AtomicInteger verticleCount = new AtomicInteger(0);

  @Override
  public void start(Future startFuture) throws Exception {

    //注册codec
    try {
      registerMessageCodecs(vertx, config());
    } catch (final CodecRegistrationException e) {
      LOGGER.error("start", "abort", "Shutting down due to one or more errors", e);
      vertx.close();
      return;
    }

    //启动
    MainVerticleDeployment deployment = new MainVerticleDeployment(config().copy());

    deployment.getDeployments().stream()
            .filter(o -> o.getDependencyVerticles().isEmpty())
            .forEach(d -> d.deploy(vertx,
                                   createHandler(deployment, d.getVerticleName(), startFuture)));
  }

  private void checkAndDeploy(MainVerticleDeployment deployment, String verticle,
                              Future startFuture) {
    deployment.getDeployments().forEach(d -> {
      boolean checkResult = d.checkPrecondition(verticle);
      if (checkResult) {
        d.deploy(vertx, createHandler(deployment, d.getVerticleName(), startFuture));
      }
    });
    if (verticleCount.get() == deployment.getDeployments().size()) {
      if (!startFuture.isComplete()) {
        startFuture.complete();
      }
    }
  }

  private Handler> createHandler(MainVerticleDeployment deployment,
                                                     String verticle, Future startFuture) {
    return ar -> {
      if (ar.failed()) {
        ar.cause().printStackTrace();
        startFuture.fail(ar.cause());
        return;
      }
      verticleCount.incrementAndGet();
      checkAndDeploy(deployment, verticle, startFuture);
    };
  }

  private void registerMessageCodecs(
          final Vertx vertx,
          final JsonObject config) {

    final JsonArray messageCodecs = config.getJsonArray("messageCodecs", new JsonArray());
    for (final Object messageCodecClassNameObject : messageCodecs.getList()) {
      if (messageCodecClassNameObject instanceof String) {
        final String messageCodecClassName = (String) messageCodecClassNameObject;
        try {
          final MessageCodec messageCodec
                  = (MessageCodec) Class.forName(messageCodecClassName).newInstance();
          vertx.eventBus().registerCodec(messageCodec);
        } catch (final InstantiationException | IllegalAccessException | ClassNotFoundException e) {
          LOGGER.warn("Failed to instantiate message codec:{}",
                      messageCodecClassName, e);
          throw new CodecRegistrationException(
                  String.format(
                          "Failed to instantiate message codec %s",
                          messageCodecClassName),
                  e);
        }
      } else {
        LOGGER.warn("Ignoring non-string message codec class name:{}",
                    messageCodecClassNameObject);
      }
    }
  }

}