package com.product.file.util;
|
|
import com.alibaba.fastjson.JSONObject;
|
import org.apache.http.HttpEntity;
|
import org.apache.http.client.ClientProtocolException;
|
import org.apache.http.client.config.RequestConfig;
|
import org.apache.http.client.methods.CloseableHttpResponse;
|
import org.apache.http.client.methods.HttpPost;
|
import org.apache.http.entity.ContentType;
|
import org.apache.http.entity.mime.MultipartEntityBuilder;
|
import org.apache.http.entity.mime.content.FileBody;
|
import org.apache.http.entity.mime.content.StringBody;
|
import org.apache.http.impl.client.CloseableHttpClient;
|
import org.apache.http.impl.client.HttpClients;
|
import org.apache.http.message.BasicNameValuePair;
|
import org.apache.http.util.EntityUtils;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
|
import java.io.File;
|
import java.io.FileOutputStream;
|
import java.io.IOException;
|
import java.io.OutputStream;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
|
public class HttpTest {
|
protected static Logger logger = LoggerFactory.getLogger(HttpTest.class);
|
|
static String uri = "http://127.0.0.1:9998/lx";
|
private static final String ACCESS_TOKEN = "token";
|
private static final String language="zh-CN";//zh-CN en
|
private static final String systemClientVersion="1.0.0";
|
private static final String systemClientType="Web";
|
|
/**
|
* 支持文件上传的接口调用
|
*/
|
public static void testFileUpload(Map<String, Object> map) {
|
CloseableHttpClient httpclient = HttpClients.createDefault();
|
try {
|
CloseableHttpClient client = null;
|
CloseableHttpResponse response = null;
|
Object token = map.get("token");
|
Object cookies = map.get("cookie");
|
String cookie[] = new String[2];
|
if (token == null) {
|
cookie = com.product.common.utils.HttpTest.login();
|
} else {
|
cookie[0] = cookies.toString();
|
cookie[1] = token.toString();
|
}
|
logger.info("cookie[0]="+cookie[0]+"------cookie[1]="+cookie[1]+"------------------------------------");
|
if (cookie == null || cookie[1] == null) {
|
return;
|
}
|
String url = null;
|
Map<String, String> fileMap = null;
|
JSONObject jsonParam = new JSONObject();
|
for (String key : map.keySet()) {// keySet获取map集合key的集合 然后在遍历key即可
|
if ("url/api".equals(key.trim())) {
|
url = map.get(key).toString();
|
} else if ("data".equals(key)) {
|
Map value = (Map) map.get(key);//
|
jsonParam.put(key, value);
|
} else if ("fileMap".equals(key)) {
|
fileMap = (Map) map.get(key);
|
} else if (map.get(key) instanceof List) {
|
List value = (List) map.get(key);//
|
jsonParam.put(key, value);
|
} else {
|
jsonParam.put(key, map.get(key));
|
}
|
}
|
jsonParam.put(ACCESS_TOKEN, cookie[1]);
|
|
List<BasicNameValuePair> parames = new ArrayList<>();
|
logger.info("上传json:" + jsonParam.toString());
|
parames.add(new BasicNameValuePair("formData", jsonParam.toString()));
|
|
HttpPost httppost = new HttpPost(uri + url);
|
httppost.addHeader("Cookie", cookie[0]);
|
httppost.addHeader("system-language-code", HttpTest.language);
|
httppost.addHeader("system-client-type", HttpTest.systemClientType);
|
httppost.addHeader("system-client-version", HttpTest.systemClientVersion);
|
RequestConfig requestConfig = RequestConfig.custom().setConnectTimeout(200000).setSocketTimeout(200000)
|
.build();
|
httppost.setConfig(requestConfig);
|
|
MultipartEntityBuilder multipartEntityBuilder = MultipartEntityBuilder.create();
|
for (Map.Entry<String, String> entry : fileMap.entrySet()) {
|
multipartEntityBuilder.addPart(entry.getKey(), new FileBody(new File(entry.getValue())));
|
}
|
StringBody comment = new StringBody(jsonParam.toString(), ContentType.TEXT_PLAIN);
|
multipartEntityBuilder.addPart("formData", comment);
|
HttpEntity reqEntity = multipartEntityBuilder.build();
|
|
System.out.println("上传参数：" + jsonParam);
|
httppost.setEntity(reqEntity);
|
System.out.println("executing request " + httppost.getRequestLine());
|
response = httpclient.execute(httppost);
|
try {
|
System.out.println(response.getStatusLine());
|
HttpEntity resEntity = response.getEntity();
|
if (resEntity != null) {
|
String responseEntityStr = EntityUtils.toString(response.getEntity());
|
System.out.println(responseEntityStr);
|
System.out.println("Response content length: " + resEntity.getContentLength());
|
}
|
EntityUtils.consume(resEntity);
|
} finally {
|
response.close();
|
}
|
} catch (ClientProtocolException e) {
|
e.printStackTrace();
|
} catch (IOException e) {
|
e.printStackTrace();
|
} finally {
|
try {
|
httpclient.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
|
/**
|
* 支持文件下载的接口调用
|
*/
|
public static void testFileDownload(Map<String, Object> map) {
|
CloseableHttpClient httpclient = HttpClients.createDefault();
|
try {
|
CloseableHttpResponse response = null;
|
Object token = map.get("token");
|
Object cookies = map.get("cookie");
|
String cookie[] = new String[2];
|
if (token == null) {
|
cookie = com.product.common.utils.HttpTest.login();
|
} else {
|
cookie[0] = cookies.toString();
|
cookie[1] = token.toString();
|
}
|
logger.info("cookie[0]="+cookie[0]+"------cookie[1]="+cookie[1]+"------------------------------------");
|
if (cookie == null || cookie[1] == null) {
|
return;
|
}
|
String url = null;
|
JSONObject jsonParam = new JSONObject();
|
for (String key : map.keySet()) {// keySet获取map集合key的集合 然后在遍历key即可
|
if ("url/api".equals(key.trim())) {
|
url = map.get(key).toString();
|
} else if ("data".equals(key)) {
|
Map value = (Map) map.get(key);//
|
jsonParam.put(key, value);
|
} else if (map.get(key) instanceof List) {
|
List value = (List) map.get(key);//
|
jsonParam.put(key, value);
|
} else {
|
jsonParam.put(key, map.get(key));
|
}
|
}
|
jsonParam.put(ACCESS_TOKEN, cookie[1]);
|
|
List<BasicNameValuePair> parames = new ArrayList<>();
|
logger.info("上传json:" + jsonParam.toString());
|
parames.add(new BasicNameValuePair("formData", jsonParam.toString()));
|
|
HttpPost httppost = new HttpPost(uri + url);
|
httppost.addHeader("Cookie", cookie[0]);
|
httppost.addHeader("system-language-code", HttpTest.language);
|
httppost.addHeader("system-client-type", HttpTest.systemClientType);
|
httppost.addHeader("system-client-version", HttpTest.systemClientVersion);
|
RequestConfig requestConfig = RequestConfig.custom().setConnectTimeout(200000).setSocketTimeout(200000).build();
|
httppost.setConfig(requestConfig);
|
|
MultipartEntityBuilder multipartEntityBuilder = MultipartEntityBuilder.create();
|
StringBody comment = new StringBody(jsonParam.toString(), ContentType.TEXT_PLAIN);
|
multipartEntityBuilder.addPart("formData", comment);
|
HttpEntity reqEntity = multipartEntityBuilder.build();
|
|
httppost.setEntity(reqEntity);
|
System.out.println("executing request " + httppost.getRequestLine());
|
response = httpclient.execute(httppost);
|
try {
|
System.out.println(response.getStatusLine());
|
HttpEntity resEntity = response.getEntity();
|
if (resEntity != null) {
|
System.out.println(reqEntity.getContentLength());
|
System.out.println(reqEntity.getContentType());
|
// String responseEntityStr = EntityUtils.toString(response.getEntity());
|
// System.out.println(responseEntityStr);
|
// System.out.println("Response content length: " + resEntity.getContentLength());
|
}
|
EntityUtils.consume(resEntity);
|
} finally {
|
response.close();
|
}
|
} catch (ClientProtocolException e) {
|
e.printStackTrace();
|
} catch (IOException e) {
|
e.printStackTrace();
|
} finally {
|
try {
|
httpclient.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
}
|
