需求:在切面中存储请求参数
思路:
通过 RequestContextHolder 获取 HttpServletRequest ,然后获取 HttpServletRequest 的 InputStream ,读取 InputStream 的数据,这便是请求参数(参数在RequestBody的情况,参数在 parameter 可以直接 request.getParameterMap())。
遇到的问题:
1. InputStream 只能被消费一次 ,请求在进入 controller 层时 @RequestBody 注解已经将 InputStream 消费过了。需要重写 HttpServletRequestWrapper 的 getInputStream() 和 getReader() 方法,这样可以多次获取 InputStream 的数据。具体代码如下:
public class RepeatableStreamRequestWrapper extends HttpServletRequestWrapper {
private final String content;
public RepeatableStreamRequestWrapper(HttpServletRequest request) {
super(request);
StringBuilder stringBuilder = new StringBuilder();
BufferedReader bufferedReader = null;
InputStream inputStream = null;
try {
inputStream = request.getInputStream();
if (inputStream != null) {
bufferedReader = new BufferedReader(new InputStreamReader(inputStream));
char[] charBuffer = new char[128];
int bytesRead = -1;
while ((bytesRead = bufferedReader.read(charBuffer)) > 0) {
stringBuilder.append(charBuffer, 0, bytesRead);
}
} else {
stringBuilder.append("");
}
} catch (IOException ex) {
} finally {
if (inputStream != null) {
try {
inputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (bufferedReader != null) {
try {
bufferedReader.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
content = stringBuilder.toString();
}
@Override
public ServletInputStream getInputStream() throws IOException {
final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(content.getBytes());
ServletInputStream servletInputStream = 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 byteArrayInputStream.read();
}
};
return servletInputStream;
}
@Override
public BufferedReader getReader() throws IOException {
return new BufferedReader(new InputStreamReader(this.getInputStream()));
}
public String getContent() {
return this.content;
}
}
之后在 Filter 里注册我们重写的 HttpServletRequestWrapper
@WebFilter(urlPatterns = "/*", filterName = "repeatableStreamFilter")
@Slf4j
public class RepeatableStreamFilter implements Filter {
@Override
public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
ServletRequest repeatableStreamRequestWrapper = null;
if (servletRequest instanceof HttpServletRequest) {
repeatableStreamRequestWrapper = new RepeatableStreamRequestWrapper((HttpServletRequest) servletRequest);
}
if (repeatableStreamRequestWrapper == null) {
filterChain.doFilter(servletRequest, servletResponse);
} else {
filterChain.doFilter(repeatableStreamRequestWrapper, servletResponse);
}
}
}
这样便可以重复获取 Inputstream 中的数据了。
2.重写完 HttpServletRequestWrapper 接口报 java.io.IOException: 断开的管道
1.有可能是我们改的 HttpServletRequestWrapper 哪里写的有问题,导致流的管道断开。
2.原来就是经常短的,只不过没有发现。
可惜,针对流在网上没找到解决方案,所以我们决定解决两端。加大nginx的缓冲空间大小或者延长请求的timeout时间。
然后增大nginx的缓冲空间 并且 精简 大数据量的 接口的 传输数据 解决了问题。