com.mars.cloud.request.util.HttpUtil Maven / Gradle / Ivy
package com.mars.cloud.request.util;
import com.alibaba.fastjson.JSONObject;
import com.mars.cloud.annotation.enums.ContentType;
import com.mars.cloud.core.cache.model.RestApiCacheModel;
import com.mars.cloud.request.rest.model.RequestParamModel;
import com.mars.cloud.request.util.model.MarsHeader;
import com.mars.cloud.request.util.model.HttpResultModel;
import com.mars.cloud.util.MarsCloudConfigUtil;
import com.mars.common.annotation.enums.ReqMethod;
import com.mars.common.constant.MarsConstant;
import com.mars.common.util.StringUtil;
import com.mars.server.server.request.model.MarsFileUpLoad;
import okhttp3.*;
import java.net.URLEncoder;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
/**
* HTTP请求工具类
*/
public class HttpUtil {
private static final String CONTENT_TYPE_JSON = "application/json;charset=UTF-8";
private static final String FORM_DATA = "multipart/form-data";
private static final String CONTENT_TYPE_STR = "Content-type".toUpperCase();
/**
* 发起请求
*
* @param restApiCacheModel
* @param params
* @return
*/
public static HttpResultModel request(RestApiCacheModel restApiCacheModel, Object[] params, ContentType contentType, MarsHeader marsHeader) throws Exception {
if (contentType == null) {
throw new Exception("必须指定ContentType");
}
if (restApiCacheModel.getReqMethod().equals(ReqMethod.GET) && !contentType.equals(ContentType.FORM)) {
throw new Exception("请求的接口，请求方式为GET，所以ContentType只能为FORM，接口名:" + restApiCacheModel.getUrl());
}
if (restApiCacheModel.getReqMethod().equals(ReqMethod.GET)) {
return formGet(restApiCacheModel, params, marsHeader);
} else {
if (contentType.equals(ContentType.FORM)) {
return formPost(restApiCacheModel, params, marsHeader);
} else if (contentType.equals(ContentType.FORM_DATA)) {
return formData(restApiCacheModel, params, marsHeader);
} else if (contentType.equals(ContentType.JSON)) {
return json(restApiCacheModel, params, marsHeader);
} else {
throw new Exception("请求的接口ContentType未知，接口名:" + restApiCacheModel.getUrl());
}
}
}
/**
* formData提交
*
* @param restApiModel
* @param params
* @return
* @throws Exception
*/
private static HttpResultModel formData(RestApiCacheModel restApiModel, Object[] params, MarsHeader marsHeader) throws Exception {
OkHttpClient okHttpClient = getOkHttpClient();
/* 发起post请求 将数据传递过去 */
MediaType formData = MediaType.parse(FORM_DATA);
MultipartBody.Builder builder = new MultipartBody.Builder();
builder.setType(formData);
/* 将参数转成统一规格的对象，进行下面的传参操作 */
Map requestParamModelMap = ParamConversionUtil.getRequestParamModelList(params);
for (RequestParamModel requestParamModel : requestParamModelMap.values()) {
if (requestParamModel.isFile()) {
// 如果是文件
Map marsFileUpLoadMap = requestParamModel.getMarsFileUpLoads();
for (MarsFileUpLoad marsFileUpLoad : marsFileUpLoadMap.values()) {
byte[] file = ParamConversionUtil.toByteArray(marsFileUpLoad.getInputStream());
RequestBody fileBody = RequestBody.create(file, MediaType.parse("application/octet-stream"));
builder.addFormDataPart(marsFileUpLoad.getName(), marsFileUpLoad.getFileName(), fileBody);
}
} else {
// 如果不是文件
Object val = requestParamModel.getValue();
if (val == null) {
continue;
}
if (val instanceof String[]) {
String[] valStr = (String[]) val;
for (String str : valStr) {
builder.addFormDataPart(requestParamModel.getName(), str);
}
} else {
builder.addFormDataPart(requestParamModel.getName(), val.toString());
}
}
}
Request request = getRequestBuilder(restApiModel, builder.build(), marsHeader)
.url(restApiModel.getUrl())
.build();
return okCall(okHttpClient, request);
}
/**
* post表单提交
*
* @param restApiModel
* @param params
* @return
*/
private static HttpResultModel formPost(RestApiCacheModel restApiModel, Object[] params, MarsHeader marsHeader) throws Exception {
OkHttpClient okHttpClient = getOkHttpClient();
JSONObject jsonParam = ParamConversionUtil.conversionToJson(params);
FormBody.Builder formBodyBuilder = new FormBody.Builder();
for (String key : jsonParam.keySet()) {
Object val = jsonParam.get(key);
if (StringUtil.isNull(val)) {
continue;
}
if (val instanceof String[]) {
String[] paramStr = (String[]) val;
if (paramStr == null) {
continue;
}
for (String str : paramStr) {
formBodyBuilder.add(key, str);
}
} else {
formBodyBuilder.add(key, val.toString());
}
}
RequestBody formBody = formBodyBuilder.build();
Request request = getRequestBuilder(restApiModel, formBody, marsHeader)
.url(restApiModel.getUrl())
.build();
return okCall(okHttpClient, request);
}
/**
* get表单提交
*
* @param restApiModel
* @param params
* @return
*/
private static HttpResultModel formGet(RestApiCacheModel restApiModel, Object[] params, MarsHeader marsHeader) throws Exception {
OkHttpClient okHttpClient = getOkHttpClient();
JSONObject jsonParam = ParamConversionUtil.conversionToJson(params);
StringBuffer paramStr = new StringBuffer();
boolean isFirst = true;
for (String key : jsonParam.keySet()) {
Object val = jsonParam.get(key);
if (StringUtil.isNull(val)) {
continue;
}
if (isFirst) {
paramStr.append("?");
} else {
paramStr.append("&");
}
if (val instanceof String[]) {
String[] paramStrings = (String[]) val;
if (paramStrings == null) {
continue;
}
for (int i = 0; i < paramStrings.length; i++) {
String va = paramStrings[i];
if (StringUtil.isNull(va)) {
continue;
}
String pStr = paramStr.toString();
if (i > 0 && !pStr.endsWith("?") && !pStr.endsWith("&")) {
paramStr.append("&");
}
paramStr.append(key);
paramStr.append("=");
paramStr.append(URLEncoder.encode(va, MarsConstant.ENCODING));
}
} else {
paramStr.append(key);
paramStr.append("=");
paramStr.append(URLEncoder.encode(val.toString(), MarsConstant.ENCODING));
}
isFirst = false;
}
Request request = getRequestBuilder(restApiModel, null, marsHeader)
.url(restApiModel.getUrl() + paramStr.toString())
.build();
return okCall(okHttpClient, request);
}
/**
* json提交
*
* @param restApiModel
* @param params
* @return
*/
private static HttpResultModel json(RestApiCacheModel restApiModel, Object[] params, MarsHeader marsHeader) throws Exception {
String jsonStrParam = "{}";
JSONObject jsonParam = ParamConversionUtil.conversionToJson(params);
if (jsonParam != null) {
jsonStrParam = jsonParam.toJSONString();
}
OkHttpClient okHttpClient = getOkHttpClient();
MediaType mediaType = MediaType.parse(CONTENT_TYPE_JSON);
RequestBody requestbody = RequestBody.create(jsonStrParam, mediaType);
Request request = getRequestBuilder(restApiModel, requestbody, marsHeader)
.url(restApiModel.getUrl())
.build();
return okCall(okHttpClient, request);
}
/**
* 根据接口的请求方式，返回不同的Builder
*
* @param restApiModel
* @param requestBody
* @return
*/
private static Request.Builder getRequestBuilder(RestApiCacheModel restApiModel, RequestBody requestBody, MarsHeader marsHeader) {
Request.Builder builder = new Request.Builder();
builder = setHeaderForRequestBuilder(builder, marsHeader);
switch (restApiModel.getReqMethod()) {
case POST:
builder.post(requestBody);
break;
case PUT:
builder.put(requestBody);
break;
case DELETE:
builder.delete(requestBody);
break;
case GET:
builder.get();
break;
}
return builder;
}
/**
* 添加请求头
* @param builder
* @param marsHeader
* @return
*/
private static Request.Builder setHeaderForRequestBuilder(Request.Builder builder, MarsHeader marsHeader){
if (marsHeader != null && marsHeader.size() > 0) {
for (String key : marsHeader.keySet()) {
List value = marsHeader.get(key);
if (value == null || key == null || key.toUpperCase().equals(CONTENT_TYPE_STR)) {
continue;
}
for(String val : value){
builder.addHeader(key, val);
}
}
}
return builder;
}
/**
* 开始请求
*
* @param okHttpClient 客户端
* @param request 请求
* @return 结果
* @throws Exception 异常
*/
private static HttpResultModel okCall(OkHttpClient okHttpClient, Request request) throws Exception {
Call call = okHttpClient.newCall(request);
Response response = call.execute();
int code = response.code();
ResponseBody responseBody = response.body();
if (code != 200) {
throw new Exception("请求接口出现异常:" + responseBody.string());
}
HttpResultModel httpResultModel = new HttpResultModel();
String head = response.header("Content-Disposition");
httpResultModel.setFileName(head);
httpResultModel.setInputStream(responseBody.byteStream());
return httpResultModel;
}
/**
* 获取okHttp客户端
*
* @return 客户端
* @throws Exception 异常
*/
private static OkHttpClient getOkHttpClient() throws Exception {
long timeOut = getTimeOut();
OkHttpClient okHttpClient = new OkHttpClient.Builder()
.connectTimeout(timeOut, TimeUnit.SECONDS)//设置连接超时时间
.readTimeout(timeOut, TimeUnit.SECONDS)//设置读取超时时间
.build();
return okHttpClient;
}
/**
* 从配置中获取超时时间
*
* @return
*/
private static long getTimeOut() {
try {
return MarsCloudConfigUtil.getMarsCloudConfig().getCloudConfig().getTimeOut();
} catch (Exception e) {
return 100L;
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy