package com.zou.metabox.common.aspect;

import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.*;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.Map;
import java.util.stream.IntStream;

/**
 * @author BIGSHU0923
 * @description com.zou.metabox 中Controller層的的日志切面
 * @since 7/30/2023  5:32 PM
 */
@Aspect
@Component
@Slf4j
public class LoggingAspect {
    /**
     * com.zou.metabox.controller 包中公共方法的切入點(diǎn)
     */
    @Pointcut("execution(public * com.zou.metabox.controller.*.*(..))")
    public void loggingPointcut(){
        // 暫不用處理
    }

    /**
     在日志切入點(diǎn)之前執(zhí)行的通知。
     記錄方法名稱和請(qǐng)求參數(shù)。
     @param joinPoint 控制器類(lèi)中的連接點(diǎn)
     */
    @Before("loggingPointcut()")
    public void logBefore(JoinPoint joinPoint){
        Object[] args = joinPoint.getArgs();
        MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
        String[] parameterNames = methodSignature.getParameterNames();
        Map<String, Object> requestMap = new HashMap<>();
        IntStream.range(0, parameterNames.length)
                .forEach(i -> requestMap.put(parameterNames[i], args[i]));
        log.info("Before method:{}|{}", joinPoint.getSignature().getName(), requestMap);
    }

    /**
     在日志切入點(diǎn)之后和方法成功執(zhí)行后執(zhí)行的通知。
     記錄方法名稱和返回結(jié)果。
     @param joinPoint 控制器類(lèi)中的連接點(diǎn)
     @param result 方法的返回結(jié)果
     */
    @AfterReturning(pointcut = "execution(public * com.zou.metabox.controller.*.*(..))", returning = "result")
    public void logResponse(JoinPoint joinPoint, Object result){
        log.info("Method response:{}|{}", joinPoint.getSignature().getName(), result.toString());
    }

}

    @Around("loggingPointcut()")
    public Object around(ProceedingJoinPoint pjp) throws Throwable{
        // 獲取類(lèi)名
        String className = pjp.getTarget().getClass().getTypeName();
        // 獲取方法名
        String methodName = pjp.getSignature().getName();
        // 獲取參數(shù)名
        String[] parameterNames = ((MethodSignature) pjp.getSignature()).getParameterNames();

        Object result = null;
        // 獲取參數(shù)值
        Object[] args = pjp.getArgs();

        // 獲取請(qǐng)求
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        // 獲取請(qǐng)求的url
        String url = request.getRequestURL().toString();

        // 請(qǐng)求參數(shù)，以參數(shù)名和值為鍵值對(duì)
        Map<String, Object> paramMap = new HashMap<>();
        IntStream.range(0, parameterNames.length).forEach(i->paramMap.put(parameterNames[i], args[i]));

        // header參數(shù)
        Enumeration<String> headerNames = request.getHeaderNames();
        Map<String, Object> headerMap = new HashMap<>();
        while (headerNames.hasMoreElements()){
            String headerName = headerNames.nextElement();
            String headerValue = request.getHeader(headerName);
            headerMap.put(headerName, headerValue);
        }

        // 打印請(qǐng)求參數(shù),記錄起始時(shí)間
        long start = System.currentTimeMillis();
        log.info("請(qǐng)求| 請(qǐng)求接口：{} | 類(lèi)名：{} | 方法：{} | header參數(shù)：{} | 參數(shù)：{} | 請(qǐng)求時(shí)間：{}", url, className, methodName, headerMap, paramMap, LocalDateTime.now());

        try {
            result = pjp.proceed();
        } catch (Exception e) {
            log.error("返回| 處理時(shí)間:{} 毫秒 | 返回結(jié)果 :{}", (System.currentTimeMillis() - start), "failed");
            throw e;
        }

        // 獲取執(zhí)行完的時(shí)間 打印返回報(bào)文
        log.info("返回| 處理時(shí)間:{} 毫秒 | 返回結(jié)果 :{}", (System.currentTimeMillis() - start), "success");
        return result;
    }