package cn.webwheel.tutorials.filter;

import cn.webwheel.Filter;
import cn.webwheel.FilterChain;

import javax.servlet.http.HttpServletRequest;

/**
 * 點擊數統計過濾器，實現cn.webwheel.Filter接口
 * 概念與api都類似與servlet filter，并且可以做到更加靈活，通過代碼配置指定其起作用的動作類
 */
public class CountFilter implements Filter {

    private int count;

    // 初始化操作，當Main.init方法調用之后執行
    public void init() {
        System.out.println("CountFilter啟動");
    }

    public boolean process(FilterChain chain) throws Exception {
        // 獲得當前的request對象，并設置count屬性為當前的訪問計數
        HttpServletRequest request = chain.getWebContext().getRequest();
        request.setAttribute("count", ++count);
        // 繼續執行
        return chain.go();
    }

    // 過濾器銷毀操作
    public void destroy() {
        System.out.println("CountFilter停止");
    }
}