package com.gllic.workweixin.config;

import com.gllic.workweixin.filter.XssFilter;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * @program: workweixin
 * @ClassName FilterConfig
 * @description:列印方法體消耗時間日誌
 * @author: Marlo
 * @create: 2021-05-12 13:29
 * @Version 1.0
 **/
@Configuration
public class FilterConfig {

    @Bean
    public FilterRegistrationBean registFilter() {
        FilterRegistrationBean registration = new FilterRegistrationBean();
        registration.setFilter(new XssFilter());
        registration.addUrlPatterns("/*");
        registration.setName("XssFilter");
        registration.setOrder(1);
        return registration;
    }

}

package com.gllic.workweixin.filter;

import com.gllic.workweixin.service.PublicService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import org.springframework.web.context.support.WebApplicationContextUtils;

import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;

/**
 * @program: workweixin-mobile-api-pro
 * @ClassName XssFilter
 * @description:
 * @author: Marlo
 * @create: 2021-07-12 21:20
 * @Version 1.0
 **/
@Slf4j
public class XssFilter implements Filter {

    @Autowired
    PublicService publicService;

    @Override
    public void init(FilterConfig filterConfig) throws ServletException {
        ServletContext sc = filterConfig.getServletContext();
        WebApplicationContext cxt = WebApplicationContextUtils.getWebApplicationContext(sc);
        if(cxt != null) {
            if (publicService == null) {
                publicService = cxt.getBean(PublicService.class);
            }
        }
    }

    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
        long start = System.currentTimeMillis();
        //HttpServletRequest httpServletRequest = ((ServletRequestAttributes)
                //RequestContextHolder.currentRequestAttributes()).getRequest();
        //String userIp = publicService.getRequestIp(httpServletRequest);
        //log.info("請求IP：{}", userIp);
        filterChain.doFilter(servletRequest, servletResponse);
        log.info("Execute cost={}", (System.currentTimeMillis() - start));
    }

    @Override
    public void destroy() {

    }

}