okhttp3.nio.HttpAsyncClientBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of buession-httpclient Show documentation
Buession Framework Http Client
The newest version!
/*
 * 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-2023 Buession.com Inc.														       |
 * +-------------------------------------------------------------------------------------------------------+
 */
package okhttp3.nio;

import okhttp3.OkHttpClient;

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

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

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

	private NioHttpClientConnectionManager connectionManager;

	protected HttpAsyncClientBuilder() {
	}

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

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

		return this;
	}

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

		return this;
	}

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

		return this;
	}

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

		return this;
	}

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

		return this;
	}

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

		return this;
	}

	public HttpAsyncClientBuilder setSSLContext(SSLContext sslContext) {
		return this;
	}

	public HttpAsyncClientBuilder setConnectionManager(NioHttpClientConnectionManager connectionManager) {
		this.connectionManager = connectionManager;
		return this;
	}

	public OkHttpClient build() {
		if(connectionManager != null){
			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;
	}

}