import com.alibaba.fastjson.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.web.client.RestTemplate;
import java.util.Map;
/**
* SpringBoot2.0使用RestTemplateTest請求API
**/
public class RestTemplateTest {
private static final Logger logger = LoggerFactory.getLogger(RestTemplateTest.class);
@Autowired
private RestTemplate restTemplate;
@Value("${xxx.xxx}")
private String accountNo;
@Value("${xxx.xxx}")
private String getUrl;
@Value("${xxx.xxx}")
private String postUrl;
private static final String AND_MARK = "&";
private static final String EQUAL_MARK = "=";
/**
* get請求
* @param parameterMap 發送內容
*/
public JSONObject get(Map<String,String> parameterMap){
long beginTime = System.currentTimeMillis();
String result;
try {
String paramString = getParams(parameterMap);
String sendUrl = getUrl + "?accountNo=" + accountNo + paramString;
logger.info("發送地址:{}",sendUrl);
result = restTemplate.getForObject(sendUrl, String.class);
logger.info("接收到結果:{}", result);
}finally {
long endTime = System.currentTimeMillis();
logger.info("開始時間:{},結束時間:{},此次調用時長爲(毫秒):{}",beginTime,endTime,endTime-beginTime);
}
return JSONObject.parseObject(result);
}
/**
* post請求
* @param contentJson 發送內容
*/
public JSONObject post(JSONObject contentJson){
long beginTime = System.currentTimeMillis();
try {
String sendUrl = postUrl + "?accountNo=" + accountNo;
logger.info("發送地址:{}",sendUrl);
logger.info("發送內容:{}",JSONObject.toJSONString(contentJson));
String result = restTemplate.postForObject(sendUrl, getJsonHttpEntity(contentJson), String.class);
logger.info("接收到結果:{}",result);
return JSONObject.parseObject(result);
}finally {
long endTime = System.currentTimeMillis();
logger.info("開始時間:{},結束時間:{},此次調用時長爲(毫秒):{}",beginTime,endTime,endTime-beginTime);
}
}
/**
* 獲取json請求體
* @param contentJson 請求內容
* @return HttpEntity
*/
private HttpEntity getJsonHttpEntity(JSONObject contentJson){
HttpHeaders headers = new HttpHeaders();
//接收字符集
headers.set("Accept-Charset", "utf-8");
//內容類型
headers.set("Content-type", MediaType.APPLICATION_JSON_UTF8_VALUE);
return new HttpEntity<>(contentJson, headers);
}
/**
* 轉換爲get方式的參數
* @param param map類型的參數
* @return String
*/
private static String getParams(Map<String, String> param) {
StringBuilder stringBuilder = new StringBuilder();
for (Map.Entry<String, String> entry : param.entrySet()) {
stringBuilder.append(AND_MARK).append(entry.getKey()).append(EQUAL_MARK).append(entry.getValue());
}
return stringBuilder.toString();
}
}