有的时候 需要在 过滤器中或者拦截器中处理一些 数据问题 需要获取 Request body 里的参数
会发现 HttpServletRequest 这个类 获取不到 application/json;charset=UTF-8格式的数据
只能获取 form/data 的参数
如果使用 form/data 格式传输 会发现 控制器 @RequestBody 这个注解 是获取不到 这种格式的数据
但是又非要用 json格式传输 就需要 重写 HttpServletRequestWrapper 这个类 使用这个类 进行获取
package com.erp.init.wrapper;
import com.erp.init.utils.StreamUtil;
import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
/**
* @author json
* @description json格式的报文请求
* @date 2018/9/28 13:51.
*/
public class JsonParameterRequestWrapper extends HttpServletRequestWrapper {
//用于保存读取body中数据
private byte[] body;
private String bodyMessage;
public JsonParameterRequestWrapper(HttpServletRequest request) throws IOException {
super(request);
//读取请求的数据保存到本类当中
body = StreamUtil.readBytes(request.getReader(), "UTF-8");
bodyMessage = new String(body,"utf-8");
}
//覆盖(重写)父类的方法
@Override
public BufferedReader getReader() throws IOException {
return new BufferedReader(new InputStreamReader(getInputStream()));
}
//覆盖(重写)父类的方法
@Override
public ServletInputStream getInputStream() throws IOException {
final ByteArrayInputStream bais = new ByteArrayInputStream(body);
return new ServletInputStream() {
@Override
public boolean isFinished() {
return false;
}
@Override
public boolean isReady() {
return false;
}
@Override
public void setReadListener(ReadListener readListener) {
}
@Override
public int read() throws IOException {
return bais.read();
}
};
}
/**
* 获取body中的数据
* @return
*/
public byte[] getBody() {
return body;
}
/**
* 把处理后的参数放到body里面
* @param body
*/
public void setBody(byte[] body) {
this.body = body;
}
public String getBodyMessage() {
return bodyMessage;
}
}
在过滤器中 使用这个类 来获取 JSON 数据
StreamUtil 工具类
package com.erp.init.utils;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
public class StreamUtil {
public static byte[] readStream(InputStream stream,int length) throws IOException{
byte[]streamData=null;
List<Integer> lengths = new ArrayList<Integer>();
List<byte[]> buffers = new ArrayList<byte[]>();
int l = 0; int totalLength = 0; byte[] buffer = null; //
while (totalLength < length && l != -1) { //
buffer = new byte[length];
l = stream.read(buffer);
if (l != -1) {
lengths.add(new Integer(l));
buffers.add(buffer);
totalLength+=l;
}
}
if(totalLength==0) {
return null;
}
l=0;
streamData = new byte[totalLength];
length =buffers.size();
int blength=0;
byte[] bbuffer=null;
for (int i = 0; i < length; i++) {
blength = ((Integer) lengths.get(i)).intValue();
bbuffer = (byte[]) buffers.get(i);
System.arraycopy(bbuffer, 0, streamData, l,blength);
l=l+blength;
}
stream=null; lengths=null; buffers=null; buffer=null;
return streamData;
}
public static byte[] readBytes(BufferedReader bufferedReader,String charset) throws IOException{
StringBuffer sb = new StringBuffer();
String s;
while ((s = bufferedReader.readLine()) != null) {
sb.append(s);
}
if(sb.length() == 0){
return "".getBytes(charset);
}
return sb.toString().getBytes(charset);
}
}
但是会有一个细节
经常会使用@RequestBody注解,这个注解是非常的好用。在请求参数传到后台的时候做一个参数检验时,使用SpringMVC的拦截器,在拦截器里把request的数据读取出来然后校验。但是在使用了拦截器的时候会出现一个问题,在拦截器读取了request的数据,在Controller里面@RequestBody注解获取Json就会失败就读取不到数据
重写HttpServletRequestWrapper方法
这种方法就是通过重写HttpServletRequestWrapper把request的保存下来,然后通过过滤器保存下来的request在填充进去,这样就可以多次读取request了
这个类已经把这些问题都解决了
粘贴复制 使用即可