Examples with ILoadBalancer com.netflix.loadbalancer.ILoadBalancer used on opensource projects

Example 16 with ILoadBalancer

use of com.netflix.loadbalancer.ILoadBalancer in project spring-cloud-gateway by spring-cloud.

the class LoadBalancerClientFilterTests method testFilter.

private ServerWebExchange testFilter(ServerWebExchange exchange, URI uri) {
    exchange.getAttributes().put(GATEWAY_REQUEST_URL_ATTR, uri);
    ArgumentCaptor<ServerWebExchange> captor = ArgumentCaptor.forClass(ServerWebExchange.class);
    when(chain.filter(captor.capture())).thenReturn(Mono.empty());
    SpringClientFactory clientFactory = mock(SpringClientFactory.class);
    ILoadBalancer loadBalancer = mock(ILoadBalancer.class);
    when(clientFactory.getLoadBalancerContext("service1")).thenReturn(new RibbonLoadBalancerContext(loadBalancer));
    when(clientFactory.getLoadBalancer("service1")).thenReturn(loadBalancer);
    when(loadBalancer.chooseServer(any())).thenReturn(new Server("service1-host1", 8081));
    RibbonLoadBalancerClient client = new RibbonLoadBalancerClient(clientFactory);
    LoadBalancerClientFilter filter = new LoadBalancerClientFilter(client);
    filter.filter(exchange, chain);
    return captor.getValue();
}

Example 17 with ILoadBalancer

use of com.netflix.loadbalancer.ILoadBalancer in project spring-cloud-netflix by spring-cloud.

the class RibbonLoadBalancingHttpClientTests method retryListenerTest.

@Test
public void retryListenerTest() throws Exception {
    int retriesNextServer = 1;
    int retriesSameServer = 1;
    boolean retryable = true;
    boolean retryOnAllOps = true;
    String serviceName = "listener";
    String host = serviceName;
    int port = 80;
    HttpMethod method = HttpMethod.POST;
    URI uri = new URI("http://" + host + ":" + port);
    CloseableHttpClient delegate = mock(CloseableHttpClient.class);
    final CloseableHttpResponse response = mock(CloseableHttpResponse.class);
    StatusLine statusLine = mock(StatusLine.class);
    doReturn(200).when(statusLine).getStatusCode();
    doReturn(statusLine).when(response).getStatusLine();
    doThrow(new IOException("boom")).doThrow(new IOException("boom again")).doReturn(response).when(delegate).execute(any(HttpUriRequest.class));
    ILoadBalancer lb = mock(ILoadBalancer.class);
    MyBackOffPolicy myBackOffPolicy = new MyBackOffPolicy();
    MyRetryListener myRetryListener = new MyRetryListener();
    RetryableRibbonLoadBalancingHttpClient client = setupClientForRetry(retriesNextServer, retriesSameServer, retryable, retryOnAllOps, serviceName, host, port, delegate, lb, "", myBackOffPolicy, false, new RetryListener[] { myRetryListener });
    RibbonApacheHttpRequest request = mock(RibbonApacheHttpRequest.class);
    doReturn(method).when(request).getMethod();
    doReturn(uri).when(request).getURI();
    doReturn(request).when(request).withNewUri(any(URI.class));
    HttpUriRequest uriRequest = mock(HttpUriRequest.class);
    doReturn(uriRequest).when(request).toRequest(any(RequestConfig.class));
    RibbonApacheHttpResponse returnedResponse = client.execute(request, null);
    verify(response, times(0)).close();
    verify(delegate, times(3)).execute(any(HttpUriRequest.class));
    verify(lb, times(2)).chooseServer(eq(serviceName));
    assertEquals(2, myBackOffPolicy.getCount());
    assertEquals(2, myRetryListener.getOnError());
}

Example 18 with ILoadBalancer

use of com.netflix.loadbalancer.ILoadBalancer in project spring-cloud-netflix by spring-cloud.

the class RibbonLoadBalancingHttpClientTests method testRetryOnPost.

@Test
public void testRetryOnPost() throws Exception {
    int retriesNextServer = 1;
    int retriesSameServer = 1;
    boolean retryable = true;
    boolean retryOnAllOps = true;
    String serviceName = "foo";
    String host = serviceName;
    int port = 80;
    HttpMethod method = HttpMethod.POST;
    URI uri = new URI("http://" + host + ":" + port);
    CloseableHttpClient delegate = mock(CloseableHttpClient.class);
    final CloseableHttpResponse response = mock(CloseableHttpResponse.class);
    StatusLine statusLine = mock(StatusLine.class);
    doReturn(200).when(statusLine).getStatusCode();
    doReturn(statusLine).when(response).getStatusLine();
    doThrow(new IOException("boom")).doThrow(new IOException("boom again")).doReturn(response).when(delegate).execute(any(HttpUriRequest.class));
    ILoadBalancer lb = mock(ILoadBalancer.class);
    MyBackOffPolicy myBackOffPolicy = new MyBackOffPolicy();
    RetryableRibbonLoadBalancingHttpClient client = setupClientForRetry(retriesNextServer, retriesSameServer, retryable, retryOnAllOps, serviceName, host, port, delegate, lb, "", myBackOffPolicy);
    RibbonApacheHttpRequest request = mock(RibbonApacheHttpRequest.class);
    doReturn(method).when(request).getMethod();
    doReturn(uri).when(request).getURI();
    doReturn(request).when(request).withNewUri(any(URI.class));
    HttpUriRequest uriRequest = mock(HttpUriRequest.class);
    doReturn(uriRequest).when(request).toRequest(any(RequestConfig.class));
    RibbonApacheHttpResponse returnedResponse = client.execute(request, null);
    verify(response, times(0)).close();
    verify(delegate, times(3)).execute(any(HttpUriRequest.class));
    verify(lb, times(2)).chooseServer(eq(serviceName));
    assertEquals(2, myBackOffPolicy.getCount());
}

Example 19 with ILoadBalancer

use of com.netflix.loadbalancer.ILoadBalancer in project spring-cloud-netflix by spring-cloud.

the class RibbonLoadBalancingHttpClientTests method retryListenerTestNoRetry.

@Test(expected = TerminatedRetryException.class)
public void retryListenerTestNoRetry() throws Exception {
    int retriesNextServer = 1;
    int retriesSameServer = 1;
    boolean retryable = true;
    boolean retryOnAllOps = true;
    String serviceName = "listener";
    String host = serviceName;
    int port = 80;
    HttpMethod method = HttpMethod.POST;
    URI uri = new URI("http://" + host + ":" + port);
    CloseableHttpClient delegate = mock(CloseableHttpClient.class);
    final CloseableHttpResponse response = mock(CloseableHttpResponse.class);
    StatusLine statusLine = mock(StatusLine.class);
    doReturn(200).when(statusLine).getStatusCode();
    doReturn(statusLine).when(response).getStatusLine();
    doThrow(new IOException("boom")).doThrow(new IOException("boom again")).doReturn(response).when(delegate).execute(any(HttpUriRequest.class));
    ILoadBalancer lb = mock(ILoadBalancer.class);
    MyBackOffPolicy myBackOffPolicy = new MyBackOffPolicy();
    MyRetryListenerNotRetry myRetryListenerNotRetry = new MyRetryListenerNotRetry();
    RetryableRibbonLoadBalancingHttpClient client = setupClientForRetry(retriesNextServer, retriesSameServer, retryable, retryOnAllOps, serviceName, host, port, delegate, lb, "", myBackOffPolicy, false, new RetryListener[] { myRetryListenerNotRetry });
    RibbonApacheHttpRequest request = mock(RibbonApacheHttpRequest.class);
    doReturn(method).when(request).getMethod();
    doReturn(uri).when(request).getURI();
    doReturn(request).when(request).withNewUri(any(URI.class));
    HttpUriRequest uriRequest = mock(HttpUriRequest.class);
    doReturn(uriRequest).when(request).toRequest(any(RequestConfig.class));
    RibbonApacheHttpResponse returnedResponse = client.execute(request, null);
}

Example 20 with ILoadBalancer

use of com.netflix.loadbalancer.ILoadBalancer in project spring-cloud-netflix by spring-cloud.

the class RibbonLoadBalancingHttpClientTests method testDoubleEncodingWithRetry.

@Test
public void testDoubleEncodingWithRetry() throws Exception {
    int retriesNextServer = 0;
    int retriesSameServer = 0;
    boolean retryable = true;
    boolean retryOnAllOps = true;
    String serviceName = "foo";
    String host = serviceName;
    int port = 80;
    HttpMethod method = HttpMethod.GET;
    final URI uri = new URI("https://" + host + ":" + port + "/a%20b");
    RibbonCommandContext context = new RibbonCommandContext(serviceName, method.toString(), uri.toString(), true, new LinkedMultiValueMap<String, String>(), new LinkedMultiValueMap<String, String>(), new ByteArrayInputStream(new String("bar").getBytes()), new ArrayList<RibbonRequestCustomizer>());
    RibbonApacheHttpRequest request = new RibbonApacheHttpRequest(context);
    CloseableHttpClient delegate = mock(CloseableHttpClient.class);
    final CloseableHttpResponse response = mock(CloseableHttpResponse.class);
    StatusLine statusLine = mock(StatusLine.class);
    doReturn(200).when(statusLine).getStatusCode();
    doReturn(statusLine).when(response).getStatusLine();
    doReturn(response).when(delegate).execute(any(HttpUriRequest.class));
    ILoadBalancer lb = mock(ILoadBalancer.class);
    RetryableRibbonLoadBalancingHttpClient client = setupClientForRetry(retriesNextServer, retriesSameServer, retryable, retryOnAllOps, serviceName, host, port, delegate, lb, "", null, true);
    client.execute(request, null);
    verify(response, times(0)).close();
    verify(delegate, times(1)).execute(argThat(new ArgumentMatcher<HttpUriRequest>() {

        @Override
        public boolean matches(HttpUriRequest argument) {
            if (argument instanceof HttpUriRequest) {
                HttpUriRequest arg = (HttpUriRequest) argument;
                return arg.getURI().equals(uri);
            }
            return false;
        }
    }));
}