top.jfunc.http.component.jdk.DefaultJdkConnectionFactory Maven / Gradle / Ivy

Go to download
package top.jfunc.http.component.jdk;

import top.jfunc.http.base.*;
import top.jfunc.http.component.AbstractRequesterFactory;
import top.jfunc.http.config.Config;
import top.jfunc.http.request.HttpRequest;
import top.jfunc.http.util.NativeUtil;
import top.jfunc.common.utils.ObjectUtil;

import javax.net.ssl.HostnameVerifier;
import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLSocketFactory;
import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URL;

/**
 * @author xiongshiyan at 2020/1/6 , contact me with email [email protected] or phone 15208384257
 */
public class DefaultJdkConnectionFactory extends AbstractRequesterFactory {
    @Override
    public HttpURLConnection doCreate(HttpRequest httpRequest) throws IOException{
        URL url = new URL(httpRequest.getCompletedUrl());
        //1.1如果需要则设置代理
        Config config = httpRequest.getConfig();
        ProxyInfo proxyInfo = ObjectUtil.defaultIfNull(httpRequest.getProxyInfo(), config.getDefaultProxyInfo());
        HttpURLConnection connection = (null != proxyInfo) ?
                (HttpURLConnection)url.openConnection(proxyInfo.getProxy()) :
                (HttpURLConnection) url.openConnection();

        configConnection(connection , httpRequest);

        return connection;
    }

    protected void configConnection(HttpURLConnection connection , HttpRequest httpRequest) throws IOException{
        ////////////////////////////////////ssl处理///////////////////////////////////
        if(connection instanceof HttpsURLConnection){
            //默认设置这些
            initSSL((HttpsURLConnection)connection , httpRequest);
        }
        ////////////////////////////////////ssl处理///////////////////////////////////

        Method method = httpRequest.getMethod();
        Config config = httpRequest.getConfig();

        boolean followRedirects = ObjectUtil.defaultIfNull(httpRequest.followRedirects() , config.followRedirects());

        if (followRedirects) {
            if(Method.GET == method || Method.POST == method){
                connection.setInstanceFollowRedirects(true);
            }
        }
        else {
            connection.setInstanceFollowRedirects(false);
        }
        ///
        /*connection.setDoInput(true);
        connection.setDoOutput(true);
        connection.setUseCaches(false);*/
        connection.setDoInput(true);
        connection.setDoOutput(config.getMethodContentStrategy().supportContent(method));

        connection.setRequestMethod(method.name());
        connection.setConnectTimeout(config.getConnectionTimeoutWithDefault(httpRequest.getConnectionTimeout()));
        connection.setReadTimeout(config.getReadTimeoutWithDefault(httpRequest.getReadTimeout()));
    }

    protected void initSSL(HttpsURLConnection connection , HttpRequest httpRequest){
        Config config = httpRequest.getConfig();
        HostnameVerifier hostnameVerifier = ObjectUtil.defaultIfNull(httpRequest.getHostnameVerifier(), config.sslHolder().getHostnameVerifier());
        SSLSocketFactory sslSocketFactory = ObjectUtil.defaultIfNull(httpRequest.getSslSocketFactory(), config.sslHolder().getSslSocketFactory());
        NativeUtil.initSSL(connection, hostnameVerifier , sslSocketFactory);
    }
}