Search in sources :

Example 1 with EndpointDiscoveryService

use of io.gravitee.definition.model.services.discovery.EndpointDiscoveryService in project gravitee-gateway by gravitee-io.

the class EndpointDiscoveryConsulVerticle method startWatch.

private void startWatch(Api api) {
    EndpointDiscoveryService discoveryService = api.getServices().get(EndpointDiscoveryService.class);
    if (api.isEnabled() && discoveryService != null && discoveryService.getProvider() == EndpointDiscoveryProvider.CONSUL) {
        LOGGER.info("Add Consul.io support for API id[{}] name[{}]", api.getId(), api.getName());
        ConsulEndpointDiscoveryConfiguration providerConfiguration = (ConsulEndpointDiscoveryConfiguration) discoveryService.getConfiguration();
        URI consulUri = URI.create(providerConfiguration.getUrl());
        ConsulClientOptions options = new ConsulClientOptions().setHost(consulUri.getHost()).setPort((consulUri.getPort() != -1) ? consulUri.getPort() : CONSUL_DEFAULT_PORT).setDc(providerConfiguration.getDc()).setAclToken(providerConfiguration.getAcl());
        if (HTTPS_SCHEME.equalsIgnoreCase(consulUri.getScheme())) {
            // SSL is not configured but the endpoint scheme is HTTPS so let's enable the SSL on Vert.x HTTP client
            // automatically
            options.setSsl(true).setTrustAll(true);
        }
        LOGGER.info("Consul.io configuration: endpoint[{}] dc[{}] acl[{}]", consulUri.toString(), options.getDc(), options.getAclToken());
        Watch<ServiceEntryList> watch = Watch.service(providerConfiguration.getService(), vertx, options);
        watch.setHandler(event -> {
            if (event.succeeded()) {
                LOGGER.debug("Receiving a Consul.io watch event for service: name[{}]", providerConfiguration.getService());
                List<ServiceEntry> entries = event.nextResult().getList();
                // Handle new services or updated services
                for (ServiceEntry service : event.nextResult().getList()) {
                    Service service1 = service.getService();
                    service1.setNodeAddress(service.getNode().getAddress());
                    handleRegisterService(api, service1);
                }
                // Handle de-registered services
                if (event.prevResult() != null) {
                    List<ServiceEntry> oldEntries = event.prevResult().getList();
                    if (oldEntries.size() > entries.size()) {
                        // Select only de-registered services
                        oldEntries.removeAll(entries);
                        for (ServiceEntry oldEntry : oldEntries) {
                            handleDeregisterService(api, oldEntry.getService());
                        }
                    }
                }
            } else {
                LOGGER.info("Unexpected error while watching services catalog", event.cause());
            }
        }).start();
        this.watches.put(api.getId(), watch);
    }
}
Also used : ConsulEndpointDiscoveryConfiguration(io.gravitee.definition.model.services.discovery.consul.ConsulEndpointDiscoveryConfiguration) EventListener(io.gravitee.common.event.EventListener) HttpClientOptions(io.gravitee.definition.model.HttpClientOptions) Logger(org.slf4j.Logger) EndpointDiscoveryService(io.gravitee.definition.model.services.discovery.EndpointDiscoveryService) EndpointDiscoveryProvider(io.gravitee.definition.model.services.discovery.EndpointDiscoveryProvider) LoggerFactory(org.slf4j.LoggerFactory) Event(io.gravitee.common.event.Event) Autowired(org.springframework.beans.factory.annotation.Autowired) Set(java.util.Set) ConsulEndpointDiscoveryConfiguration(io.gravitee.definition.model.services.discovery.consul.ConsulEndpointDiscoveryConfiguration) HashMap(java.util.HashMap) DiscoveredEndpoint(io.gravitee.gateway.services.endpoint.discovery.consul.endpoint.DiscoveredEndpoint) Future(io.vertx.core.Future) io.vertx.ext.consul(io.vertx.ext.consul) Endpoint(io.gravitee.definition.model.Endpoint) List(java.util.List) Reactable(io.gravitee.gateway.reactor.Reactable) EventManager(io.gravitee.common.event.EventManager) AbstractVerticle(io.vertx.core.AbstractVerticle) Map(java.util.Map) Api(io.gravitee.gateway.handlers.api.definition.Api) URI(java.net.URI) ReactorEvent(io.gravitee.gateway.reactor.ReactorEvent) EndpointDiscoveryService(io.gravitee.definition.model.services.discovery.EndpointDiscoveryService) EndpointDiscoveryService(io.gravitee.definition.model.services.discovery.EndpointDiscoveryService) List(java.util.List) URI(java.net.URI)

Aggregations

Event (io.gravitee.common.event.Event)1 EventListener (io.gravitee.common.event.EventListener)1 EventManager (io.gravitee.common.event.EventManager)1 Endpoint (io.gravitee.definition.model.Endpoint)1 HttpClientOptions (io.gravitee.definition.model.HttpClientOptions)1 EndpointDiscoveryProvider (io.gravitee.definition.model.services.discovery.EndpointDiscoveryProvider)1 EndpointDiscoveryService (io.gravitee.definition.model.services.discovery.EndpointDiscoveryService)1 ConsulEndpointDiscoveryConfiguration (io.gravitee.definition.model.services.discovery.consul.ConsulEndpointDiscoveryConfiguration)1 Api (io.gravitee.gateway.handlers.api.definition.Api)1 Reactable (io.gravitee.gateway.reactor.Reactable)1 ReactorEvent (io.gravitee.gateway.reactor.ReactorEvent)1 DiscoveredEndpoint (io.gravitee.gateway.services.endpoint.discovery.consul.endpoint.DiscoveredEndpoint)1 AbstractVerticle (io.vertx.core.AbstractVerticle)1 Future (io.vertx.core.Future)1 io.vertx.ext.consul (io.vertx.ext.consul)1 URI (java.net.URI)1 HashMap (java.util.HashMap)1 List (java.util.List)1 Map (java.util.Map)1 Set (java.util.Set)1