責任鏈設計模式模擬Java Web過濾器的實現

 

Tomcat容器以及一些Web框架都使用了責任鏈設計模式來對請求和響應進行逐步的處理或者過濾，本文使用Java語言來模擬過濾器的實現。其序列圖以下：

 

若是簡單的使用責任鏈模式，用一個List來實現層層過濾，則會致使request的處理是按照順序來執行，response也會按照順序來執行，這就與咱們指望的response按照倒序來執行相悖。所以要用一種方式實現request順序處理，response倒序處理。

實現

類圖大體以下：

 

首先，寫兩個實體類，Request和Response

Request.java

[java] view plain copy

1. public class Request {  
2.     private String requestStr;  
3.   
4.     public String getRequestStr() {  
5.         return requestStr;  
6.     }  
7.   
8.     public void setRequestStr(String requestStr) {  
9.         this.requestStr = requestStr;  
10.     }  
11. }  

Response.java[java] view plain copy

1. public class Response {  
2.     private String responseStr;  
3.   
4.     public String getResponseStr() {  
5.         return responseStr;  
6.     }  
7.   
8.     public void setResponseStr(String responseStr) {  
9.         this.responseStr = responseStr;  
10.     }  
11. }  

 

面向接口編程，編寫過濾器接口。

Filter.java

[java] view plain copy

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

這裏每次進行過濾後，都將過濾器鏈指向下一個過濾器，進行過濾，執行完後，再逐層返回，有點相似於遞歸。

 

編寫過濾器鏈

FilterChain.java

[java] view plain copy

1. import java.util.ArrayList;  
2. import java.util.List;  
3.   
4. public class FilterChain implements Filter{  
5.     private List<Filter> filters = new ArrayList<>();  
6.     private int index = 0;  
7.       
8.     public FilterChain addFilter(Filter f){  
9.         this.filters.add(f);  
10.         return this;  
11.     }  
12.   
13.     @Override  
14.     public void doFilter(Request request, Response response, FilterChain chain) {  
15.         if(index == filters.size()) return;  
16.           
17.         Filter f = filters.get(index);  
18.         index ++;  
19.         f.doFilter(request, response, chain);  
20.     }  
21. }  

該類中進行了兩個操做，一個是過濾器的添加，一個是每一次過濾完成後，將索引加一後，再到下一個過濾器進行過濾，當走到盡頭，再逐層返回。這有點相似於雙向鏈表，先從頭指針日後走，到了盡頭再往前走。

 

編寫兩個過濾器類

HtmlFilter.java

[java] view plain copy

1. public class HtmlFilter implements Filter {  
2.     @Override  
3.     public void doFilter(Request request, Response response, FilterChain chain) {  
4.         request.setRequestStr(request.getRequestStr().replace('<', '[').replace('>', ']') + "---HtmlFilter");  
5.         chain.doFilter(request, response, chain);  
6.         response.setResponseStr(response.getResponseStr() + "---HtmlFilter");  
7.     }  
8. }  

將尖括號替換爲中括號。

 

SeansitiveFilter.java

[java] view plain copy

1. public class SensitiveFilter implements Filter {  
2.     @Override  
3.     public void doFilter(Request request, Response response,FilterChain chain) {  
4.         request.setRequestStr(request.getRequestStr().replace("被就業", "就業").replace("敏感", "") + "---SensitiveFilter");  
5.         chain.doFilter(request, response, chain);  
6.         response.setResponseStr(response.getResponseStr() + "---SensitiveFilter");  
7.     }  
8. }  

去掉敏感詞彙

 

最後編寫測試用例

Main.java

[java] view plain copy

1. public class Main {  
2.     public static void main(String[] args) {  
3.         String msg = "你們好:),<script>,敏感,被就業";  
4.         Request request = new Request();  
5.         request.setRequestStr(msg);  
6.         Response response = new Response();  
7.         response.setResponseStr("response");  
8.           
9.         FilterChain fc = new FilterChain();  
10.         fc.addFilter(new HtmlFilter()).addFilter(new SensitiveFilter());  
11.         fc.doFilter(request, response, fc);  
12.         System.out.println(request.getRequestStr());  
13.         System.out.println(response.getResponseStr());  
14.     }  
15. }  

最終輸出的結果爲：[java] view plain copy

1. 你們好:),[script],,就業---HtmlFilter---SensitiveFilter  
2. response---SensitiveFilter---HtmlFilter  

 

若是由於某一緣由不進行接下來的過濾，則在FilterChain中加一個判斷條件，在第幾個過濾器不知足條件時，則不執行下一個過濾器的過濾，直接return便可，這樣就直接返回了，response不會再進行相應過濾器的加工處理。