init
This commit is contained in:
Mauve
@@ -0,0 +1,310 @@
|
||||
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;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user