com.qaprosoft.zafira.util.http.HttpClient Maven / Gradle / Ivy

Go to download
/*******************************************************************************
 * Copyright 2013-2019 Qaprosoft (http://www.qaprosoft.com).
 *
 * 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.qaprosoft.zafira.util.http;

import java.util.Map;
import java.util.function.Function;

import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;

import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.qaprosoft.zafira.client.Path;
import com.sun.jersey.api.client.Client;
import com.sun.jersey.api.client.ClientResponse;
import com.sun.jersey.api.client.WebResource;
import com.sun.jersey.api.client.config.DefaultClientConfig;
import com.sun.jersey.core.util.MultivaluedMapImpl;

public class HttpClient {

    private static final Logger LOGGER = LoggerFactory.getLogger(HttpClient.class);

    private static final Integer CONNECT_TIMEOUT = 60000;
    private static final Integer READ_TIMEOUT = 60000;

    private static Client client;

    static {
        client = Client.create(new DefaultClientConfig(GensonProvider.class));
        client.setConnectTimeout(CONNECT_TIMEOUT);
        client.setReadTimeout(READ_TIMEOUT);
    }

    public static Executor uri(Path path, String serviceUrl, Object... parameters) {
        String url = path.build(serviceUrl, parameters);
        return uri(url, null);
    }

    public static Executor uri(Path path, Map queryParameters, String serviceUrl, Object... parameters) {
        String url = path.build(serviceUrl, parameters);
        return uri(url, queryParameters);
    }

    private static Executor uri(String url, Map queryParameters) {
        WebResource webResource = client.resource(url);
        if (queryParameters != null) {
            MultivaluedMap requestParameters = new MultivaluedMapImpl();
            queryParameters.forEach(requestParameters::add);
            webResource = webResource.queryParams(requestParameters);
        }
        return new Executor(webResource);
    }

    public static class Executor {

        private WebResource.Builder builder;
        private String errorMessage;

        public Executor(WebResource webResource) {
            builder = webResource.type(MediaType.APPLICATION_JSON)
                                 .accept(MediaType.APPLICATION_JSON);
        }

        public  Response get(Class responseClass) {
            return execute(responseClass, builder -> builder.get(ClientResponse.class));
        }

        public  Response post(Class responseClass, Object requestEntity) {
            return execute(responseClass, builder -> builder.post(ClientResponse.class, requestEntity));
        }

        public  Response put(Class responseClass, Object requestEntity) {
            return execute(responseClass, builder -> builder.put(ClientResponse.class, requestEntity));
        }

        public  Response delete(Class responseClass) {
            return execute(responseClass, builder -> builder.delete(ClientResponse.class));
        }

        public Executor type(String mediaType) {
            builder.type(mediaType);
            return this;
        }

        public Executor accept(String mediaType) {
            builder.accept(mediaType);
            return this;
        }

        public Executor withAuthorization(String authToken) {
            return withAuthorization(authToken, null);
        }

        public Executor withAuthorization(String authToken, String project) {
            initHeaders(builder, authToken, project);
            return this;
        }

        private static void initHeaders(WebResource.Builder builder, String authToken, String project) {
            if (!StringUtils.isEmpty(authToken)) {
                builder.header("Authorization", authToken);
            }
            if (!StringUtils.isEmpty(project)) {
                builder.header("Project", project);
            }
        }

        private  Response execute(Class responseClass, Function methodBuilder) {
            Response rs = new Response<>();
            try {
                ClientResponse response = methodBuilder.apply(builder);
                int status = response.getStatus();
                rs.setStatus(status);
                boolean deserializeResponseBody = responseClass != null && response.hasEntity() && !responseClass.isAssignableFrom(Void.class) && status == 200;
                if (deserializeResponseBody) {
                    rs.setObject(response.getEntity(responseClass));
                }
            } catch (Exception e) {
                String message = errorMessage == null ? e.getMessage() : e.getMessage() + ". " + errorMessage;
                LOGGER.error(message, e);
            }
            return rs;
        }

        public Executor onFailure(String message) {
            this.errorMessage = message;
            return this;
        }

    }

    public static class Response {

        private int status;
        private T object;

        public Response() {
        }

        Response(int status, T object) {
            this.status = status;
            this.object = object;
        }

        public int getStatus() {
            return status;
        }

        public void setStatus(int status) {
            this.status = status;
        }

        public T getObject() {
            return object;
        }

        public void setObject(T object) {
            this.object = object;
        }
    }

}