com.aliyun.datahub.client.http.HttpResponse Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aliyun-sdk-datahub Show documentation
There is a newer version: 2.25.6
package com.aliyun.datahub.client.http;

import com.aliyun.datahub.client.common.DatahubConstant;
import com.aliyun.datahub.client.model.BaseResult;

import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.Response;
import java.util.HashMap;
import java.util.Map;

public class HttpResponse {
    private int httpStatus;
    private Map headers = new HashMap<>();
    private Response response;

    public HttpResponse(HttpRequest request, Response response) {
        this.response = response;
        this.httpStatus = response.getStatus();

        for (String key : response.getHeaders().keySet()) {
            headers.put(key.toLowerCase(), response.getHeaderString(key));
        }
    }

    public  T getEntity(Class returnType) {
        try {
            if (returnType != null) {
                T result = getEntityInter(returnType);
                if (result == null
                        && BaseResult.class.isAssignableFrom(returnType)) {
                    try {
                        result = returnType.newInstance();
                        String requestId = headers.get(DatahubConstant.X_DATAHUB_REQUEST_ID);
                        ((BaseResult)result).setRequestId(requestId);
                    } catch (InstantiationException | IllegalAccessException e) {
                        //
                    }
                }
                return result;
            }
            return null;
        } finally {
            response.close();
        }
    }

    private  T getEntityInter(Class returnType) {
        if (isMediaSupport()) {
            return response.readEntity(returnType);
        } else {
            String content = response.readEntity(String.class);
            if (String.class.equals(returnType)) {
                return returnType.cast(content);
            } else {
                return null;
            }
        }
    }

    private boolean isMediaSupport() {
        String mediaType = headers.get(HttpHeaders.CONTENT_TYPE.toLowerCase());
        return mediaType != null && (
                mediaType.contains("application/json") || mediaType.contains("application/x-protobuf"));
    }

    public int getHttpStatus() {
        return httpStatus;
    }

    public void close() {
        this.response.close();
    }

    public Map getHeaders() {
        return headers;
    }

    public String getHeader(String key) {
        return headers.get(key);
    }
}