JavaWeb 之 Filter 敏感词汇过滤案例

需求：

　　1、 对day17_case案例录入的数据进行敏感词汇过滤
　　2、 敏感词汇参考 src路径下的《敏感词汇.txt》
　　3、 如果是敏感词汇，替换为 ***

分析：

　　1、 对request对象进行增强。增强获取参数相关方法
　　2、 放行。传递代理对象

代码实现：

 import org.springframework.cglib.proxy.InvocationHandler;
 import org.springframework.cglib.proxy.Proxy;

 import javax.servlet.*;
 import javax.servlet.annotation.WebFilter;
 import java.io.BufferedReader;
 import java.io.FileReader;
 import java.io.IOException;
 import java.lang.reflect.Method;
 import java.util.ArrayList;
 import java.util.List;

 /**
  * 敏感词汇过滤器
  */
 @WebFilter("/*")
 public class SensitiveWordsFilter implements Filter {

     public void doFilter(ServletRequest req, ServletResponse resp, FilterChain chain) throws ServletException, IOException {
         // 1 创建代理对象，增强 getparameter 方法
         ServletRequest proxy_req = (ServletRequest) Proxy.newProxyInstance(req.getClass().getClassLoader(), req.getClass().getInterfaces(), new InvocationHandler() {
             @Override
             public Object invoke(Object o, Method method, Object[] args) throws Throwable {

                 // 增强 getparameter 方法
                 // 判断是否是该方法
                 if(method.getName().equals("getParameter")) {
                     // 增强返回值
                     // 获取返回值
                     String value = (String) method.invoke(req,args);

                     if(value != null) {
                         for (String str : list) {
                             if(value.contains(str)){
                                 value = value.replaceAll(str,"***");
                             }
                         }
                     }

                     return  value;
                 }

                 return method.invoke(req,args);
             }
         });
         // 2 放行，传递增强的代理对象

         chain.doFilter(proxy_req, resp);
     }

     private List<String> list = new ArrayList<String>();  // 敏感词汇集合
     public void init(FilterConfig config) throws ServletException {

         try {
             // 1 加载文件
             // 获取文件的真实路径
             ServletContext servletContext = config.getServletContext();
             String realPath = servletContext.getRealPath("/WEB-INF/classes/敏感词汇.txt");
             // 2 读取文件

             BufferedReader br = new BufferedReader(new FileReader(realPath));

             // 3 将文件的每一行添加到 list 中

             String line = null;
             while((line =  br.readLine()) != null) {
                 list.add(line);
             }

             br.close();

             System.out.println(list);
         } catch (Exception e) {
             e.printStackTrace();
         }

     }

     public void destroy() {
     }

 }