okhttp3.HttpClientBuilder Maven / Gradle / Ivy

Go to download
/*
 * Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements.
 * See the NOTICE file distributed with this work for additional information regarding copyright ownership.
 * The ASF licenses this file to you under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is
 * distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and limitations under the License.
 *
 * =========================================================================================================
 *
 * This software consists of voluntary contributions made by many individuals on behalf of the
 * Apache Software Foundation. For more information on the Apache Software Foundation, please see
 * .
 *
 * +-------------------------------------------------------------------------------------------------------+
 * | License: http://www.apache.org/licenses/LICENSE-2.0.txt 										       |
 * | Author: Yong.Teng  													       |
 * | Copyright @ 2013-2024 Buession.com Inc.														       |
 * +-------------------------------------------------------------------------------------------------------+
 */
package okhttp3;

import javax.net.ssl.HostnameVerifier;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocketFactory;
import java.util.Optional;
import java.util.concurrent.TimeUnit;

/**
 * @author Yong.Teng
 * @since 2.3.0
 */
public class HttpClientBuilder {

	private final OkHttpClient.Builder builder = new OkHttpClient.Builder();

	private HttpClientConnectionManager connectionManager;

	protected HttpClientBuilder() {
	}

	public static HttpClientBuilder create() {
		return new HttpClientBuilder();
	}

	public HttpClientBuilder setRetryOnConnectionFailure(Boolean retryOnConnectionFailure) {
		if(retryOnConnectionFailure != null){
			builder.retryOnConnectionFailure(retryOnConnectionFailure);
		}

		return this;
	}

	public HttpClientBuilder setConnectTimeout(long connectTimeout) {
		if(connectTimeout > -1){
			builder.connectTimeout(connectTimeout, TimeUnit.MILLISECONDS);
		}

		return this;
	}

	public HttpClientBuilder setReadTimeout(long readTimeout) {
		if(readTimeout > 0){
			builder.readTimeout(readTimeout, TimeUnit.MILLISECONDS);
		}

		return this;
	}

	public HttpClientBuilder setWriteTimeout(long writeTimeout) {
		if(writeTimeout > 0){
			builder.writeTimeout(writeTimeout, TimeUnit.MILLISECONDS);
		}

		return this;
	}

	public HttpClientBuilder setFollowRedirects(Boolean followRedirects) {
		if(followRedirects != null){
			builder.followRedirects(followRedirects);
		}

		return this;
	}

	public HttpClientBuilder setSSLSocketFactory(SSLSocketFactory sslSocketFactory) {
		if(sslSocketFactory != null){
			builder.sslSocketFactory(sslSocketFactory);
		}

		return this;
	}

	public HttpClientBuilder setSSLHostnameVerifier(HostnameVerifier hostnameVerifier) {
		if(hostnameVerifier != null){
			builder.hostnameVerifier(hostnameVerifier);
		}

		return this;
	}

	public HttpClientBuilder setSSLContext(SSLContext sslContext) {
		if(sslContext != null){
		}

		return this;
	}

	public HttpClientBuilder setConnectionManager(HttpClientConnectionManager connectionManager) {
		this.connectionManager = connectionManager;
		return this;
	}

	public OkHttpClient build() {
		Optional.ofNullable(connectionManager)
				.ifPresent((connectionManager)->builder.connectionPool(connectionManager.getConnectionPool()));

		OkHttpClient client = builder.build();

		if(connectionManager != null){
			if(connectionManager.getMaxRequests() > 0){
				client.dispatcher().setMaxRequests(connectionManager.getMaxRequests());
			}
			client.dispatcher().setMaxRequestsPerHost(connectionManager.getMaxRequestsPerHost());
		}

		return client;
	}

}