com.orbitz.consul.Consul#Builder

Source File: ConsulFactory.java From dropwizard-consul with Apache License 2.0

6 votes

@JsonIgnore
public Consul build() {

  final Consul.Builder builder = Consul.builder().withHostAndPort(endpoint).withPing(servicePing);

  aclToken.ifPresent(
      token -> {
        // setting both acl token here and with header, supplying an auth
        // header. This should cover both use cases: endpoint supports
        // legacy ?token query param and other case in which endpoint
        // requires an X-Consul-Token header.
        // @see https://www.consul.io/api/index.html#acls
        builder.withAclToken(token).withHeaders(ImmutableMap.of(CONSUL_AUTH_HEADER_KEY, token));
      });

  return builder.build();
}

Source File: ConsulService.java From ARCHIVE-wildfly-swarm with Apache License 2.0

6 votes

@Override
public void start(StartContext startContext) throws StartException {

    Consul.Builder builder = Consul.builder();


    // pool because of multiple threads.
    ResteasyClientBuilder clientBuilder = new ResteasyClientBuilder();
    clientBuilder = clientBuilder.connectionPoolSize(20);

    builder.withClientBuilder(clientBuilder);
    builder.withUrl(this.url);

    try {
        this.consul = builder.build();
    } catch (Exception e) {
        throw new StartException("Failed to connect consul at "+url, e);
    }
}

Source File: ClusterModuleConsulProviderTest.java From skywalking with Apache License 2.0

6 votes

@Test
@SuppressWarnings("unchecked")
public void prepare() throws Exception {
    ClusterModuleConsulConfig consulConfig = new ClusterModuleConsulConfig();
    consulConfig.setHostPort("10.0.0.1:1000,10.0.0.2:1001");
    Whitebox.setInternalState(provider, "config", consulConfig);

    Consul consulClient = mock(Consul.class);
    Consul.Builder builder = mock(Consul.Builder.class);
    when(builder.build()).thenReturn(consulClient);

    PowerMockito.mockStatic(Consul.class);
    when(Consul.builder()).thenReturn(builder);
    when(builder.withConnectTimeoutMillis(anyLong())).thenReturn(builder);

    when(builder.withMultipleHostAndPort(anyCollection(), anyLong())).thenReturn(builder);
    provider.prepare();

    ArgumentCaptor<Collection> addressCaptor = ArgumentCaptor.forClass(Collection.class);
    ArgumentCaptor<Long> timeCaptor = ArgumentCaptor.forClass(long.class);
    verify(builder).withMultipleHostAndPort(addressCaptor.capture(), timeCaptor.capture());

    List<HostAndPort> address = (List<HostAndPort>) addressCaptor.getValue();
    assertEquals(2, address.size());
    assertEquals(Lists.newArrayList(HostAndPort.fromParts("10.0.0.1", 1000), HostAndPort.fromParts("10.0.0.2", 1001)), address);
}

Source File: ClusterModuleConsulProviderTest.java From skywalking with Apache License 2.0

6 votes

@Test
public void prepareSingle() throws Exception {
    ClusterModuleConsulConfig consulConfig = new ClusterModuleConsulConfig();
    consulConfig.setHostPort("10.0.0.1:1000");
    Whitebox.setInternalState(provider, "config", consulConfig);

    Consul consulClient = mock(Consul.class);
    Consul.Builder builder = mock(Consul.Builder.class);
    when(builder.build()).thenReturn(consulClient);

    PowerMockito.mockStatic(Consul.class);
    when(Consul.builder()).thenReturn(builder);
    when(builder.withConnectTimeoutMillis(anyLong())).thenCallRealMethod();

    when(builder.withHostAndPort(any())).thenReturn(builder);

    provider.prepare();

    ArgumentCaptor<HostAndPort> hostAndPortArgumentCaptor = ArgumentCaptor.forClass(HostAndPort.class);
    verify(builder).withHostAndPort(hostAndPortArgumentCaptor.capture());

    HostAndPort address = hostAndPortArgumentCaptor.getValue();
    assertEquals(HostAndPort.fromParts("10.0.0.1", 1000), address);
}

Source File: ConsulClient.java From pragmatic-microservices-lab with MIT License

5 votes

public static Consul build() {
    Consul.Builder consulBuilder = Consul.builder();
    ConfigProvider.getConfig()
            .getOptionalValue(CONFIG_CONSUL_URL, String.class)
            .ifPresent(url -> {
                consulBuilder.withUrl(url);
            });
    return consulBuilder.build();
}

Source File: ConsulService.java From thorntail with Apache License 2.0

5 votes

@Override
public void start(StartContext startContext) throws StartException {

    Consul.Builder builder = Consul.builder();

    builder.withUrl(this.url);

    try {
        this.consul = builder.build();
    } catch (Exception e) {
        throw new StartException("Failed to connect consul at " + url, e);
    }
}

Source File: ConsulConfigurationWatcherRegister.java From skywalking with Apache License 2.0

5 votes

public ConsulConfigurationWatcherRegister(ConsulConfigurationCenterSettings settings) {
    super(settings.getPeriod());

    this.configItemKeyedByName = new ConcurrentHashMap<>();
    this.cachesByKey = new ConcurrentHashMap<>();

    List<HostAndPort> hostAndPorts = Splitter.on(",")
                                             .splitToList(settings.getHostAndPorts())
                                             .parallelStream()
                                             .map(hostAndPort -> HostAndPort.fromString(hostAndPort)
                                                                            .withDefaultPort(DEFAULT_PORT))
                                             .collect(Collectors.toList());

    Consul.Builder builder = Consul.builder().withConnectTimeoutMillis(3000);

    if (hostAndPorts.size() == 1) {
        builder.withHostAndPort(hostAndPorts.get(0));
    } else {
        builder.withMultipleHostAndPort(hostAndPorts, 5000);
    }

    if (StringUtils.isNotEmpty(settings.getAclToken())) {
        builder.withAclToken(settings.getAclToken());
    }

    consul = builder.build().keyValueClient();
}

Source File: ClusterModuleConsulProvider.java From skywalking with Apache License 2.0

5 votes

@Override
public void prepare() throws ServiceNotProvidedException, ModuleStartException {
    try {
        List<Address> addressList = ConnectUtils.parse(config.getHostPort());

        List<HostAndPort> hostAndPorts = new ArrayList<>();
        for (Address address : addressList) {
            hostAndPorts.add(HostAndPort.fromParts(address.getHost(), address.getPort()));
        }

        Consul.Builder consulBuilder = Consul.builder()
                                             //                    we should set this value or it will be blocked forever
                                             .withConnectTimeoutMillis(3000);

        if (StringUtils.isNotEmpty(config.getAclToken())) {
            consulBuilder.withAclToken(config.getAclToken());
        }

        if (hostAndPorts.size() > 1) {
            client = consulBuilder.withMultipleHostAndPort(hostAndPorts, 5000).build();
        } else {
            client = consulBuilder.withHostAndPort(hostAndPorts.get(0)).build();
        }
    } catch (ConnectStringParseException | ConsulException e) {
        throw new ModuleStartException(e.getMessage(), e);
    }

    ConsulCoordinator coordinator = new ConsulCoordinator(config, client);
    this.registerServiceImplementation(ClusterRegister.class, coordinator);
    this.registerServiceImplementation(ClusterNodesQuery.class, coordinator);
}

Source File: ConsulBundle.java From dropwizard-consul with Apache License 2.0

4 votes

@Override
public void initialize(Bootstrap<?> bootstrap) {
  // Replace variables with values from Consul KV. Please override
  // getConsulAgentHost() and getConsulAgentPort() if Consul is not
  // listening on the default localhost:8500.
  try {
    LOGGER.debug("Connecting to Consul at {}:{}", getConsulAgentHost(), getConsulAgentPort());

    final Consul.Builder builder =
        Consul.builder()
            .withHostAndPort(HostAndPort.fromParts(getConsulAgentHost(), getConsulAgentPort()));

    getConsulAclToken()
        .ifPresent(
            token -> {
              // setting both ACL token here and with header, supplying an
              // auth header. This should cover both use cases: endpoint
              // supports legacy ?token query param and other case
              // in which endpoint requires an X-Consul-Token header.
              // @see https://www.consul.io/api/index.html#acls

              LOGGER.debug("Using Consul ACL token: {}", token);

              builder
                  .withAclToken(token)
                  .withHeaders(ImmutableMap.of(CONSUL_AUTH_HEADER_KEY, token));
            });

    // using Consul as a configuration substitution provider
    bootstrap.setConfigurationSourceProvider(
        new SubstitutingSourceProvider(
            bootstrap.getConfigurationSourceProvider(),
            new ConsulSubstitutor(builder.build(), strict, substitutionInVariables)));

  } catch (ConsulException e) {
    LOGGER.warn(
        "Unable to query Consul running on {}:{}," + " disabling configuration substitution",
        getConsulAgentHost(),
        getConsulAgentPort(),
        e);
  }
}

Source File: ConsulModule.java From nano-framework with Apache License 2.0

4 votes

private Consul build(final ConsulConfig cfg) {
    final Consul.Builder builder = Consul.builder();

    final HostAndPort hostAndPort = cfg.getHostAndPort();
    if (hostAndPort != null) {
        builder.withHostAndPort(cfg.getHostAndPort());
    }

    final String username = cfg.getUsernaem();
    final String password = cfg.getPassword();
    if (username != null && password != null) {
        builder.withBasicAuth(cfg.getUsernaem(), cfg.getPassword());
    }

    final URL url = cfg.getUrl();
    if (url != null) {
        builder.withUrl(cfg.getUrl());
    }

    final String token = cfg.getToken();
    if (token != null) {
        builder.withAclToken(cfg.getToken());
    }

    final Map<String, String> headers = cfg.getHeaders();
    if (!CollectionUtils.isEmpty(headers)) {
        builder.withHeaders(cfg.getHeaders());
    }

    final Long connectTimeout = cfg.getConnectTimeout();
    if (connectTimeout != null) {
        builder.withConnectTimeoutMillis(cfg.getConnectTimeout());
    }

    final Long readTimeout = cfg.getReadTimeout();
    if (readTimeout != null) {
        builder.withReadTimeoutMillis(cfg.getReadTimeout());
    }

    final Long writeTimeout = cfg.getWriteTimeout();
    if (writeTimeout != null) {
        builder.withWriteTimeoutMillis(cfg.getWriteTimeout());
    }

    return builder.build();
}

Java Code Examples for com.orbitz.consul.Consul#Builder