com.budjb.httprequests.jersey1.JerseyHttpResponse Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2016-2018 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.budjb.httprequests.jersey1;

import com.budjb.httprequests.HttpEntity;
import com.budjb.httprequests.HttpRequest;
import com.budjb.httprequests.HttpResponse;
import com.budjb.httprequests.MultiValuedMap;
import com.budjb.httprequests.converter.EntityConverterManager;
import com.sun.jersey.api.client.ClientResponse;

import javax.ws.rs.core.MediaType;
import java.io.IOException;

/**
 * An {@link HttpResponse} implementation that wraps a {@link ClientResponse}.
 */
class JerseyHttpResponse extends HttpResponse {
    /**
     * Jersey Client response.
     */
    private final ClientResponse response;

    /**
     * Constructor.
     *
     * @param request          Request properties used to make the request.
     * @param converterManager Converter manager.
     * @param response         Jersey Client response.
     */
    JerseyHttpResponse(HttpRequest request, EntityConverterManager converterManager, ClientResponse response) throws IOException {
        super(converterManager, request, response.getStatus(), new MultiValuedMap(response.getHeaders()), parseEntity(response));

        this.response = response;

        if (!hasEntity()) {
            close();
        }
    }

    /**
     * Parses the entity from the response.
     *
     * @param response Jersey response.
     * @return The parsed HTTP entity, or {@code null} if none is available.
     * @throws IOException When an IO exception occurs.
     */
    private static HttpEntity parseEntity(ClientResponse response) throws IOException {
        if (!response.hasEntity()) {
            return null;
        }

        String contentType = null;
        String charSet = null;

        if (response.getType() != null) {
            MediaType mediaType = response.getType();
            contentType = mediaType.getType() + "/" + mediaType.getSubtype();
            if (mediaType.getParameters().containsKey("charset")) {
                charSet = mediaType.getParameters().get("charset");
            }
        }

        return new HttpEntity(response.getEntityInputStream(), contentType, charSet);
    }

    /**
     * Returns the Jersey Client response.
     *
     * @return The Jersey Client response.
     */
    public ClientResponse getResponse() {
        return response;
    }
}