com.base4j.util.http.HttpResponse Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of base4j-util Show documentation
base4j公共工具包
The newest version!
package com.base4j.util.http;

import java.io.ByteArrayInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.util.List;
import java.util.Map.Entry;
import java.util.zip.GZIPInputStream;

import com.base4j.util.IoUtil;
import com.base4j.util.StrUtil;
import com.base4j.util.exceptions.HttpException;
import com.base4j.util.io.FastByteArrayOutputStream;
import com.base4j.util.lang.Conver;

/**
 * Http响应类
 *
 * @author Looly
 */
public class HttpResponse extends HttpBase {

    /**
     * 读取服务器返回的流保存至内存
     */
    private FastByteArrayOutputStream out;

    /**
     * 读取响应信息
     *
     * @param httpConnection Http连接对象
     * @return HttpResponse
     */
    public static HttpResponse readResponse(HttpConnection httpConnection) {
        final HttpResponse httpResponse = new HttpResponse();

        try {
            httpResponse.status = httpConnection.responseCode();
            httpResponse.headers = httpConnection.headers();
            httpResponse.charset = httpConnection.charset();

            InputStream in;
            if (httpResponse.status < HttpURLConnection.HTTP_BAD_REQUEST) {
                in = httpConnection.getInputStream();
            } else {
                in = httpConnection.getErrorStream();
            }
            httpResponse.readBody(in);
        } catch (IOException e) {
            if (e instanceof FileNotFoundException) {
                //服务器无返回内容，忽略之
            } else {
                throw new HttpException(e.getMessage(), e);
            }
        }

        return httpResponse;
    }

    /**
     * 响应状态码
     */
    private int status;

    public HttpResponse() {
    }

    /**
     * 获取状态码
     *
     * @return 状态码
     */
    public int getStatus() {
        return status;
    }

    // ---------------------------------------------------------------- Http Response Header start

    /**
     * 获取内容编码
     *
     * @return String
     */
    public String contentEncoding() {
        return header(Header.CONTENT_ENCODING);
    }

    /**
     * @return 是否为gzip压缩过的内容
     */
    public boolean isGzip() {
        String contentEncoding = contentEncoding();
        return contentEncoding != null && contentEncoding.equalsIgnoreCase("gzip");
    }
    // ---------------------------------------------------------------- Http Response Header end

    // ---------------------------------------------------------------- Body start

    /**
     * 获得服务区响应流
     *
     * @return 响应流
     */
    public InputStream bodyStream() {
        return new ByteArrayInputStream(this.out.toByteArray());
    }

    /**
     * 获取响应流字节码
     *
     * @return byte[]
     */
    public byte[] bodyBytes() {
        if (null == this.out) {
            return null;
        }
        return this.out.toByteArray();
    }

    /**
     * 获取响应主体
     *
     * @return String
     */
    public String body() {
        try {
            return HttpUtil.getString(bodyStream(), this.charset, null == this.charset);
        } catch (IOException e) {
            throw new HttpException(e);
        }
    }
    // ---------------------------------------------------------------- Body end

    @Override
    public String toString() {
        StringBuilder sb = StrUtil.builder();
        sb.append("Request Headers: ").append(StrUtil.CRLF);
        for (Entry> entry : this.headers.entrySet()) {
            sb.append("    ").append(entry).append(StrUtil.CRLF);
        }

        sb.append("Request Body: ").append(StrUtil.CRLF);
        sb.append("    ").append(this.body()).append(StrUtil.CRLF);

        return sb.toString();
    }

    // ---------------------------------------------------------------- Private method start

    /**
     * 读取主体
     *
     * @param in 输入流
     * @return 自身
     * @throws IOException
     */
    private void readBody(InputStream in) throws IOException {
        if (isGzip()) {
            in = new GZIPInputStream(in);
        }
        int contentLength = Conver.toInt(header(Header.CONTENT_LENGTH), 0);
        this.out = contentLength > 0 ? new FastByteArrayOutputStream(contentLength) : new FastByteArrayOutputStream();
        IoUtil.copy(in, this.out);
    }
    // ---------------------------------------------------------------- Private method end
}