Example usage for org.apache.http.config SocketConfig custom

List of usage examples for org.apache.http.config SocketConfig custom

Introduction

In this page you can find the example usage for org.apache.http.config SocketConfig custom.

Prototype

public static Builder custom() 

Source Link

Usage

From source file:org.apache.http.impl.conn.TestPoolingHttpClientConnectionManager.java

@Test
public void testProxyConnectAndUpgrade() throws Exception {
    final HttpHost target = new HttpHost("somehost", -1, "https");
    final HttpHost proxy = new HttpHost("someproxy", 8080);
    final InetAddress remote = InetAddress.getByAddress(new byte[] { 10, 0, 0, 1 });
    final InetAddress local = InetAddress.getByAddress(new byte[] { 127, 0, 0, 1 });
    final HttpRoute route = new HttpRoute(target, local, proxy, true);

    final CPoolEntry entry = new CPoolEntry(LogFactory.getLog(getClass()), "id", route, conn, -1,
            TimeUnit.MILLISECONDS);
    entry.markRouteComplete();/*  www  . jav a  2s .c  o  m*/
    Mockito.when(future.isCancelled()).thenReturn(Boolean.FALSE);
    Mockito.when(conn.isOpen()).thenReturn(true);
    Mockito.when(future.isCancelled()).thenReturn(false);
    Mockito.when(future.get(1, TimeUnit.SECONDS)).thenReturn(entry);
    Mockito.when(pool.lease(route, null, null)).thenReturn(future);

    final ConnectionRequest connRequest1 = mgr.requestConnection(route, null);
    final HttpClientConnection conn1 = connRequest1.get(1, TimeUnit.SECONDS);
    Assert.assertNotNull(conn1);

    final ConnectionSocketFactory plainsf = Mockito.mock(ConnectionSocketFactory.class);
    final LayeredConnectionSocketFactory sslsf = Mockito.mock(LayeredConnectionSocketFactory.class);
    final Socket socket = Mockito.mock(Socket.class);
    final HttpClientContext context = HttpClientContext.create();
    final SocketConfig sconfig = SocketConfig.custom().build();
    final ConnectionConfig cconfig = ConnectionConfig.custom().build();

    mgr.setDefaultSocketConfig(sconfig);
    mgr.setDefaultConnectionConfig(cconfig);

    Mockito.when(dnsResolver.resolve("someproxy")).thenReturn(new InetAddress[] { remote });
    Mockito.when(schemePortResolver.resolve(proxy)).thenReturn(8080);
    Mockito.when(schemePortResolver.resolve(target)).thenReturn(8443);
    Mockito.when(socketFactoryRegistry.lookup("http")).thenReturn(plainsf);
    Mockito.when(socketFactoryRegistry.lookup("https")).thenReturn(sslsf);
    Mockito.when(plainsf.createSocket(Mockito.<HttpContext>any())).thenReturn(socket);
    Mockito.when(plainsf.connectSocket(Mockito.anyInt(), Mockito.eq(socket), Mockito.<HttpHost>any(),
            Mockito.<InetSocketAddress>any(), Mockito.<InetSocketAddress>any(), Mockito.<HttpContext>any()))
            .thenReturn(socket);

    mgr.connect(conn1, route, 123, context);

    Mockito.verify(dnsResolver, Mockito.times(1)).resolve("someproxy");
    Mockito.verify(schemePortResolver, Mockito.times(1)).resolve(proxy);
    Mockito.verify(plainsf, Mockito.times(1)).createSocket(context);
    Mockito.verify(plainsf, Mockito.times(1)).connectSocket(123, socket, proxy,
            new InetSocketAddress(remote, 8080), new InetSocketAddress(local, 0), context);

    Mockito.when(conn.getSocket()).thenReturn(socket);

    mgr.upgrade(conn1, route, context);

    Mockito.verify(schemePortResolver, Mockito.times(1)).resolve(target);
    Mockito.verify(sslsf, Mockito.times(1)).createLayeredSocket(socket, "somehost", 8443, context);

    mgr.routeComplete(conn1, route, context);
}

From source file:org.apache.http.testserver.HttpServer.java

public void start() throws IOException {
    Asserts.check(this.server == null, "Server already running");
    this.server = ServerBootstrap.bootstrap()
            .setSocketConfig(SocketConfig.custom().setSoTimeout(this.timeout).build())
            .setServerInfo("TEST-SERVER/1.1").setConnectionFactory(new LoggingConnFactory())
            .setExceptionLogger(new SimpleExceptionLogger()).setExpectationVerifier(this.expectationVerifier)
            .setHandlerMapper(this.reqistry).create();
    this.server.start();
}

From source file:org.apache.manifoldcf.agents.output.elasticsearch.ElasticSearchConnector.java

protected HttpClient getSession() throws ManifoldCFException {
    if (client == null) {
        connectionManager = new PoolingHttpClientConnectionManager();

        int socketTimeout = 900000;
        int connectionTimeout = 60000;

        // Set up connection manager
        connectionManager = new PoolingHttpClientConnectionManager();

        CredentialsProvider credentialsProvider = new BasicCredentialsProvider();

        RequestConfig.Builder requestBuilder = RequestConfig.custom().setCircularRedirectsAllowed(true)
                .setSocketTimeout(socketTimeout).setStaleConnectionCheckEnabled(true)
                .setExpectContinueEnabled(true).setConnectTimeout(connectionTimeout)
                .setConnectionRequestTimeout(socketTimeout);

        client = HttpClients.custom().setConnectionManager(connectionManager).setMaxConnTotal(1)
                .disableAutomaticRetries().setDefaultRequestConfig(requestBuilder.build())
                .setDefaultSocketConfig(SocketConfig.custom()
                        //.setTcpNoDelay(true)
                        .setSoTimeout(socketTimeout).build())
                .setDefaultCredentialsProvider(credentialsProvider)
                .setRequestExecutor(new HttpRequestExecutor(socketTimeout)).build();

    }/*from ww w .  j a v  a 2 s. co m*/
    expirationTime = System.currentTimeMillis() + EXPIRATION_INTERVAL;
    return client;
}

From source file:org.jboss.javadoc.PostFile.java

public StatusLine post() throws IOException {

    mpEntityBuilder.addBinaryBody("--upload-file", this.filePath);

    String encoding = new String(Base64.encodeBase64((username + ":" + password).getBytes()));
    requestBuilderPost.addHeader("Authorization", "Basic " + encoding);
    requestBuilderPost.addHeader("Accept", "*/*");
    requestBuilderPost.setVersion(HttpVersion.HTTP_1_1);
    requestBuilderPost.setEntity(mpEntityBuilder.build());

    SocketConfig socketConfig = SocketConfig.custom().setTcpNoDelay(true).setSoTimeout(10000).build();
    CloseableHttpClient closeableHttpClient = HttpClientBuilder.create().setDefaultSocketConfig(socketConfig)
            .build();//from   ww w.  j  a  va2s .  c om
    CloseableHttpResponse response = closeableHttpClient.execute(requestBuilderPost.build());

    HttpEntity resEntity = response.getEntity();
    if (resEntity != null) {
        EntityUtils.consume(resEntity);
    }
    return response.getStatusLine();
}

From source file:org.xwiki.extension.repository.http.internal.DefaultHttpClientFactory.java

@Override
public HttpClientBuilder createHttpClientBuilder(Map<String, String> properties) {
    HttpClientBuilder httpClientBuilder = createHttpClientBuilder(properties.get("auth.user"),
            properties.get("auth.password"));

    // Set socket timeouts
    BasicHttpClientConnectionManager connectionManager = new BasicHttpClientConnectionManager();
    SocketConfig.Builder socketConfigBuilder = SocketConfig.custom();
    socketConfigBuilder.setSoTimeout(getIntProperty(properties, SOCKET_TIMEOUT, 30000));
    connectionManager.setSocketConfig(socketConfigBuilder.build());
    httpClientBuilder.setConnectionManager(connectionManager);

    // Set request timeouts
    RequestConfig.Builder requestBuilder = RequestConfig.custom();
    requestBuilder = requestBuilder.setConnectTimeout(getIntProperty(properties, CONNECTION_TIMEOUT, 30000));
    httpClientBuilder.setDefaultRequestConfig(requestBuilder.build());

    return httpClientBuilder;
}