com.buession.httpclient.apache.ApacheResponseBuilder 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 com.buession.httpclient.apache;

import com.buession.httpclient.core.AbstractResponseHeaderParse;
import com.buession.httpclient.core.ProtocolVersion;
import com.buession.httpclient.core.AbstractResponseBuilder;
import com.buession.httpclient.core.Response;
import com.buession.httpclient.core.StatusLine;
import com.google.common.collect.Multimap;
import org.apache.http.Header;
import org.apache.http.HttpEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.entity.BufferedHttpEntity;
import org.apache.http.util.EntityUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;

/**
 * Apache Http Client 响应构建器
 *
 * @author Yong.Teng
 */
public class ApacheResponseBuilder extends AbstractResponseBuilder {

	private final static Logger logger = LoggerFactory.getLogger(ApacheResponseBuilder.class);

	@Override
	public Response build(org.apache.http.HttpResponse httpResponse){
		final Response response = new Response();
		final org.apache.http.StatusLine responseStatusLine = httpResponse.getStatusLine();
		final org.apache.http.ProtocolVersion responseProtocolVersion = responseStatusLine.getProtocolVersion();
		final ApacheResponseHeaderParse responseHeaderParse = new ApacheResponseHeaderParse();

		response.setProtocolVersion(ProtocolVersion.createInstance(responseProtocolVersion.getProtocol(),
				responseProtocolVersion.getMajor(), responseProtocolVersion.getMinor()));
		response.setStatusLine(
				new StatusLine(responseStatusLine.getStatusCode(), responseStatusLine.getReasonPhrase()));
		response.setHeaders(responseHeaderParse.parse(httpResponse.getAllHeaders()));

		if(httpResponse.getEntity() != null){
			response.setContentLength(httpResponse.getEntity().getContentLength());

			try{
				HttpEntity httpEntity = new BufferedHttpEntity(httpResponse.getEntity());

				response.setInputStream(httpEntity.getContent());
				response.setBody(EntityUtils.toString(httpEntity));

				if(httpResponse.getEntity().getContent() != null){
					httpResponse.getEntity().getContent().close();
				}
			}catch(IOException e){
				logger.error("Response entity to body error.", e);
			}
		}

		if(httpResponse instanceof CloseableHttpResponse){
			try{
				((CloseableHttpResponse) httpResponse).close();
			}catch(IOException e){
				logger.error("Close HTTP Response error.", e);
			}
		}

		return response;
	}

	/**
	 * Apache HttpClient 响应头解析器
	 *
	 * @author Yong.Teng
	 * @since 2.1.2
	 */
	private static class ApacheResponseHeaderParse extends AbstractResponseHeaderParse {

		@Override
		protected void doParse(final org.apache.http.Header[] headers, final Multimap headersMap){
			for(org.apache.http.Header header : headers){
				if(header.getElements() != null){
					headersMap.put(header.getName(), header.getValue());
				}
			}
		}

	}

}