package com.fioc.ect.fproject.web.interceptors;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.core.NamedThreadLocal;

import org.springframework.web.servlet.HandlerInterceptor;

import org.springframework.web.servlet.ModelAndView;

import com.fioc.ect.fproject.utils.DateUtil;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import java.text.SimpleDateFormat;

/**

* @Author wulei

* @Title: LogInterceptor

* @Description: 日志拦截器, 用于记录请求的时间

* @Date 2018/10/25 13:14

*/

public class LogInterceptor implements HandlerInterceptor {

    /**

    * logger

    */

    private static final Logger logger = LoggerFactory.getLogger(LogInterceptor.class);

    private static final ThreadLocal<Long> startTimeThreadLocal = new NamedThreadLocal<>("ThreadLocal StartTime");

    @Override

    public boolean preHandle(HttpServletRequest request, HttpServletResponse response,

                            Object handler) {

        if (logger.isDebugEnabled()) {

            // 1、开始时间

            long beginTime = System.currentTimeMillis();

            // 线程绑定变量（该数据只有当前请求的线程可见）

            startTimeThreadLocal.set(beginTime);

            logger.debug("开始计时: {}  URI: {}", new SimpleDateFormat("hh:mm:ss.SSS")

                    .format(beginTime), request.getRequestURI());

        }

        return true;

    }

    @Override

    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler,

                          ModelAndView modelAndView) {

    }

    @Override

    public void afterCompletion(HttpServletRequest request, HttpServletResponse response,

                                Object handler, Exception ex) {

        // 打印JVM信息。

        if (logger.isDebugEnabled()) {

            // 得到线程绑定的局部变量（开始时间）

            long beginTime = startTimeThreadLocal.get();

            // 2、结束时间

            long endTime = System.currentTimeMillis();

            logger.debug("计时结束：{}  耗时：{}  URI: {}  最大内存: {}m  已分配内存: {}m  已分配内存中的剩余空间: {}m  最大可用内存: {}m",

                    new SimpleDateFormat("hh:mm:ss.SSS").format(endTime), DateUtil.formatDateTime(endTime - beginTime),

                    request.getRequestURI(), Runtime.getRuntime().maxMemory() / 1024 / 1024,

                    Runtime.getRuntime().totalMemory() / 1024 / 1024, Runtime.getRuntime().freeMemory() / 1024 / 1024,

                    (Runtime.getRuntime().maxMemory() - Runtime.getRuntime().totalMemory() + Runtime.getRuntime().freeMemory()) / 1024 / 1024);

        }

    }

}