Filter实现全站违法关键词屏蔽

思路:客户端请求服务器数据,经过Filter过滤(请求放行,响应拦截),服务器向客户端返回数据时,在Filter中修改掉返回数据中违法的部分。

修改服务器的响应需要自定义一个HttpServletResponse的包装类,继承自HttpServletResponseWrapper即可实现自己的HttpServletResponse的包装类。

大概是这样的:

//package com.ecshop.tools;
import java.io.CharArrayWriter;
import java.io.InputStream;
import java.io.PrintWriter;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;
public class ShieldKeywordResponseWrapper extends HttpServletResponseWrapper{
private CharArrayWriter caw;
private PrintWriter pw;
public ShieldKeywordResponseWrapper(HttpServletResponse response) {
super(response);
// 这个是我们保存返回结果的地方
caw = new CharArrayWriter();
// 这个是包装PrintWriter的,让所有结果通过这个PrintWriter写入到caw中
pw = new PrintWriter(caw);
}
@Override
public PrintWriter getWriter() {
return pw;
}
public String getResult() {
return caw.toString();
}
}

因为违法关键词都包含在响应文本消息中,服务器向请求客户端响应文本消息必须获取HttpServletResponse对象中的PrintWriter文本输出流对象进行输出消息,所以我们的这个封装类覆盖HttpServletResponse的getWriter是很重要的一步,我这个封装类的意思是让服务器响应数据时使用我们修改过的PrintWriter文本输出流,输出的数据并不立刻响应给客户,而是把输出的数据都保留在了CharArrayWriter这个对象中,当控制层执行完毕我们就可以在Filter中进行获取CharArrayWriter对象中的数据进行修改了,修改完毕我们在返回给你客户,就是我们想要的效果了。(而默认HttpServletResponse的PrintWriter对象是由tomcat自己所扩展自PrintWriter的CoyoteWriter对象,CoyoteWriter对象所打印的数据都保留在OutputBuffer对象(也是tomcat自己所扩展自Writer的一个对象)中,我们不方便获取它的文本数据,所以我们修改掉它,当OutputBuffer执行flush就发送消息,或者不执行让servlet结束后自动发送消息)。

以上是我看了一点tomcat源码然后瞎推理,对不对无所谓。。

当我们HttpServletResponse的包装类弄完剩下的就好办了。

//我们看下过滤器中的doFilter函数部分代码就足够了
public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
HttpServletResponse res= (HttpServletResponse) response;
   //我们刚刚做的包装类在这里
ShieldKeywordResponseWrapper skrw= new ShieldKeywordResponseWrapper(res);
chain.doFilter(request, skrw);
//在servlet处理完毕后我们开始做处理
String result = skrw.getResult();
//替换关键词为*
Properties p = new Properties();
p.load(ShieldKeywords.class.getResourceAsStream("/keywords.properties"));
Set<Entry<Object, Object>> se= p.entrySet();
for(Entry<Object, Object> e:se){
result = result.replace(String.valueOf(e.getKey()), String.valueOf(e.getValue()));
}
// 输出最终的结果
PrintWriter out = response.getWriter();
out.write(result);
//以下两句无作用,CharArrayWriter中为空函数,tomcat自己提供的OutputBuffer执行这两句才有作用
out.flush();
out.close();
}

最后看下keywords.properties文件的内容:

违法词1=*
违法词2=*

建议web.xml中过滤器配置时只映射响应文本数据的请求url,因有的请求是要获取一个文件流,服务器HttpServletResponse响应时文本输出流和字节流是冲突的,正常情况下两者只可取一,不然会报错

上一篇:Webupload + MVC 之上传下载


下一篇:用H5的canvas做时钟