com.yhzdys.myosotis.misc.MyosotisHttpClient Maven / Gradle / Ivy

Go to download
package com.yhzdys.myosotis.misc;

import org.apache.http.Header;
import org.apache.http.HeaderIterator;
import org.apache.http.HttpResponse;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.conn.ConnectionKeepAliveStrategy;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.protocol.HTTP;
import org.apache.http.protocol.HttpContext;

import java.io.Closeable;
import java.util.concurrent.TimeUnit;

/**
 * for myosotis internal use only
 * not expose {@link Closeable#close()} method, so current client can't be close(design on purpose)
 */
public final class MyosotisHttpClient {

    private static final MyosotisHttpClient instance = new MyosotisHttpClient(createClient());
    private final CloseableHttpClient httpClient;

    private MyosotisHttpClient(CloseableHttpClient client) {
        this.httpClient = client;
    }

    public static MyosotisHttpClient getInstance() {
        return instance;
    }

    private static CloseableHttpClient createClient() {
        try {
            return HttpClientBuilder.create()
                    .setMaxConnTotal(1024)
                    .setMaxConnPerRoute(1024)
                    .evictIdleConnections(60, TimeUnit.SECONDS)
                    .setKeepAliveStrategy(new MyosotisKeepAliveStrategy())
                    .build();
        } catch (Throwable t) {
            LoggerFactory.getLogger().error(t.getMessage(), t);
        }
        return null;
    }

    public CloseableHttpResponse execute(HttpUriRequest request) throws Exception {
        return httpClient.execute(request);
    }

    private static class MyosotisKeepAliveStrategy implements ConnectionKeepAliveStrategy {

        @Override
        public long getKeepAliveDuration(HttpResponse httpResponse, HttpContext httpContext) {
            HeaderIterator iterator = httpResponse.headerIterator(HTTP.CONN_KEEP_ALIVE);
            while (iterator.hasNext()) {
                Header header = iterator.nextHeader();
                String name = header.getName();
                String value = header.getValue();
                if (value != null && name.equalsIgnoreCase("timeout")) {
                    try {
                        return Long.parseLong(value) * 1000;
                    } catch (Exception ignore) {
                    }
                }
            }
            // 10min.
            return TimeUnit.MINUTES.toMillis(10);
        }
    }
}