SpringBoot拦截所有请求,记录日志表

1.创建一个类并实现HandlerInterceptor 拦截器

import org.springframework.core.env.Environment;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.resource.ResourceHttpRequestHandler;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.net.InetAddress;
import java.util.concurrent.Executor;


public class RequestInterceptor implements HandlerInterceptor {

    private static final String START_TIME = "start_time";
    private static final String REQUEST_LOG = "request_log";

    //环境
    private Environment env;
    private JdbcTemplate jdbcTemplate;
    private Executor executor;

    public RequestInterceptor(Environment env, JdbcTemplate jdbcTemplate, Executor executor) {
        this.env = env;
        this.jdbcTemplate = jdbcTemplate;
        this.executor = executor;
    }

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
            throws Exception {
        try {
            if (handler instanceof ResourceHttpRequestHandler) {
                // 处理静态资源的请求
                return true;
            }
            //其他请求
            RequestLog log = new RequestLog();
            log.setFunctionalModule(env.getProperty("spring.application.name"));
            log.setUrl(request.getRequestURI());
            log.setIpAddres(InetAddress.getLocalHost().getHostAddress());
            log.setNativeIp(IpGetNativeUtils.getIpAddress());
            long startTime = System.currentTimeMillis();
            request.setAttribute(START_TIME, startTime);
            request.setAttribute(REQUEST_LOG, log);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            return true;
        }
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler,
                           ModelAndView modelAndView) throws Exception {

    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler,
                                Exception ex) throws Exception {
        try {
            if (handler instanceof ResourceHttpRequestHandler) {
                // 处理静态资源的请求
            } else {
                // 处理其他请求
                long startTime = (Long) request.getAttribute(START_TIME);
                RequestLog log = (RequestLog) request.getAttribute(REQUEST_LOG);
                long endTime = System.currentTimeMillis();
                long executeTime = endTime - startTime;
                log.setResponseTime(executeTime + "");
                //异步保存请求日志记录
                executor.execute(() -> {
                    jdbcTemplate.update(RequestDayStaticSQL.SAVE_SQL,
                            log.getIpAddres(),
                            log.getFunctionalModule(),
                            log.getUrl(),
                            log.getResponseTime(),
                            log.getNativeIp()
                    );
                });

            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

这里通过Environment对象来获取自己配置文件中的内容,通过Executor来进行异步执行,通过jdbc模板类来进行数据存储落库

2,注册拦截器

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.web.servlet.config.annotation.InterceptorRegistration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

import javax.annotation.Resource;
import java.util.concurrent.Executor;

@Configuration
public class AppConfig implements WebMvcConfigurer {

    @Autowired
    private Environment env;
    @Autowired
    private JdbcTemplate jdbcTemplate;

    @Resource(name = "requestLogThreadPool")
    private Executor taskExecutor;

    @Value("${requestLog.onState:false}")
    private Boolean requestLogOnState;


    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        if (requestLogOnState) {
            InterceptorRegistration interceptorRegistration = registry.addInterceptor(new RequestInterceptor(env, jdbcTemplate, taskExecutor));
            interceptorRegistration.addPathPatterns("/**");
            interceptorRegistration.excludePathPatterns("/static/**");
        }
    }
}

requestLogOnState是否开启该拦截器,此处是在配置文件中设置的开关,根据需求,也可以把开关放在其他位置