com.yuweix.kuafu.http.request.AbstractHttpRequest Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of kuafu-http Show documentation
Kuafu
The newest version!
package com.yuweix.kuafu.http.request;


import java.util.List;
import jakarta.servlet.http.Cookie;

import com.alibaba.fastjson2.TypeReference;
import com.yuweix.kuafu.http.CallbackResponseHandler;
import com.yuweix.kuafu.http.DefaultHttpDelete;
import com.yuweix.kuafu.http.HttpContextAdaptor;
import com.yuweix.kuafu.http.HttpMethod;
import com.yuweix.kuafu.http.response.ErrorHttpResponse;
import com.yuweix.kuafu.http.strategy.connect.KeepAliveStrategy;
import com.yuweix.kuafu.http.strategy.redirect.NeedRedirectStrategy;
import com.yuweix.kuafu.http.strategy.retry.NotNeedRetryHandler;
import org.apache.http.Header;
import org.apache.http.HttpRequestInterceptor;
import org.apache.http.HttpResponseInterceptor;
import org.apache.http.HttpStatus;
import org.apache.http.client.HttpRequestRetryHandler;
import org.apache.http.client.RedirectStrategy;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.methods.HttpEntityEnclosingRequestBase;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpPut;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.conn.ConnectionKeepAliveStrategy;
import org.apache.http.conn.socket.LayeredConnectionSocketFactory;
import org.apache.http.entity.ContentType;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.message.BasicHeader;
import org.apache.http.protocol.HTTP;
import com.yuweix.kuafu.http.ssl.TrustAllSslSocketFactory;
import com.yuweix.kuafu.http.response.HttpResponse;


/**
 * @author yuwei
 */
public abstract class AbstractHttpRequest> implements HttpRequest {
	private HttpUriRequest httpUriRequest;
	private String url;
	private HttpMethod method;
	private Class responseTypeClass;
	private TypeReference responseTypeReference;
	private List cookieList;
	private List headerList;
	private LayeredConnectionSocketFactory sslSocketFactory;
	private ConnectionKeepAliveStrategy keepAliveStrategy;
	private RequestConfig requestConfig;
	private HttpRequestRetryHandler retryHandler;
	private RedirectStrategy redirectStrategy;
	private List firstRequestInterceptorList;
	private List lastRequestInterceptorList;
	private List firstResponseInterceptorList;
	private List lastResponseInterceptorList;
	private String charset;


	protected AbstractHttpRequest() {
		this.responseTypeClass = String.class;
	}

	protected void setHttpUriRequest(HttpUriRequest httpUriRequest) {
		this.httpUriRequest = httpUriRequest;
	}


	@SuppressWarnings("unchecked")
	public T url(String url) {
		this.url = url;
		return (T) this;
	}
	public String getUrl() {
		return url;
	}

	@SuppressWarnings("unchecked")
	public T method(HttpMethod method) {
		this.method = method;
		return (T) this;
	}
	public HttpMethod getMethod() {
		return method;
	}

	@SuppressWarnings("unchecked")
	public T responseType(Class responseTypeClass) {
		this.responseTypeClass = responseTypeClass;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T responseType(TypeReference typeReference) {
		this.responseTypeReference = typeReference;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T cookieList(List cookieList) {
		this.cookieList = cookieList;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T headerList(List headerList) {
		this.headerList = headerList;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T sslSocketFactory(LayeredConnectionSocketFactory sslSocketFactory) {
		this.sslSocketFactory = sslSocketFactory;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T keepAliveStrategy(ConnectionKeepAliveStrategy keepAliveStrategy) {
		this.keepAliveStrategy = keepAliveStrategy;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T requestConfig(RequestConfig requestConfig) {
		this.requestConfig = requestConfig;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T retryHandler(HttpRequestRetryHandler retryHandler) {
		this.retryHandler = retryHandler;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T redirectStrategy(RedirectStrategy redirectStrategy) {
		this.redirectStrategy = redirectStrategy;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T firstRequestInterceptorList(List firstRequestInterceptorList) {
		this.firstRequestInterceptorList = firstRequestInterceptorList;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T lastRequestInterceptorList(List lastRequestInterceptorList) {
		this.lastRequestInterceptorList = lastRequestInterceptorList;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T firstResponseInterceptorList(List firstResponseInterceptorList) {
		this.firstResponseInterceptorList = firstResponseInterceptorList;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T lastResponseInterceptorList(List lastResponseInterceptorList) {
		this.lastResponseInterceptorList = lastResponseInterceptorList;
		return (T) this;
	}

	@SuppressWarnings("unchecked")
	public T charset(String charset) {
		this.charset = charset;
		return (T) this;
	}
	public String getCharset() {
		return charset;
	}

	protected ContentType getHeaderContentType() {
		return null;
	}

	protected HttpEntityEnclosingRequestBase getRequestBase() {
		if (HttpMethod.POST.equals(method)) {
			return new HttpPost(url);
		} else if (HttpMethod.PUT.equals(method)) {
			return new HttpPut(url);
		} else if (HttpMethod.DELETE.equals(method)) {
			return new DefaultHttpDelete(url);
		} else {
			return new HttpPost(url);
		}
	}

	protected HttpResponse execute0() {
		/**
		 * header
		 */
		ContentType hct = getHeaderContentType();
		if (hct != null) {
			httpUriRequest.setHeader(new BasicHeader(HTTP.CONTENT_TYPE, hct.toString()));
		}
		if (headerList != null && headerList.size() > 0) {
			for (Header header: headerList) {
				httpUriRequest.setHeader(header);
			}
		}
		/**
		 * cookie
		 */
		if (cookieList != null && cookieList.size() > 0) {
			StringBuilder builder = new StringBuilder("");
			for (Cookie cookie: cookieList) {
				builder.append(cookie.getName()).append("=").append(cookie.getValue()).append(";");
			}
			builder.deleteCharAt(builder.length() - 1);
			httpUriRequest.setHeader("Cookie", builder.toString());
		}
		/**
		 * context
		 */
		HttpContextAdaptor context = HttpContextAdaptor.create();

		HttpClientBuilder builder = HttpClients.custom()
											.setDefaultRequestConfig(requestConfig)
											.setKeepAliveStrategy(keepAliveStrategy == null ? KeepAliveStrategy.get() : keepAliveStrategy)
											.setDefaultCookieStore(context.getCookieStore())
											.setRetryHandler(retryHandler == null ? NotNeedRetryHandler.get() : retryHandler)
											.setRedirectStrategy(redirectStrategy == null ? NeedRedirectStrategy.get() : redirectStrategy)
											.setSSLSocketFactory(sslSocketFactory == null ? TrustAllSslSocketFactory.get() : sslSocketFactory);

		/**
		 * add first http request interceptor list
		 */
		if (firstRequestInterceptorList != null && firstRequestInterceptorList.size() > 0) {
			for (HttpRequestInterceptor interceptor: firstRequestInterceptorList) {
				builder.addInterceptorFirst(interceptor);
			}
		}

		/**
		 * add last http request interceptor list
		 */
		if (lastRequestInterceptorList != null && lastRequestInterceptorList.size() > 0) {
			for (HttpRequestInterceptor interceptor: lastRequestInterceptorList) {
				builder.addInterceptorLast(interceptor);
			}
		}

		/**
		 * add first http response interceptor list
		 */
		if (firstResponseInterceptorList != null && firstResponseInterceptorList.size() > 0) {
			for (HttpResponseInterceptor interceptor: firstResponseInterceptorList) {
				builder.addInterceptorFirst(interceptor);
			}
		}

		/**
		 * add last http response interceptor list
		 */
		if (lastResponseInterceptorList != null && lastResponseInterceptorList.size() > 0) {
			for (HttpResponseInterceptor interceptor: lastResponseInterceptorList) {
				builder.addInterceptorLast(interceptor);
			}
		}

		CloseableHttpClient client = builder.build();
		CallbackResponseHandler handler = CallbackResponseHandler.create()
																.responseType(responseTypeClass)
																.responseType(responseTypeReference)
																.context(context)
																.charset(charset);
		try {
			return client.execute(httpUriRequest, handler, context);
		} catch (Exception e) {
			return new ErrorHttpResponse<>(HttpStatus.SC_INTERNAL_SERVER_ERROR, e.toString());
		}
	}
}