Spring cached 的 wrapper 类读取请求响应内容

JAVA乐园

共 9260字,需浏览 19分钟

 ·

2021-05-11 20:03

0x01:wrapper 类说明

Servlet 的 request body 以及 response body 一旦流被读取了,就无法再次被读取了,因此这对于有些要做拦截业务请求来说,比较麻烦。那么如何处理这个业务场景呢?利用 filter,然后重写 HttpServletRequest 和 HttpServletResponse 包装一层,然后 proceed,最后 response 完之后在把cached 的 body 设置回原始响应。

Spring提供了 ContentCachingRequestWrapper 以及 ContentCachingResponseWrapper 两个类,来解决这类问题。

ContentCachingRequestWrapper wrappedRequest = new ContentCachingRequestWrapper(request);
ContentCachingResponseWrapper wrappedResponse = new ContentCachingResponseWrapper(response);

读取完 Response body 之后

wrappedResponse.copyBodyToResponse();

通过这个设置回去,就可以使得接口调用者可以正常接收响应了。


0x02:日志拦截实例

  • 继承OncePerRequestFilter,顾名思义该filter能够确保在一次请求只通过一次该 filter

  • 实现 Ordered,设置filter的执行顺序,一般设置最低级别执行

package com.filter;

import java.io.IOException;

import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.core.Ordered;
import org.springframework.web.filter.OncePerRequestFilter;

public class LogFilter extends OncePerRequestFilter implements Ordered {

    @Override
    public int getOrder() {
        // TODO Auto-generated method stub
        return 0;
    }

    @Override
    protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain)
            throws ServletException, IOException 
{
        // TODO Auto-generated method stub

    }

}

完整代码大致如下

package com.filter;

import java.io.IOException;

import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.Ordered;
import org.springframework.stereotype.Component;
import org.springframework.web.filter.OncePerRequestFilter;
import org.springframework.web.util.ContentCachingRequestWrapper;
import org.springframework.web.util.ContentCachingResponseWrapper;
import org.springframework.web.util.WebUtils;

@Component
public class LogFilter extends OncePerRequestFilter implements Ordered {

    
@Value("${openLog}")
    
private boolean openLog;
    
@Override
    
public int getOrder() {
        
return Ordered.LOWEST_PRECEDENCE-8;
    }

    
@Override
    
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain)
            throws ServletException, IOException 
{
        ContentCachingRequestWrapper wrappedRequest = 
null;
        ContentCachingResponseWrapper wrappedResponse = 
null;
        
try{
            
if(openLog){
                printHender(request);
                
printCookie(request);
                
printParameter(request);
                wrappedRequest = 
new ContentCachingRequestWrapper(request);
                wrappedResponse = 
new ContentCachingResponseWrapper(response);
                printRequest(wrappedRequest);
                printResponse(wrappedResponse);
            }
else{
                filterChain.doFilter(request, response);
            }
        }
finally {
            
if(wrappedResponse!=null){
                wrappedResponse.copyBodyToResponse();
            }
        }

    }

    
/**
     * 打印返回体
     * 
     * @param wrappedRequest
     */

    
private void printResponse(ContentCachingResponseWrapper wrappedResponse) {
        
try{
            ContentCachingResponseWrapper resp =  WebUtils.getNativeResponse(wrappedResponse, ContentCachingResponseWrapper.class);
            
byte[] responseBody = resp.getContentAsByteArray();
            String body = 
new String(responseBody, resp.getCharacterEncoding());
            
//TODO 打印

        }
catch (Exception e) {
            
// TODO: handle exception
        }

    }

    
/**
     * 打印请求体
     * 
     * @param wrappedRequest
     */

    
private void printRequest(ContentCachingRequestWrapper wrappedRequest) {
        
try{
            ContentCachingRequestWrapper req =  WebUtils.getNativeRequest(wrappedRequest, ContentCachingRequestWrapper.class);
            
byte[] reqestBody = req.getContentAsByteArray();
            String body = 
new String(reqestBody, req.getCharacterEncoding());
            
//TODO 打印

        }
catch (Exception e) {
            
// TODO: handle exception
        }
    }

    
/**
     * 打印参数
     * 
     * @param request
     */

    
private void printParameter(HttpServletRequest request) {
        
// TODO Auto-generated method stub

    }

    
/**
     * 打印cookie
     * 
     * @param request
     */

    
private void printCookie(HttpServletRequest request) {
        
// TODO Auto-generated method stub

    }

    
/**
     * 打印请求头
     * 
     * @param request
     */

    
private void printHender(HttpServletRequest request) {
        
// TODO Auto-generated method stub

    }

}

喜欢,在看


浏览 50
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报