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() {
    }

}