Java模擬FilterChain的實現 (Chain Of Responsibility模式)

　　有這樣一個應用場景，對於傳入后台的Request和返回頁面的Response信息，如果頁面的Request帶有敏感信息或危險字符串，就會對系統造成影響，所以我們會寫過濾器(Filter)處理Request，不同的信息處理會有不同的處理方式，如果我們只用一個Filter來處理所有的信息，擴展性就不夠強，所以會針對不同的信息處理寫不同的Filter，然后把Filter放在一個FilterChain中，以實現信息處理的“熱拔插”，這樣一個FilterChain就像一條職責鏈，鏈上的每個節點處理不同的功能，同時對於Response我們希望能以Resuest處理的倒序進行返回處理，以達到完全過濾敏感信息的效果，功能圖如下所示。

　　AFilter到DFilter構成了一個FilterChain。

　　現在來模擬這個Servlet中的FilterChain。

　　首先定義接口Filter,因為所有的Filter必須遵循這個規范，才能裝載在FilterChain中。

　　

public interface Filter {
    void doFilter(Request request,Response response,FilterChain chain);
}

 

　　接下來，編寫FilterChain以及各個Filter。

　　

public class FilterChain implements Filter{
    List<Filter> fs = new ArrayList<Filter>();
    int index = 0;
    
    
    public FilterChain addFilter(Filter f) {
        fs.add(f);
        return this;
    }
    

    @Override
    public void doFilter(Request request, Response response,FilterChain chain) {
        if(index == fs.size()) return;
        Filter f = fs.get(index);
        index ++;
        f.doFilter(request, response, chain);
    }
    
    
}

public class HTMLFilter implements Filter {


    @Override
    public void doFilter(Request request, Response response,FilterChain chain) {
        request.requestStr = request.requestStr.replace("<", "[").replace(">", "]") + "--HTML--";
        chain.doFilter(request, response, chain);
        response.responseStr += "--HTML--";
    }
}

public class SensitiveFilter implements Filter {


    @Override
    public void doFilter(Request request, Response response,FilterChain chain) {
        request.requestStr = request.requestStr.replace("被就業", "就業").replace("敏感", "**") + "--Sensitive--";
        chain.doFilter(request, response, chain);
        response.responseStr += "--Sensitive--";
    }

    
}

　　最后測試類。

　　

public class Test {

    /**
     * @param args
     */
    public static void main(String[] args) {
        String msg = "測試，<script>，被就業，敏感信息";
        FilterChain fc = new FilterChain();
        fc.addFilter(new HTMLFilter()).addFilter(new SensitiveFilter());
        Request request = new Request();
        request.setRequestStr(msg);
        Response response = new Response();
        response.setResponseStr("response");
        fc.doFilter(request, response,fc);
        System.out.println(request.getRequestStr());
        System.out.println(response.getResponseStr());
    }

}

 

　　測試結果如下（箭頭為Filter走向）: