Mauve
311 lines
12 KiB
Java
311 lines
12 KiB
Java
package com.mgm.mcrm.common.httpclient;
|
||
|
||
import org.apache.hc.client5.http.entity.UrlEncodedFormEntity;
|
||
import org.apache.hc.client5.http.entity.mime.HttpMultipartMode;
|
||
import org.apache.hc.client5.http.entity.mime.MultipartEntityBuilder;
|
||
import org.apache.hc.core5.http.ContentType;
|
||
import org.apache.hc.core5.http.HttpEntity;
|
||
import org.apache.hc.core5.http.HttpHeaders;
|
||
import org.apache.hc.core5.http.NameValuePair;
|
||
import org.apache.hc.core5.http.io.entity.ByteArrayEntity;
|
||
import org.apache.hc.core5.http.io.entity.FileEntity;
|
||
import org.apache.hc.core5.http.io.entity.StringEntity;
|
||
import org.apache.hc.core5.http.message.BasicHeader;
|
||
import org.apache.hc.core5.http.message.BasicNameValuePair;
|
||
import org.slf4j.Logger;
|
||
import org.slf4j.LoggerFactory;
|
||
|
||
import java.io.File;
|
||
import java.io.UnsupportedEncodingException;
|
||
import java.lang.reflect.Field;
|
||
import java.lang.reflect.Modifier;
|
||
import java.nio.charset.Charset;
|
||
import java.util.*;
|
||
import java.util.Map.Entry;
|
||
|
||
|
||
public class HttpEntityBuilder {
|
||
|
||
//传入参数特定类型
|
||
public static final String ENTITY_STRING="$ENTITY_STRING$";
|
||
public static final String ENTITY_JSON="$ENTITY_JSON$";
|
||
public static final String ENTITY_FILE="$ENTITY_FILEE$";
|
||
public static final String ENTITY_BYTES="$ENTITY_BYTES$";
|
||
public static final String ENTITY_INPUTSTREAM="$ENTITY_INPUTSTREAM$";
|
||
public static final String ENTITY_SERIALIZABLE="$ENTITY_SERIALIZABLE$";
|
||
public static final String ENTITY_MULTIPART="$ENTITY_MULTIPART$";
|
||
private static final List<String> SPECIAL_ENTITIY = Arrays.asList(ENTITY_STRING, ENTITY_JSON, ENTITY_BYTES, ENTITY_FILE, ENTITY_INPUTSTREAM, ENTITY_SERIALIZABLE, ENTITY_MULTIPART);
|
||
|
||
/**
|
||
* 是否开启debug，
|
||
*/
|
||
private static boolean debug = false;
|
||
|
||
private static final Logger log = LoggerFactory.getLogger(HttpEntityBuilder.class);
|
||
|
||
|
||
/**
|
||
* 检测url是否含有参数，如果有，则把参数加到参数列表中
|
||
*
|
||
* @param url 资源地址
|
||
* @param nvps 参数列表
|
||
* @param encoding 编码
|
||
* @return 返回去掉参数的url
|
||
* @throws UnsupportedEncodingException 不支持的编码异常
|
||
*/
|
||
public static String checkHasParas(String url, List<NameValuePair> nvps, String encoding) throws UnsupportedEncodingException {
|
||
// 检测url中是否存在参数
|
||
if (url.contains("?") && url.indexOf("?") < url.indexOf("=")) {
|
||
String[] paras = url.substring(url.indexOf("?") + 1).split("&");
|
||
for (String para : paras) {
|
||
int pos = para.indexOf("=");
|
||
if (pos >= 0) {
|
||
nvps.add(new BasicNameValuePair(para.substring(0, pos), para.substring(pos + 1)));
|
||
} else {
|
||
nvps.add(new BasicNameValuePair(para, ""));
|
||
}
|
||
}
|
||
url = url.substring(0, url.indexOf("?"));
|
||
}
|
||
return url;
|
||
}
|
||
|
||
/**
|
||
*
|
||
* 参数转换，将map中的参数，转到参数列表中
|
||
*
|
||
* @param nvps 参数列表
|
||
* @param map 参数列表（map）
|
||
* @param encoding 编码
|
||
* @return 返回HttpEntity
|
||
* @throws UnsupportedEncodingException 不支持的编码异常
|
||
*/
|
||
public static HttpEntity map2HttpEntity(List<NameValuePair> nvps, Map<String, Object> map, String encoding) throws UnsupportedEncodingException {
|
||
HttpEntity entity = null;
|
||
if(map!=null && map.size()>0){
|
||
boolean isSpecial = false;
|
||
// 拼接参数
|
||
for (Entry<String, Object> entry : map.entrySet()) {
|
||
if(SPECIAL_ENTITIY.contains(entry.getKey())){//判断是否在之中
|
||
isSpecial = true;
|
||
if(ENTITY_STRING.equals(entry.getKey())){//string
|
||
entity = new StringEntity(String.valueOf(entry.getValue()), Charset.forName(encoding));
|
||
break;
|
||
}else if(ENTITY_JSON.equals(entry.getKey())){//json
|
||
ContentType jsonContentType;
|
||
|
||
entity = new StringEntity(String.valueOf(entry.getValue()), Charset.forName(encoding));
|
||
String contentType = "application/json";
|
||
if (encoding != null && !encoding.trim().isEmpty()) {
|
||
// 创建带有编码的JSON ContentType
|
||
jsonContentType = ContentType.APPLICATION_JSON.withCharset(Charset.forName(encoding));
|
||
}else {
|
||
// 使用默认的JSON ContentType（通常UTF-8）
|
||
jsonContentType = ContentType.APPLICATION_JSON;
|
||
}
|
||
// 创建实体时直接传入ContentType
|
||
entity = new StringEntity(String.valueOf(entry.getValue()), jsonContentType);
|
||
break;
|
||
}else if(ENTITY_BYTES.equals(entry.getKey())){//file
|
||
byte[] data = (byte[]) entry.getValue();
|
||
entity = new ByteArrayEntity(data, ContentType.APPLICATION_OCTET_STREAM);
|
||
break;
|
||
}else if(ENTITY_FILE.equals(entry.getKey())){//file
|
||
if(File.class.isAssignableFrom(entry.getValue().getClass())){
|
||
entity = new FileEntity((File)entry.getValue(), ContentType.APPLICATION_OCTET_STREAM);
|
||
}else if(entry.getValue().getClass()==String.class){
|
||
entity = new FileEntity(new File((String) entry.getValue()), ContentType.create("text/plain", "UTF-8"));
|
||
}
|
||
break;
|
||
}else if(ENTITY_INPUTSTREAM.equals(entry.getKey())){//inputstream
|
||
// entity = new InputStreamEntity();
|
||
break;
|
||
}else if(ENTITY_SERIALIZABLE.equals(entry.getKey())){//serializeable
|
||
// entity = new SerializableEntity()
|
||
break;
|
||
}else if(ENTITY_MULTIPART.equals(entry.getKey())){//MultipartEntityBuilder
|
||
File[] files = null;
|
||
if(File.class.isAssignableFrom(entry.getValue().getClass().getComponentType())){
|
||
files=(File[])entry.getValue();
|
||
}else if(entry.getValue().getClass().getComponentType()==String.class){
|
||
String[] names = (String[]) entry.getValue();
|
||
files = new File[names.length];
|
||
for (int i = 0; i < names.length; i++) {
|
||
files[i] = new File(names[i]);
|
||
}
|
||
}
|
||
MultipartEntityBuilder builder = MultipartEntityBuilder.create();
|
||
builder.setCharset(Charset.forName(encoding));// 设置请求的编码格式
|
||
builder.setMode(HttpMultipartMode.LEGACY);// 设置浏览器兼容模式
|
||
int count = 0;
|
||
for (File file : files) {
|
||
// //把文件转换成流对象FileBody
|
||
// FileBody fileBody = new FileBody(file);
|
||
// builder.addPart(String.valueOf(map.get(ENTITY_MULTIPART+".name")) + count++, fileBody);
|
||
builder.addBinaryBody(String.valueOf(map.get(ENTITY_MULTIPART+".name")) + count++,file);
|
||
}
|
||
boolean forceRemoveContentTypeCharset = (boolean)map.get(ENTITY_MULTIPART+".rmCharset");
|
||
Map<String, Object> m = new HashMap<String, Object>();
|
||
m.putAll(map);
|
||
m.remove(ENTITY_MULTIPART);
|
||
m.remove(ENTITY_MULTIPART+".name");
|
||
m.remove(ENTITY_MULTIPART+".rmCharset");
|
||
Iterator<Entry<String, Object>> iterator = m.entrySet().iterator();
|
||
// 发送的数据
|
||
while (iterator.hasNext()) {
|
||
Entry<String, Object> e = iterator.next();
|
||
builder.addTextBody(e.getKey(), String.valueOf(e.getValue()), ContentType.create("text/plain", encoding));
|
||
}
|
||
entity = builder.build();// 生成 HTTP POST 实体
|
||
|
||
//强制去除contentType中的编码设置，否则，在某些情况下会导致上传失败
|
||
if(forceRemoveContentTypeCharset){
|
||
removeContentTypeCharset(encoding, entity);
|
||
}
|
||
break;
|
||
}else {
|
||
nvps.add(new BasicNameValuePair(entry.getKey(), String.valueOf(entry.getValue())));
|
||
}
|
||
}else{
|
||
nvps.add(new BasicNameValuePair(entry.getKey(), String.valueOf(entry.getValue())));
|
||
}
|
||
}
|
||
if(!isSpecial) {
|
||
entity = new UrlEncodedFormEntity(nvps, Charset.forName(encoding));
|
||
}
|
||
}
|
||
return entity;
|
||
}
|
||
|
||
/**
|
||
* 移除content-type中的charset
|
||
*
|
||
* @param encoding 编码
|
||
* @param entity 请求参数及数据信息
|
||
*/
|
||
private static void removeContentTypeCharset(String encoding, HttpEntity entity) {
|
||
try {
|
||
Class<?> clazz = entity.getClass();
|
||
Field field = clazz.getDeclaredField("contentType");
|
||
field.setAccessible(true); //将字段的访问权限设为true：即去除private修饰符的影响
|
||
if(Modifier.isFinal(field.getModifiers())){
|
||
Field modifiersField = Field.class.getDeclaredField("modifiers"); //去除final修饰符的影响，将字段设为可修改的
|
||
modifiersField.setAccessible(true);
|
||
modifiersField.setInt(field, field.getModifiers() & ~Modifier.FINAL);
|
||
}
|
||
BasicHeader o = (BasicHeader) field.get(entity);
|
||
field.set(entity, new BasicHeader(HttpHeaders.CONTENT_TYPE, o.getValue().replace("; charset="+encoding,"")));
|
||
} catch (NoSuchFieldException e) {
|
||
HttpEntityBuilder.exception(e);
|
||
} catch (SecurityException e) {
|
||
HttpEntityBuilder.exception(e);
|
||
} catch (IllegalArgumentException e) {
|
||
HttpEntityBuilder.exception(e);
|
||
} catch (IllegalAccessException e) {
|
||
HttpEntityBuilder.exception(e);
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* 生成参数
|
||
* 参数格式：k1=v1&k2=v2
|
||
*
|
||
* @param paras 参数列表
|
||
* @return 返回参数列表（map）
|
||
*/
|
||
public static Map<String,Object> buildParas(String paras){
|
||
String[] p = paras.split("&");
|
||
String[][] ps = new String[p.length][2];
|
||
int pos = 0;
|
||
for (int i = 0; i < p.length; i++) {
|
||
pos = p[i].indexOf("=");
|
||
ps[i][0]=p[i].substring(0,pos);
|
||
ps[i][1]=p[i].substring(pos+1);
|
||
pos = 0;
|
||
}
|
||
return buildParas(ps);
|
||
}
|
||
|
||
/**
|
||
* 生成参数
|
||
* 参数类型：{{"k1","v1"},{"k2","v2"}}
|
||
*
|
||
* @param paras 参数列表
|
||
* @return 返回参数列表（map）
|
||
*/
|
||
public static Map<String,Object> buildParas(String[][] paras){
|
||
// 创建参数队列
|
||
Map<String,Object> map = new HashMap<String, Object>();
|
||
for (String[] para: paras) {
|
||
map.put(para[0], para[1]);
|
||
}
|
||
return map;
|
||
}
|
||
|
||
/**
|
||
* 打印消息
|
||
*
|
||
* @param msg 消息
|
||
*/
|
||
public static void info(String msg){
|
||
if(debug){
|
||
log.info(msg);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 打印消息和异常堆栈
|
||
*
|
||
* @param msg 异常消息
|
||
* @param t 异常
|
||
*/
|
||
public static void infoException(String msg, Throwable t){
|
||
if(debug){
|
||
log.info(msg,t);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 打印错误消息
|
||
*
|
||
* @param msg 异常消息
|
||
*/
|
||
public static void error(String msg){
|
||
log.error(msg);
|
||
}
|
||
|
||
/**
|
||
* 打印错误消息和异常堆栈
|
||
*
|
||
* @param msg 异常消息
|
||
* @param t 异常
|
||
*/
|
||
public static void errorException(String msg, Throwable t){
|
||
log.error(msg, t);
|
||
}
|
||
|
||
/**
|
||
* 打印异常堆栈
|
||
*
|
||
* @param t 异常
|
||
*/
|
||
public static void exception(Throwable t){
|
||
log.error(String.valueOf(t));
|
||
}
|
||
|
||
/**
|
||
* 开启打印日志
|
||
*/
|
||
public static void debug() {
|
||
debug(true);
|
||
}
|
||
/**
|
||
* 开启或关闭打印日志
|
||
* @param debug 是否开启debug
|
||
*/
|
||
public static void debug(boolean debug) {
|
||
HttpEntityBuilder.debug = debug;
|
||
}
|
||
}
|