Examples with ServiceInstance org.springframework.cloud.client.ServiceInstance used on opensource projects

Example 21 with ServiceInstance

use of org.springframework.cloud.client.ServiceInstance in project spring-cloud-netflix by spring-cloud.

the class RibbonLoadBalancerClientTests method testReconstructUriWithSecureClientConfig.

@Test
public void testReconstructUriWithSecureClientConfig() throws Exception {
    RibbonServer server = getRibbonServer();
    IClientConfig config = mock(IClientConfig.class);
    when(config.get(CommonClientConfigKey.IsSecure)).thenReturn(true);
    when(clientFactory.getClientConfig(server.getServiceId())).thenReturn(config);
    RibbonLoadBalancerClient client = getRibbonLoadBalancerClient(server);
    ServiceInstance serviceInstance = client.choose(server.getServiceId());
    URI uri = client.reconstructURI(serviceInstance, new URL("http://" + server.getServiceId()).toURI());
    assertEquals(server.getHost(), uri.getHost());
    assertEquals(server.getPort(), uri.getPort());
    assertEquals("https", uri.getScheme());
}

Example 22 with ServiceInstance

use of org.springframework.cloud.client.ServiceInstance in project spring-cloud-netflix by spring-cloud.

the class RibbonLoadBalancerClientTests method testReconstructUriWithPath.

private void testReconstructUriWithPath(String scheme, String path) {
    RibbonServer server = getRibbonServer();
    IClientConfig config = mock(IClientConfig.class);
    when(config.get(CommonClientConfigKey.IsSecure)).thenReturn(true);
    when(clientFactory.getClientConfig(server.getServiceId())).thenReturn(config);
    RibbonLoadBalancerClient client = getRibbonLoadBalancerClient(server);
    ServiceInstance serviceInstance = client.choose(server.getServiceId());
    URI expanded = new DefaultUriBuilderFactory().expand(scheme + "://" + server.getServiceId() + path);
    URI reconstructed = client.reconstructURI(serviceInstance, expanded);
    assertThat(reconstructed).hasPath(path);
}

Example 23 with ServiceInstance

use of org.springframework.cloud.client.ServiceInstance in project spring-cloud-netflix by spring-cloud.

the class RibbonLoadBalancerClientTests method testChoose.

@Test
public void testChoose() {
    RibbonServer server = getRibbonServer();
    RibbonLoadBalancerClient client = getRibbonLoadBalancerClient(server);
    ServiceInstance serviceInstance = client.choose(server.getServiceId());
    assertServiceInstance(server, serviceInstance);
}

Example 24 with ServiceInstance

use of org.springframework.cloud.client.ServiceInstance in project spring-cloud-netflix by spring-cloud.

the class RibbonLoadBalancerClientTests method testReconstructSchemelessUriWithoutClientConfig.

public void testReconstructSchemelessUriWithoutClientConfig(RibbonServer server, String expectedScheme) throws Exception {
    IClientConfig config = mock(IClientConfig.class);
    when(config.get(CommonClientConfigKey.IsSecure)).thenReturn(null);
    when(clientFactory.getClientConfig(server.getServiceId())).thenReturn(config);
    RibbonLoadBalancerClient client = getRibbonLoadBalancerClient(server);
    ServiceInstance serviceInstance = client.choose(server.getServiceId());
    URI uri = client.reconstructURI(serviceInstance, new URI("//" + server.getServiceId()));
    assertEquals(server.getHost(), uri.getHost());
    assertEquals(server.getPort(), uri.getPort());
    assertEquals(expectedScheme, uri.getScheme());
}

Example 25 with ServiceInstance

use of org.springframework.cloud.client.ServiceInstance in project spring-cloud-netflix by spring-cloud.

the class RetryableRibbonLoadBalancingHttpClient method execute.

@Override
public RibbonApacheHttpResponse execute(final RibbonApacheHttpRequest request, final IClientConfig configOverride) throws Exception {
    final RequestConfig.Builder builder = RequestConfig.custom();
    IClientConfig config = configOverride != null ? configOverride : this.config;
    RibbonProperties ribbon = RibbonProperties.from(config);
    builder.setConnectTimeout(ribbon.connectTimeout(this.connectTimeout));
    builder.setSocketTimeout(ribbon.readTimeout(this.readTimeout));
    builder.setRedirectsEnabled(ribbon.isFollowRedirects(this.followRedirects));
    final RequestConfig requestConfig = builder.build();
    final LoadBalancedRetryPolicy retryPolicy = loadBalancedRetryFactory.createRetryPolicy(this.getClientName(), this);
    RetryCallback<RibbonApacheHttpResponse, Exception> retryCallback = context -> {
        // on retries the policy will choose the server and set it in the context
        // extract the server and update the request being made
        RibbonApacheHttpRequest newRequest = request;
        if (context instanceof LoadBalancedRetryContext) {
            ServiceInstance service = ((LoadBalancedRetryContext) context).getServiceInstance();
            validateServiceInstance(service);
            if (service != null) {
                // Reconstruct the request URI using the host and port set in the retry context
                newRequest = newRequest.withNewUri(UriComponentsBuilder.newInstance().host(service.getHost()).scheme(service.getUri().getScheme()).userInfo(newRequest.getURI().getUserInfo()).port(service.getPort()).path(newRequest.getURI().getPath()).query(newRequest.getURI().getQuery()).fragment(newRequest.getURI().getFragment()).build().encode().toUri());
            }
        }
        newRequest = getSecureRequest(newRequest, configOverride);
        HttpUriRequest httpUriRequest = newRequest.toRequest(requestConfig);
        final HttpResponse httpResponse = RetryableRibbonLoadBalancingHttpClient.this.delegate.execute(httpUriRequest);
        if (retryPolicy.retryableStatusCode(httpResponse.getStatusLine().getStatusCode())) {
            throw new HttpClientStatusCodeException(RetryableRibbonLoadBalancingHttpClient.this.clientName, httpResponse, HttpClientUtils.createEntity(httpResponse), httpUriRequest.getURI());
        }
        return new RibbonApacheHttpResponse(httpResponse, httpUriRequest.getURI());
    };
    LoadBalancedRecoveryCallback<RibbonApacheHttpResponse, HttpResponse> recoveryCallback = new LoadBalancedRecoveryCallback<RibbonApacheHttpResponse, HttpResponse>() {

        @Override
        protected RibbonApacheHttpResponse createResponse(HttpResponse response, URI uri) {
            return new RibbonApacheHttpResponse(response, uri);
        }
    };
    return this.executeWithRetry(request, retryPolicy, retryCallback, recoveryCallback);
}