com.nepxion.discovery.common.util.RestUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of discovery-common Show documentation
Nepxion Discovery is an enhancement for Spring Cloud Discovery
There is a newer version: 6.22.0
package com.nepxion.discovery.common.util;

/**
 * Title: Nepxion Discovery
 * Description: Nepxion Discovery
 * Copyright: Copyright (c) 2017-2050
 * Company: Nepxion
 * @author Haojun Ren
 * @version 1.0
 */

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.apache.commons.collections4.MapUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.http.HttpHeaders;
import org.springframework.web.client.ResponseErrorHandler;
import org.springframework.web.client.RestTemplate;

import com.fasterxml.jackson.core.type.TypeReference;
import com.nepxion.discovery.common.constant.DiscoveryConstant;
import com.nepxion.discovery.common.handler.DiscoveryResponseErrorHandler;

public class RestUtil {
    public static HttpHeaders processHeader(HttpHeaders httpHeaders, Map headerMap) {
        if (MapUtils.isNotEmpty(headerMap)) {
            for (Map.Entry entry : headerMap.entrySet()) {
                String key = entry.getKey();
                String value = entry.getValue();
                httpHeaders.add(key, value);
            }
        }

        return httpHeaders;
    }

    public static String processParameter(String url, Map parameterMap) {
        if (MapUtils.isNotEmpty(parameterMap)) {
            StringBuilder parameterStringBuilder = new StringBuilder();

            int index = 0;
            for (Map.Entry entry : parameterMap.entrySet()) {
                String key = entry.getKey();
                String value = entry.getValue();

                parameterStringBuilder.append(key + DiscoveryConstant.EQUALS + value);
                if (index < parameterMap.size() - 1) {
                    parameterStringBuilder.append("&");
                }

                index++;
            }

            String parameter = parameterStringBuilder.toString();
            parameter = StringUtils.isNotEmpty(parameter) ? "?" + parameter : "";

            url += parameter;
        }

        return url;
    }

    public static HttpHeaders processCookie(HttpHeaders httpHeaders, Map cookieMap) {
        if (MapUtils.isNotEmpty(cookieMap)) {
            List cookieList = new ArrayList();
            for (Map.Entry entry : cookieMap.entrySet()) {
                String key = entry.getKey();
                String value = entry.getValue();

                cookieList.add(key + DiscoveryConstant.EQUALS + value);
            }

            httpHeaders.put(DiscoveryConstant.COOKIE, cookieList);
        }

        return httpHeaders;
    }

    public static  T fromJson(RestTemplate restTemplate, String result, TypeReference typeReference) {
        try {
            return JsonUtil.fromJson(result, typeReference);
        } catch (Exception e) {
            String cause = getCause(restTemplate);
            if (StringUtils.isNotEmpty(cause)) {
                throw new IllegalArgumentException(cause);
            }

            throw e;
        }
    }

    public static String getCause(RestTemplate restTemplate) {
        ResponseErrorHandler responseErrorHandler = restTemplate.getErrorHandler();
        if (responseErrorHandler instanceof DiscoveryResponseErrorHandler) {
            DiscoveryResponseErrorHandler discoveryResponseErrorHandler = (DiscoveryResponseErrorHandler) responseErrorHandler;

            return discoveryResponseErrorHandler.getCause();
        }

        return null;
    }
}