自用过滤器代码

自用过滤器代码

import javax.servlet.*;
import javax.servlet.annotation.WebFilter;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import java.io.IOException;
import java.util.Objects;

@WebFilter(urlPatterns = "/*")
public class MyFilter implements Filter {

    @Override
    public void init(FilterConfig filterConfig) throws ServletException {
        // 初始化方法，可以在这里做一些初始化的操作
        System.out.println("MyFilter is initializing...");
    }

    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain)
            throws IOException, ServletException {
        // 实际的过滤逻辑在这里
        HttpServletRequest request = (HttpServletRequest) servletRequest;
        // 可以在这里对请求进行预处理
        //记录IP访问次数
        String ip = request.getRemoteAddr();
        //统计网文次数并输出
        HttpSession session = request.getSession();
        Integer count = (Integer) session.getAttribute("count");
        count = Objects.isNull(count)?1:++count;
        System.out.println("ip"+ip+"\tcount:"+count);
        session.setAttribute("count",count);
        // 继续处理请求链，方行
        filterChain.doFilter(servletRequest, servletResponse);

    }

    @Override
    public void destroy() {
        // 销毁方法，过滤器实例销毁时调用
        System.out.println("MyFilter is destroying...");
    }
}

对应的controller

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class FilterController {
    @GetMapping("/filter")
    public String index(){return "hello,filter";}

}

对应的config

import com.yase.system.filter.MyFilter;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class FilterConfig {

    @Bean
    public FilterRegistrationBean<MyFilter> myFilterRegistration() {
        FilterRegistrationBean<MyFilter> registration = new FilterRegistrationBean<>();
        registration.setFilter(new MyFilter());
        registration.addUrlPatterns("/*"); // 设置过滤器的URL模式
        return registration;
    }
}