java使用Filter过滤器对Response返回值进行修改

转：java使用Filter过滤器对Response返回值进行修改

练习时只做了对request 的处理，这里记录一下，filter 对 response的处理。

原文地址：java使用Filter过滤器对Response返回值进行修改

有时候在开发过程中会有这样一个场景，有时需要对返回的数据进行处理。比如加密啊，添加某些标记啊等等。这种情况我们可以通过过滤器去处理。

过滤器是什么就不介绍了，百度一下，你就知道。

以下是代码：

返回值输出代理类：这个类主要是为了吧Response里面的返回值获取到，因为直接Response没有提供直接拿到返回值的方法。所以要通过代理来取得返回值

 1 package com.channel.filter;
 2
 3
 4 import java.io.ByteArrayOutputStream;
 5 import java.io.IOException;
 6
 7 import javax.servlet.ServletOutputStream;
 8 import javax.servlet.WriteListener;
 9 import javax.servlet.http.HttpServletResponse;
10 import javax.servlet.http.HttpServletResponseWrapper;
11
12
13 /**
14  * 返回值输出代理类
15  *
16  * @Title: ResponseWrapper
17  * @Description:
18  * @author kokJuis
19  * @date 上午9:52:11
20  */
21 public class ResponseWrapper extends HttpServletResponseWrapper
22 {
23
24     private ByteArrayOutputStream buffer;
25
26     private ServletOutputStream out;
27
28     public ResponseWrapper(HttpServletResponse httpServletResponse)
29     {
30         super(httpServletResponse);
31         buffer = new ByteArrayOutputStream();
32         out = new WrapperOutputStream(buffer);
33     }
34
35     @Override
36     public ServletOutputStream getOutputStream()
37         throws IOException
38     {
39         return out;
40     }
41
42     @Override
43     public void flushBuffer()
44         throws IOException
45     {
46         if (out != null)
47         {
48             out.flush();
49         }
50     }
51
52     public byte[] getContent()
53         throws IOException
54     {
55         flushBuffer();
56         return buffer.toByteArray();
57     }
58
59     class WrapperOutputStream extends ServletOutputStream
60     {
61         private ByteArrayOutputStream bos;
62
63         public WrapperOutputStream(ByteArrayOutputStream bos)
64         {
65             this.bos = bos;
66         }
67
68         @Override
69         public void write(int b)
70             throws IOException
71         {
72             bos.write(b);
73         }
74
75         @Override
76         public boolean isReady()
77         {
78
79             // TODO Auto-generated method stub
80             return false;
81
82         }
83
84         @Override
85         public void setWriteListener(WriteListener arg0)
86         {
87
88             // TODO Auto-generated method stub
89
90         }
91     }
92
93 }

过滤器类：

 1 package com.channel.filter;
 2
 3
 4 import java.io.IOException;
 5
 6 import javax.servlet.Filter;
 7 import javax.servlet.FilterChain;
 8 import javax.servlet.FilterConfig;
 9 import javax.servlet.ServletException;
10 import javax.servlet.ServletOutputStream;
11 import javax.servlet.ServletRequest;
12 import javax.servlet.ServletResponse;
13 import javax.servlet.http.HttpServletResponse;
14
15 import com.channel.util.RsaEncryptUtil;
16
17
18 /**
19  * 返回值输出过滤器，这里用来加密返回值
20  *
21  * @Title: ResponseFilter
22  * @Description:
23  * @author kokJuis
24  * @date 上午9:52:42
25  */
26 public class ResponseFilter implements Filter
27 {
28
29     @Override
30     public void doFilter(ServletRequest request, ServletResponse response, FilterChain filterChain)
31         throws IOException, ServletException
32     {
33         ResponseWrapper wrapperResponse = new ResponseWrapper((HttpServletResponse)response);//转换成代理类
34         // 这里只拦截返回，直接让请求过去，如果在请求前有处理，可以在这里处理
35         filterChain.doFilter(request, wrapperResponse);
36         byte[] content = wrapperResponse.getContent();//获取返回值
37     //判断是否有值
38         if (content.length > 0)
39         {
40
41             String str = new String(content, "UTF-8");
42             System.out.println("返回值:" + str);
43             String ciphertext = null;
44
45             try
46             {
47                 //......根据需要处理返回值
48             }
49             catch (Exception e)
50             {
51                 e.printStackTrace();
52             }
53         //把返回值输出到客户端
54             ServletOutputStream out = response.getOutputStream();
55             out.write(ciphertext.getBytes());
56             out.flush();
57         }
58
59     }
60
61     @Override
62     public void init(FilterConfig arg0)
63         throws ServletException
64     {
65
66     }
67
68     @Override
69     public void destroy()
70     {
71
72     }
73
74 }

配置很简单：

spring boot：

 1  　　/**
 2      * 配置过滤器，这里过滤器主要是对返回值做后继处理
 3      *
 4      * @return
 5      */
 6     @Bean
 7     public FilterRegistrationBean someFilterRegistration()
 8     {
 9         FilterRegistrationBean registration = new FilterRegistrationBean();
10         registration.setFilter(new ResponseFilter());// 配置一个返回值加密过滤器
11         registration.addUrlPatterns("/*");
12         registration.addInitParameter("paramName", "paramValue");
13         registration.setName("responseFilter");
14         return registration;
15     }

spring mvc 下直接在web.xml下添加配置即可

 1 <!-- 返回值过滤器 -->
 2     <filter>
 3         <filter-name>responseFilter</filter-name>
 4         <filter-class>com.im.filter.ResponseFilter</filter-class>
 5     </filter>
 6
 7     <filter-mapping>
 8         <filter-name>responseFilter</filter-name>
 9         <url-pattern>/*</url-pattern>
10     </filter-mapping>