springboot 自定义注解记录控制器执行时间（HandlerInterceptor实现）

待我称王封你为后i 2023-02-26 15:24 7阅读 0赞

--------------------

### springboot 自定义注解记录控制器执行时间（HandlerInterceptor实现） ###

### \*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\* ###

**示例**

**\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\***

**myannotation 层**

**@LogRecord：自定义注解记录控制器执行时间**

@Documented
    @Target(ElementType.METHOD)
    @Retention(RetentionPolicy.RUNTIME)
    public @interface LogRecord {
    
        String methodDesc() default "";
    }

**\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\***

**interceptor 层**

**CustomInterceptor**

@Component
    public class CustomInterceptor implements HandlerInterceptor {
    
        private Logger logger= LoggerFactory.getLogger(CustomInterceptor.class.getName());
    
        private ThreadLocal<Long> threadLocal=new ThreadLocal<>();
    
        @Override
        public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
            HandlerMethod handlerMethod=(HandlerMethod)handler;
            Method method=handlerMethod.getMethod();
    
            LogRecord logRecord=method.getAnnotation(LogRecord.class);
            if (logRecord!=null){
                threadLocal.set(System.currentTimeMillis());
            }
    
            return true;
        }
    
        @Override
        public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
            HandlerMethod handlerMethod=(HandlerMethod)handler;
            Method method=handlerMethod.getMethod();
    
            LogRecord logRecord=method.getAnnotation(LogRecord.class);
            if (logRecord!=null){
                long startTime=threadLocal.get();
                long endTime=System.currentTimeMillis();
                long costTime=endTime-startTime;
    
                String requestUri=request.getRequestURI();
                String methodName=method.getDeclaringClass().getName()+"."+method.getName();
                String methodDesc=logRecord.methodDesc();
    
                logger.info("requestUri({})  methodName({})  methodDesc({}) ==> 花费时间：{}毫秒",requestUri,methodName,methodDesc,costTime);
            }
        }
    }

**\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\***

**config 层**

**WebConfig**

@Configuration
    public class WebConfig implements WebMvcConfigurer {
    
        @Resource
        private CustomInterceptor customInterceptor;
    
        @Override
        public void addInterceptors(InterceptorRegistry registry) {
            registry.addInterceptor(customInterceptor);
        }
    }

**\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\***

**controller 层**

**HelloController**

@RestController
    public class HelloController {
    
        @LogRecord(methodDesc = "hello")
        @RequestMapping("/hello")
        public String hello() throws Exception{
            System.out.println("hello");
            TimeUnit.SECONDS.sleep(4);
    
            return "hello";
        }
    }

### \*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\* ###

**使用测试**

**localhost:8080/hello**

2020-07-16 10:54:41.574  INFO 2056 --- [nio-8080-exec-1] o.s.web.servlet.DispatcherServlet        : Completed initialization in 5 ms
    hello
    2020-07-16 10:54:45.642  INFO 2056 --- [nio-8080-exec-1] c.e.demo.interceptor.CustomInterceptor   : requestUri(/hello)  methodName(com.example.demo.controller.HelloController.hello)  methodDesc(hello) ==> 花费时间：4053毫秒

**/hello 方法执行时间 4053毫秒**