cc.kebei.expands.request.http.simple.SimpleResponse Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of onion-expands-request Show documentation

请求模拟工具:http,ftp,webservice

There is a newer version: 3.0.6

Show newest version

package cc.kebei.expands.request.http.simple;


import org.apache.http.Header;
import org.apache.http.HttpResponse;
import org.apache.http.util.EntityUtils;
import cc.kebei.expands.request.http.Response;

import java.io.IOException;
import java.io.InputStream;
import java.util.Optional;

/**
 * Created by Kebei on 16-6-28.
 */
public class SimpleResponse implements Response {
    HttpResponse response;

    public SimpleResponse(HttpResponse response) {
        this.response = response;
    }

    @Override
    public int getCode() {
        return response.getStatusLine().getStatusCode();
    }

    @Override
    public String asString() throws IOException {
        return new String(asBytes());
    }

    @Override
    public byte[] asBytes() throws IOException {
        return EntityUtils.toByteArray(response.getEntity());
    }

    @Override
    public InputStream asStream() throws IOException {
        return response.getEntity().getContent();
    }

    @Override
    public  T getNativeResponse() {
        return (T) response;
    }

    @Override
    public String toString() {
        try {
            return asString();
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    @Override
    public String getHeader(String header) {
        return Optional
                .ofNullable(response.getFirstHeader(header))
                .map(Header::getValue)
                .orElse(null);
    }

    @Override
    public long contentLength() {
        return response.getEntity().getContentLength();
    }
}