springboot Java HttpServletRequest 转 CURL 命令 【Servlet】利用 HttpServletRequestWrapper 实现对request body的二次读取，可用于记录日志
https://blog.csdn.net/weixin_43833834/article/details/119816469
http://www.manongjc.com/detail/25-hjxgtagcwdpkpuv.html
https://blog.csdn.net/woluoyifan/article/details/81280702
private static final String FORMAT_HEADER = "-H \"%1$s:%2$s\"";
private static final String FORMAT_METHOD = "-X %1$s";
private static final String FORMAT_BODY = "-d '%1$s'";
private static final String FORMAT_URL = "\"%1$s\"";
private static final String CONTENT_TYPE = "Content-Type";
/**
* <p>
* HttpServletRequest 转化为 CURL 命令
* </p>
*
* @param request request
* @return String
* @author Tophua
* @since 2021/8/19
*/
public String getCurl(HttpServletRequest request) {
String curl;
try {
List<String> parts = new ArrayList<>();
parts.add("curl");
String url = request.getRequestURL().toString();
String method = request.getMethod();
String contentType = request.getContentType();
String queryString = request.getQueryString();
parts.add(String.format(ControllerLogAspect.FORMAT_METHOD, method.toUpperCase()));
Map<String, String> headers = new HashMap<>(16);
Enumeration<String> headerNames = request.getHeaderNames();
while (headerNames.hasMoreElements()) {
String key = headerNames.nextElement();
headers.put(key, request.getHeader(key));
}
headers.forEach((k, v) -> parts.add(String.format(ControllerLogAspect.FORMAT_HEADER, k, v)));
if (StrUtil.isNotEmpty(contentType) && !headers.containsKey(ControllerLogAspect.CONTENT_TYPE)) {
parts.add(String.format(ControllerLogAspect.FORMAT_HEADER, ControllerLogAspect.CONTENT_TYPE, contentType));
}
if (StrUtil.isNotEmpty(queryString)) {
url = HttpUtil.urlWithForm(url, queryString, CharsetUtil.CHARSET_UTF_8, false);
}
if (ContentType.isFormUrlEncode(contentType) && CollUtil.isNotEmpty(request.getParameterMap())) {
request.getParameterMap().forEach((k, v) ->
parts.add(StrUtil.format("--data-urlencode '{}={}'", k, ArrayUtil.get(v, 0))));
}
if (StrUtil.startWithIgnoreCase(contentType, ContentType.JSON.toString())) {
BodyReaderHttpServletRequestWrapper wrapper = (BodyReaderHttpServletRequestWrapper) request;
// String body = StrUtil.utf8Str(wrapper.getCachedBody());
String body = IoUtil.readUtf8(wrapper.getInputStream());
if (StrUtil.isNotEmpty(body)) {
parts.add(String.format(ControllerLogAspect.FORMAT_BODY, body));
}
}
parts.add(String.format(ControllerLogAspect.FORMAT_URL, url));
curl = StrUtil.join(" ", parts);
} catch (Exception e) {
e.printStackTrace();
curl = null;
}
return curl;
}
import javax.servlet.*;
import javax.servlet.annotation.WebFilter;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
@WebFilter(filterName = "httpServletRequestWrapperFilter", urlPatterns = "/*")
public class HttpServletRequestWrapperFilter implements Filter {
@Override
public void init(FilterConfig filterConfig) throws ServletException {
}
@Override
public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
throws IOException, ServletException {
if (request instanceof HttpServletRequest) {
ServletRequest requestWrapper = new BodyReaderHttpServletRequestWrapper((HttpServletRequest) request);
chain.doFilter(requestWrapper, response);
} else {
chain.doFilter(request, response);
}
}
@Override
public void destroy() {
}
}
import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.*;
public class BodyReaderHttpServletRequestWrapper extends HttpServletRequestWrapper {
private final byte[] bytes;
public BodyReaderHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
super(request);
try (BufferedInputStream bis = new BufferedInputStream(request.getInputStream());
ByteArrayOutputStream baos = new ByteArrayOutputStream()) {
byte[] buffer = new byte[1024];
int len;
while ((len = bis.read(buffer)) > 0) {
baos.write(buffer, 0, len);
}
this.bytes = baos.toByteArray();
String body = new String(this.bytes);
System.out.println(body);
} catch (IOException ex) {
throw ex;
}
}
@Override
public ServletInputStream getInputStream() throws IOException {
final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(this.bytes);
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 byteArrayInputStream.read();
}
};
}
@Override
public BufferedReader getReader() throws IOException {
return new BufferedReader(new InputStreamReader(this.getInputStream()));
}
}
最近项目上有一个需求，在api上收到的请求，需要在springmvc转化成实体参数之前把request body读取出来记录日志。
在通常的响应流程上，使用了request.getInputStream()之后，流就会失效，即这个request body的流只能读取一次，这也是流本身的特性所致（当然，还有一种特殊的流——推回输入流PushbackInputStream）。
通过servlet api，可以通过继承 HttpServletRequestWrapper 这个类，在 filter 中 对 request 进行一次封装，再传进chain中，如此便可以实现日志的记录。
最近项目上有一个需求，在api上收到的请求，需要在springmvc转化成实体参数之前把request body读取出来记录日志。
在通常的响应流程上，使用了request.getInputStream()之后，流就会失效，即这个request body的流只能读取一次，这也是流本身的特性所致（当然，还有一种特殊的流——推回输入流PushbackInputStream）。
通过servlet api，可以通过继承 HttpServletRequestWrapper 这个类，在 filter 中 对 request 进行一次封装，再传进chain中，如此便可以实现日志的记录。
————————————————
版权声明：本文为CSDN博主「苏笛南风」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/woluoyifan/article/details/81280702
