关于使用过滤器通过HttpServletResponseWrapper包装HttpServletResponse实现获取response中的返回数据进行处理的问题

最近在项目遇到对响应回来的是response中的数据进行处理的问题，即相关的处理过程中遇到的问题记录下来

下面是最简单的自定义Wrapper

public class ResponseWrapper extends HttpServletResponseWrapper {

    private ByteArrayOutputStream buffer;

    private ServletOutputStream out;

    public ResponseWrapper(HttpServletResponse httpServletResponse) {
        super(httpServletResponse);
        buffer = new ByteArrayOutputStream();
        out = new WrapperOutputStream(buffer);
    }

    @Override
    public ServletOutputStream getOutputStream() throws IOException {
        return out;
    }

    @Override
    public void flushBuffer() throws IOException {
        if (out != null) {
            out.flush();
        }
    }

    public byte[] getContent() throws IOException {
        flushBuffer();
        return buffer.toByteArray();
    }

    class WrapperOutputStream extends ServletOutputStream {
        private ByteArrayOutputStream bos;

        public WrapperOutputStream(ByteArrayOutputStream bos) {
            this.bos = bos;
        }

        @Override
        public void write(int b) throws IOException {
            bos.write(b);
        }

        @Override
        public boolean isReady() {

            // TODO Auto-generated method stub
            return false;

        }

        @Override
        public void setWriteListener(WriteListener arg0) {

            // TODO Auto-generated method stub

        }
    }
}

 

接下来是在 Filter中处理返回的response，

@Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
            throws IOException, ServletException {
        WrapperResponse wrapperResponse = new WrapperResponse(resp); 

        chain.doFilter(request, wrapperResponse);

        byte[] bytes = wrapperResponse.getContent(); 
        //此處的bytes為response返回來的數據，根據自身需求就response数据进行压缩，或者是进行加密等等一系列操作
        wrapperResponse.setContentLength(-1);
        ServletOutputStream out = response.getOutputStream();
        out.write(bytes);
        out.flush();
       
    }

 

其中 wrapperResponse.setContentLength(-1);添加这一行代码的作用是为了防止在使用框架过程中会出现返回的数据不全的问题题也可能会出现的问题页面响应速度变慢，一般来说此处功能就可以实现，但是如果你定义的filter拦截的是全部的请求，其中包括了文件下载的请求，而恰恰你的文件下载方法带有返回值对象，像这样

public ResultInfo donload(HttpServletRequest request,HttpServletResponse response,String Url){
        ResultInfo result =new ResultInfo();
        
        donloadFile(request,response,Url);
        result.setSuccess();
        result.setStatus(200);
    }

那么你filter中的byte[] bytes = wrapperResponse.getContent(); 数据会多出你封装的ResulInfo的信息，也就是说你下载的文件要比实际文件多出来那么些，这个具体如何处理要看你实际的业务需求了，在此建议如果使用过滤器和Wrapper的方式进行处理的话过滤器最好只拦截你想要处理的请求