package com.cloud.common.utils;
|
|
import com.alibaba.fastjson.JSONObject;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.http.HttpEntity;
|
import org.springframework.http.HttpHeaders;
|
import org.springframework.http.HttpMethod;
|
import org.springframework.http.MediaType;
|
import org.springframework.http.converter.FormHttpMessageConverter;
|
import org.springframework.stereotype.Component;
|
import org.springframework.util.LinkedMultiValueMap;
|
import org.springframework.util.MultiValueMap;
|
import org.springframework.web.client.RestClientException;
|
import org.springframework.web.client.RestTemplate;
|
|
import javax.annotation.PostConstruct;
|
import java.util.Iterator;
|
import java.util.List;
|
import java.util.Set;
|
import java.util.regex.Matcher;
|
import java.util.regex.Pattern;
|
|
/**
|
* @author lp
|
*/
|
@Component
|
public class RestTemplateUtil {
|
|
private static final Pattern QUERY_PARAM_PATTREN = Pattern.compile("([^&=]+)(=?)([^&]+)?");
|
|
@Autowired
|
private RestTemplate restTemplate;
|
|
private static RestTemplateUtil restTemplateUtil;
|
|
@PostConstruct
|
public void init() {
|
restTemplateUtil = this;
|
restTemplateUtil.restTemplate = this.restTemplate;
|
}
|
|
|
/**
|
* httpclient POST调用，可以携带token,[-封装的postForEntity接口-]
|
* @param url 请求地址
|
* @param paramMap 请求参数，必须是MultiValueMap类型
|
* @param responseType 返回类型
|
* @param isNeedToken 请求header是否需要携带token
|
* @return T
|
* @throws RestClientException
|
*/
|
public static <T> T postHeaderForEntity(String url, MultiValueMap<String, String> paramMap, Class<T> responseType, boolean isNeedToken)
|
throws RestClientException {
|
HttpHeaders headers = new HttpHeaders();
|
headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED);
|
if(isNeedToken) {
|
headers.add("Authorization", "Bearer " + RequestUtil.getTokenByParam());
|
}
|
HttpEntity<MultiValueMap<String, String>> request = new HttpEntity<MultiValueMap<String, String>>(paramMap, headers);
|
return restTemplateUtil.restTemplate.postForEntity(url, request, responseType).getBody();
|
}
|
|
|
public static <T> T getHeaderForEntity(String url, MultiValueMap<String, Object> paramMap, Class<T> responseType, boolean isNeedToken)
|
throws RestClientException {
|
HttpHeaders headers = new HttpHeaders();
|
headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED);
|
if(isNeedToken) {
|
headers.add("Authorization", "Bearer " + RequestUtil.getTokenByParam());
|
}
|
HttpEntity<MultiValueMap<String, Object>> request = new HttpEntity<MultiValueMap<String, Object>>(paramMap, headers);
|
return restTemplateUtil.restTemplate.exchange(url, HttpMethod.GET, request, responseType).getBody();
|
|
}
|
|
|
/**
|
* @param url
|
* @param params
|
* @return
|
* @Title: get
|
* @author: lp
|
* @Description: TODO
|
* @return: String
|
*/
|
public static String get(String url, JSONObject params,boolean isNeedToken) {
|
if(isNeedToken){
|
params.put("access_token", RequestUtil.getTokenByParam());
|
}
|
String response = restTemplateUtil.restTemplate.getForObject(expandURL(url, params.keySet()), String.class, params);
|
return response;
|
}
|
|
/**
|
* @param url
|
* @param params
|
* @param mediaType
|
* @param isNeedToken
|
* @return
|
* @Title: post
|
* @author: lp
|
* @Description: 将参数都拼接在url之后
|
* @return: String
|
*/
|
public static String post(String url, JSONObject params, MediaType mediaType,boolean isNeedToken) {
|
// 拿到header信息
|
HttpHeaders requestHeaders = new HttpHeaders();
|
requestHeaders.setContentType(mediaType);
|
if(isNeedToken) {
|
requestHeaders.add("Authorization", "Bearer " + RequestUtil.getTokenByParam());
|
}
|
HttpEntity<JSONObject> requestEntity = (mediaType == MediaType.APPLICATION_JSON
|
|| mediaType == MediaType.APPLICATION_JSON_UTF8) ? new HttpEntity<JSONObject>(params, requestHeaders)
|
: new HttpEntity<JSONObject>(null, requestHeaders);
|
String result = (mediaType == MediaType.APPLICATION_JSON || mediaType == MediaType.APPLICATION_JSON_UTF8)
|
? restTemplateUtil.restTemplate.postForObject(url, requestEntity, String.class)
|
: restTemplateUtil.restTemplate.postForObject(expandURL(url, params.keySet()), requestEntity, String.class, params);
|
return result;
|
}
|
|
/**
|
* @param url
|
* @param params
|
* @param mediaType
|
* @param clz
|
* @return
|
* @Title: post
|
* @author: lp
|
* @Description: 发送json或者form格式数据
|
* @return: String
|
*/
|
public static <T> T post(String url, JSONObject params, MediaType mediaType, Class<T> clz,boolean isNeedToken) {
|
//这是为 MediaType.APPLICATION_FORM_URLENCODED 格式HttpEntity 数据 添加转换器
|
//还有就是，如果是APPLICATION_FORM_URLENCODED方式发送post请求，
|
//也可以直接HttpHeaders requestHeaders = new HttpHeaders(createMultiValueMap(params)，true)，就不用增加转换器了
|
restTemplateUtil.restTemplate.getMessageConverters().add(new FormHttpMessageConverter());
|
// 设置header信息
|
HttpHeaders requestHeaders = new HttpHeaders();
|
requestHeaders.setContentType(mediaType);
|
if(isNeedToken) {
|
requestHeaders.add("Authorization", "Bearer " + RequestUtil.getTokenByParam());
|
}
|
HttpEntity<?> requestEntity = (
|
mediaType == MediaType.APPLICATION_JSON
|
|| mediaType == MediaType.APPLICATION_JSON_UTF8)
|
? new HttpEntity<JSONObject>(params, requestHeaders)
|
: (mediaType == MediaType.APPLICATION_FORM_URLENCODED
|
? new HttpEntity<MultiValueMap>(createMultiValueMap(params), requestHeaders)
|
: new HttpEntity<>(null, requestHeaders));
|
|
T result = (mediaType == MediaType.APPLICATION_JSON || mediaType == MediaType.APPLICATION_JSON_UTF8)
|
? restTemplateUtil.restTemplate.postForObject(url, requestEntity, clz)
|
: restTemplateUtil.restTemplate.postForObject(mediaType == MediaType.APPLICATION_FORM_URLENCODED
|
? url
|
: expandURL(url, params.keySet()), requestEntity, clz, params);
|
|
return result;
|
}
|
|
|
private static MultiValueMap<String, String> createMultiValueMap(JSONObject params) {
|
MultiValueMap<String, String> map = new LinkedMultiValueMap<>();
|
for(String key : params.keySet()) {
|
if(params.get(key) instanceof List) {
|
for(Iterator<String> it = ((List<String>) params.get(key)).iterator(); it.hasNext(); ) {
|
String value = it.next();
|
map.add(key, value);
|
}
|
} else {
|
map.add(key, params.getString(key));
|
}
|
}
|
return map;
|
}
|
|
|
private static String expandURL(String url, Set<?> keys) {
|
Matcher mc = RestTemplateUtil.QUERY_PARAM_PATTREN.matcher(url);
|
StringBuilder sb = new StringBuilder(url);
|
if (!mc.find()) {
|
sb.append("&");
|
} else {
|
sb.append("?");
|
}
|
|
for (Object key : keys) {
|
sb.append(key).append("=").append("{").append(key).append("}").append("&");
|
}
|
return sb.deleteCharAt(sb.length() - 1).toString();
|
}
|
|
}
|
