play.libs.ws.HttpAsyncResponse Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of framework Show documentation

RePlay is a fork of the Play1 framework, created by Codeborne.

There is a newer version: 2.6.3

package play.libs.ws;

import com.ning.http.client.Response;
import play.mvc.Http;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * An HTTP response wrapper
 */
public class HttpAsyncResponse extends HttpResponse {

    private final Response response;

    /**
     * You shouldn't have to create an HttpResponse yourself
     *
     * @param response
     *            The given response
     */
    public HttpAsyncResponse(Response response) {
        this.response = response;
    }

    /**
     * The HTTP status code
     *
     * @return the status code of the http response
     */
    @Override
    public Integer getStatus() {
        return this.response.getStatusCode();
    }

    /**
     * the HTTP status text
     *
     * @return the status text of the http response
     */
    @Override
    public String getStatusText() {
        return this.response.getStatusText();
    }

    @Override
    public String getHeader(String key) {
        return response.getHeader(key);
    }

    @Override
    public List getHeaders() {
        return ((Map>) response.getHeaders()).entrySet().stream()
          .map(entry -> new Http.Header(entry.getKey(), entry.getValue()))
          .collect(Collectors.toList());
    }

    @Override
    public String getString() {
        try {
            return response.getResponseBody(getEncoding().name());
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    @Override
    public String getString(Charset encoding) {
        try {
            return response.getResponseBody(encoding.name());
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * get the response as a stream
     *
     * @return an inputstream
     */
    @Override
    public InputStream getStream() {
        try {
            return response.getResponseBodyAsStream();
        } catch (IllegalStateException e) {
            return new ByteArrayInputStream(new byte[] {}); // Workaround
                                                            // AHC's bug on
                                                            // empty
                                                            // responses
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

}