cn.zcltd.http.response.HttpResponseResult Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of btg-http Show documentation
btg-http
There is a newer version: 4.0.10
package cn.zcltd.http.response;

import cn.zcltd.btg.sutil.EmptyUtil;
import org.apache.http.HttpResponse;
import org.apache.http.client.methods.HttpUriRequest;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * http请求返回结果
 */
public class HttpResponseResult {
    private int status;
    private HttpUriRequest request;
    private HttpResponse response;
    private String charset = "UTF-8";
    private byte[] responseByte;
    private Map> headerMap = new HashMap<>();

    public HttpResponseResult() {

    }

    public HttpResponseResult(byte[] responseByte) {
        this.responseByte = responseByte;
    }

    /**
     * 添加header
     *
     * @param key   名称
     * @param value 值
     */
    public void addHeader(String key, String value) {
        if (headerMap.containsKey(key)) {
            headerMap.get(key).add(value);
            return;
        }
        headerMap.put(key, new ArrayList() {{
            add(value);
        }});
    }

    /**
     * 添加一组header
     *
     * @param headers 值
     */
    public void addHeaders(Map> headers) {
        headerMap.putAll(headers);
    }

    /**
     * 获取header（第一个值）
     *
     * @param key 名称
     * @return 值
     */
    public String getHeader(String key) {
        List values = headerMap.get(key);
        if (EmptyUtil.isNotEmpty(values)) {
            return values.get(0);
        }
        return null;
    }

    /**
     * 获取header
     *
     * @param key 名称
     * @return 值
     */
    public List getHeaderMult(String key) {
        return headerMap.get(key);
    }

    public InputStream getResponseBody() throws IOException, UnsupportedOperationException {
        return new ByteArrayInputStream(responseByte);
    }

    public int getStatus() {
        return status;
    }

    public void setStatus(int status) {
        this.status = status;
    }

    public HttpUriRequest getRequest() {
        return request;
    }

    public void setRequest(HttpUriRequest request) {
        this.request = request;
    }

    public HttpResponse getResponse() {
        return response;
    }

    public void setResponse(HttpResponse response) {
        this.response = response;
    }

    public String getCharset() {
        return charset;
    }

    public void setCharset(String charset) {
        this.charset = charset;
    }

    public byte[] getResponseByte() {
        return responseByte;
    }

    public void setResponseByte(byte[] responseByte) {
        this.responseByte = responseByte;
    }

    public Map> getHeaderMap() {
        return headerMap;
    }

    public void setHeaderMap(Map> headerMap) {
        this.headerMap = headerMap;
    }

    @Override
    public String toString() {
        return "HttpResponseResult{" +
                "headerMap=" + headerMap +
                "responseByte='" + responseByte.length + '\'' +
                '}';
    }
}