Spring Boot 中实现一个过滤器相当简单,实现 javax.servlet.Filter 接口即可。

下面以实现一个记录接口访问日志及请求耗时的过滤器为例:

1、定义 ApiAccessFilter 类,并实现 Filter 接口

@Slf4j
@WebFilter(filterName = "ApiAccessFilter", urlPatterns = "/*")
public class ApiAccessFilter implements Filter {
 
 
    @Override
    public void init(FilterConfig filterConfig) {
 
    }
 
    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse,
        FilterChain filterChain) throws IOException, ServletException {
 
        HttpServletRequest request = (HttpServletRequest) servletRequest;
 
        Long requestId = IdGenUtils.nextIdByMem(); // 请求ID,这个是我业务中的id,大家可自行决定是否需要
        long start = System.currentTimeMillis(); // 请求进入时间
 
        log.info("[Api Access] start. id: {}, uri: {}, method: {}, client: {}", requestId,
            request.getRequestURI(), request.getMethod(), getIP(request));
        filterChain.doFilter(servletRequest, servletResponse);
        log.info("[Api Access]   end. id: {}, duration: {}ms", requestId,
            System.currentTimeMillis() - start);
 
    }
 
    @Override
    public void destroy() {
 
    }
 
    /**
     * 获取IP地址
     *
     * @param request 请求
     * @return request发起客户端的IP地址
     */
    private String getIP(HttpServletRequest request) {
        if (request == null) {
            return "0.0.0.0";
        }
 
        String Xip = request.getHeader("X-Real-IP");
        String XFor = request.getHeader("X-Forwarded-For");
 
        String UNKNOWN_IP = "unknown";
        if (StringUtils.isNotEmpty(XFor) && !UNKNOWN_IP.equalsIgnoreCase(XFor)) {
            //多次反向代理后会有多个ip值,第一个ip才是真实ip
            int index = XFor.indexOf(",");
            if (index != -1) {
                return XFor.substring(0, index);
            } else {
                return XFor;
            }
        }
 
        XFor = Xip;
        if (StringUtils.isNotEmpty(XFor) && !UNKNOWN_IP.equalsIgnoreCase(XFor)) {
            return XFor;
        }
 
        if (StringUtils.isBlank(XFor) || UNKNOWN_IP.equalsIgnoreCase(XFor)) {
            XFor = request.getHeader("Proxy-Client-IP");
        }
        if (StringUtils.isBlank(XFor) || UNKNOWN_IP.equalsIgnoreCase(XFor)) {
            XFor = request.getHeader("WL-Proxy-Client-IP");
        }
        if (StringUtils.isBlank(XFor) || UNKNOWN_IP.equalsIgnoreCase(XFor)) {
            XFor = request.getHeader("HTTP_CLIENT_IP");
        }
        if (StringUtils.isBlank(XFor) || UNKNOWN_IP.equalsIgnoreCase(XFor)) {
            XFor = request.getHeader("HTTP_X_FORWARDED_FOR");
        }
        if (StringUtils.isBlank(XFor) || UNKNOWN_IP.equalsIgnoreCase(XFor)) {
            XFor = request.getRemoteAddr();
        }
        return XFor;
    }
}

2、启用该过滤器

springboot 中两种启用过滤器的方式,第一种在 FilterRegistrationBean 中注册该 Filter,第二种,采用注解的方式启用

个人觉得注册方式比较麻烦,所有本例中使用的是第二种。

在 Filter 中添加注解 @WebFilter(filterName = "ApiAccessFilter", urlPatterns = "/"),配置了过滤器名和需要过滤的请求地址,/ 表示过滤所有请求。然后在启动类上,通过 @ServletComponentScan 注解,指明该过滤器即可

@SpringBootApplication
@ServletComponentScan("com.yclouds.service.demo")
public class YCloudsServiceDemoApplication {
 
    public static void main(String[] args) {
        SpringApplication.run(YCloudsServiceDemoApplication.class, args);
    }
}

3、效果展示

ea502fe13df1689b09be52c2581a3ba3.png

发送了两次 http 请求,每次都会打印日志内容,并且统计了整个请求的耗时情况。

GitHub 地址:

https://github.com/ye17186/spring-boot-learn


作者:ye17186

来源链接:

https://blog.csdn.net/ye17186/article/details/88818728

e7c3aeac56a1883e1926896a4d89699d.png

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐