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();
}
}