zuul获取参数工具类,代码如下:
package com.openailab.oascloud.gateway.util;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.parser.Feature;
import com.google.common.collect.Maps;
import com.netflix.zuul.context.RequestContext;
import com.netflix.zuul.http.HttpServletRequestWrapper;
import com.netflix.zuul.http.ServletInputStreamWrapper;
import com.openailab.oascloud.common.util.IPUtil;
import com.openailab.oascloud.gateway.filter.CharacterEncodeFilter;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.apache.commons.fileupload.servlet.ServletRequestContext;
import org.apache.tomcat.util.http.fileupload.FileItemIterator;
import org.apache.tomcat.util.http.fileupload.FileItemStream;
import org.apache.tomcat.util.http.fileupload.util.Streams;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.StreamUtils;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.nio.charset.Charset;
import java.util.*;
/**
* @Classname: com.openailab.oascloud.gateway.util.ParamUtil
* @Description: 描述
* @Author: zxzhang
* @Date: 2019/6/27
*/
public class ParamUtil {
private static final Logger LOGGER = LoggerFactory.getLogger(CharacterEncodeFilter.class);
/**
* 获取请求参数(Get||Post)
*
* @param ctx
* @return java.util.Map
* @author zxzhang
* @date 2019/6/27
*/
public static Map getRequestParams(RequestContext ctx) {
String method = ctx.getRequest().getMethod();
String uri = ctx.getRequest().getRequestURI();
//判断是POST请求还是GET请求(不同请求获取参数方式不同)
LinkedHashMap param = null;
try {
if (uri.startsWith("/zuul")) {
byte[] requestByte = saveaIns(ctx.getRequest().getInputStream());
rewriteRequest(ctx, requestByte);
param = Maps.newLinkedHashMap();
DiskFileItemFactory factory = new DiskFileItemFactory();
ServletFileUpload upload = new ServletFileUpload(factory);
upload.setHeaderEncoding("UTF-8");
List list = upload.parseRequest(new ServletRequestContext(ctx.getRequest()));
for (FileItem item : list) {
String name = item.getFieldName();
if (item.isFormField()) {
String value = item.getString("UTF-8");
param.put(name, value);
} else {
String filename = item.getName();
param.put(name, filename);
}
}
rewriteRequest(ctx, requestByte);
} else {
if ("GET".equals(method.toUpperCase())) {
Map> map = ctx.getRequestQueryParams();
if (!(Objects.isNull(map) || map.isEmpty())) {
param = Maps.newLinkedHashMap();
for (Map.Entry> entry : map.entrySet()) {
param.put(entry.getKey(), entry.getValue().get(0));
}
}
} else if ("POST".equals(method.toUpperCase())) {
try (InputStream inputStream = ctx.getRequest().getInputStream()) {
String body = StreamUtils.copyToString(inputStream, Charset.forName("UTF-8"));
LOGGER.info("***************原始参数:{}***************", body);
param = JSONObject.parseObject(body, LinkedHashMap.class, Feature.OrderedField);
} catch (IOException e) {
e.printStackTrace();
}
}
}
} catch (Exception e) {
LOGGER.error("***************ParamUtil->getRequestParams throw Exception:{}***************", e);
}
return param;
}
/**
* 设置请求参数(Get||Post)
*
* @param ctx
* @return void
* @author zxzhang
* @date 2019/6/28
*/
public static void setRequestParams(RequestContext ctx) {
String uri = ctx.getRequest().getRequestURI();
String source = uri.indexOf("/platform") > 0 ? "platform" : "application";
String client = uri.indexOf("/web") > 0 ? "web" : "app";
String method = ctx.getRequest().getMethod();
HttpServletRequest request = ctx.getRequest();
if (uri.startsWith("/zuul")) {
return;
} else {
//判断是POST请求还是GET请求(不同请求获取参数方式不同)
if ("GET".equals(method.toUpperCase())) {
Map> param = ctx.getRequestQueryParams();
if (Objects.isNull(param) || param.isEmpty()) {
param = Maps.newHashMap();
}
param.put("source", Arrays.asList(source));
param.put("client", Arrays.asList(client));
param.put("ip", Arrays.asList(IPUtil.getClientIp(request)));
ctx.setRequestQueryParams(param);
} else if ("POST".equals(method.toUpperCase())) {
try (InputStream inputStream = ctx.getRequest().getInputStream()) {
String body = StreamUtils.copyToString(inputStream, Charset.forName("UTF-8"));
Map param = JSONObject.parseObject(body);
if (Objects.isNull(param) || param.isEmpty()) {
param = Maps.newHashMap();
}
param.put("source", source);
param.put("client", client);
param.put("ip", IPUtil.getClientIp(request));
// 重写上下文的HttpServletRequestWrapper
final byte[] paramBytes = param.toString().getBytes();
rewriteRequest(ctx, paramBytes);
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
private static void rewriteRequest(RequestContext ctx, byte[] paramBytes) {
ctx.setRequest(new HttpServletRequestWrapper(ctx.getRequest()) {
@Override
public ServletInputStream getInputStream() throws IOException {
return new ServletInputStreamWrapper(paramBytes);
}
@Override
public int getContentLength() {
return paramBytes.length;
}
@Override
public long getContentLengthLong() {
return paramBytes.length;
}
});
}
/**
* 保存流对象(输入流在第二次使用的时候会失效)
* 在需要用到InputStream的地方再封装成InputStream
* ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(buf);
* Workbook wb = new HSSFWorkbook(byteArrayInputStream);//byteArrayInputStream 继承了InputStream,故这样用并没有问题
* 如果只需要用到一次inputstream流,就不用这样啦,直接用就OK
*
* @param ins
*/
public static byte[] saveaIns(InputStream ins) {
byte[] buf = null;
try {
if (ins != null) {
buf = org.apache.commons.io.IOUtils.toByteArray(ins);//ins为InputStream流
}
} catch (IOException e) {
e.printStackTrace();
}
return buf;
}
}