org.asynchttpclient.ResponseBase Maven / Gradle / Ivy

Go to download
package org.asynchttpclient;

import static org.asynchttpclient.util.HttpUtils.*;
import static org.asynchttpclient.util.MiscUtils.isNonEmpty;
import io.netty.handler.codec.http.HttpHeaders;

import java.net.SocketAddress;
import java.nio.charset.Charset;
import java.util.Collections;
import java.util.List;
import java.util.Map;

import org.asynchttpclient.cookie.Cookie;
import org.asynchttpclient.uri.Uri;

public abstract class ResponseBase implements Response {

    protected final List bodyParts;
    protected final HttpResponseHeaders headers;
    protected final HttpResponseStatus status;
    private List cookies;

    protected ResponseBase(HttpResponseStatus status, HttpResponseHeaders headers, List bodyParts) {
        this.bodyParts = bodyParts;
        this.headers = headers;
        this.status = status;
    }

    protected abstract List buildCookies();

    protected Charset calculateCharset(Charset charset) {

        if (charset == null) {
            String contentType = getContentType();
            if (contentType != null)
                charset = parseCharset(contentType); // parseCharset can return
                                                     // null
        }
        return charset != null ? charset : DEFAULT_CHARSET;
    }

    @Override
    public final int getStatusCode() {
        return status.getStatusCode();
    }

    @Override
    public final String getStatusText() {
        return status.getStatusText();
    }

    @Override
    public final Uri getUri() {
        return status.getUri();
    }

    @Override
    public SocketAddress getRemoteAddress() {
        return status.getRemoteAddress();
    }

    @Override
    public SocketAddress getLocalAddress() {
        return status.getLocalAddress();
    }

    @Override
    public final String getContentType() {
        return headers != null ? getHeader("Content-Type") : null;
    }

    @Override
    public final String getHeader(String name) {
        return headers != null ? getHeaders().get(name) : null;
    }

    @Override
    public final List getHeaders(String name) {
        return headers != null ? getHeaders().getAll(name) : Collections. emptyList();
    }

    @Override
    public final HttpHeaders getHeaders() {
        return headers != null ? headers.getHeaders() : HttpHeaders.EMPTY_HEADERS;
    }

    @Override
    public final boolean isRedirected() {
        switch (status.getStatusCode()) {
        case 301:
        case 302:
        case 303:
        case 307:
        case 308:
            return true;
        default:
            return false;
        }
    }

    @Override
    public List getCookies() {

        if (headers == null) {
            return Collections.emptyList();
        }

        if (cookies == null) {
            cookies = buildCookies();
        }
        return cookies;

    }

    @Override
    public boolean hasResponseStatus() {
        return status != null;
    }

    @Override
    public boolean hasResponseHeaders() {
        return headers != null && !headers.getHeaders().isEmpty();
    }

    @Override
    public boolean hasResponseBody() {
        return isNonEmpty(bodyParts);
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        sb.append(getClass().getSimpleName()).append(" {\n")//
        .append("\tstatusCode=").append(getStatusCode()).append("\n")//
        .append("\theaders=\n");
        
        for (Map.Entry header: getHeaders()) {
            sb.append("\t\t").append(header.getKey()).append(": ").append(header.getValue()).append("\n");
        }
        sb.append("\tbody=\n").append(getResponseBody()).append("\n")//
                .append("}").toString();
        return sb.toString();
    }
}