springboot專案如何使用切面記錄使用者操作紀錄檔

2022-11-01 14:03:06

1、引入springboot-aop整合jar

Spring-boot-start-aop

2、application.yml中啟用宣告

#spring設定
spring:
  #切面啟用
  aop:
    proxy-target-class: true
    auto: true

3、自定義一個攔截controller的註解

package cn.annotation;
 
import java.lang.annotation.*;
 
/**
 * Title: SystemControllerLog
 * @date 2018年8月31日
 * @version V1.0
 * Description:  自定義註解，攔截controller
 */
 
@Target({ElementType.PARAMETER, ElementType.METHOD})//作用在引數和方法上
@Retention(RetentionPolicy.RUNTIME)//執行時註解
@Documented//表明這個註解應該被 javadoc工具記錄
public @interface SystemControllerLog {
    String description() default "";
}

4、自定義一個攔截service的註解

package cn.annotation;
 
import java.lang.annotation.*;
 
/**
 * Title: SystemControllerLog
 * @date 2018年8月31日
 * @version V1.0
 * Description:  自定義註解，攔截service
 */
@Target({ElementType.PARAMETER, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface SystemServiceLog {
    String description() default "";
}

5、定義紀錄檔記錄切面

package cn.annotation;
 
import cn.pojo.Action;
import cn.pojo.User;
import cn.service.ActionService;
import cn.utils.IpUtils;
import cn.utils.JsonUtils;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.AfterThrowing;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
 
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import java.lang.reflect.Method;
import java.util.Date;
 
/**
 * Title: SystemControllerLog
 * @date 2018年8月31日
 * @version V1.0
 * Description: 切點類
 */
@Aspect
@Component
@SuppressWarnings("all")
public class SystemLogAspect {
    //注入Service用於把紀錄檔儲存資料庫，實際專案入庫採用佇列做非同步
    @Resource
    private ActionService actionService;
    //本地異常紀錄檔記錄物件
    private static final Logger logger = LoggerFactory.getLogger(SystemLogAspect.class);
    //Service層切點
    @Pointcut("@annotation(cn.annotation.SystemServiceLog)")
    public void serviceAspect(){
    }
 
    //Controller層切點
    @Pointcut("@annotation(cn.oa.annotation.SystemControllerLog)")
    public void controllerAspect(){
    }
 
    /**
     * @Description  前置通知  用於攔截Controller層記錄使用者的操作
     * @date 2018年9月3日 10:38
     */
 
    @Before("controllerAspect()")
    public void doBefore(JoinPoint joinPoint){
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        HttpSession session = request.getSession();
        //讀取session中的使用者
        User user = (User) session.getAttribute("user");
 
        String ip = IpUtils.getIpAddr(request);
 
        try {
            //*========控制檯輸出=========*//
            System.out.println("==============前置通知開始==============");
            System.out.println("請求方法" + (joinPoint.getTarget().getClass().getName() + "." + joinPoint.getSignature().getName()));
            System.out.println("方法描述：" + getControllerMethodDescription(joinPoint));
            System.out.println("請求人："+user.getUsername());
            System.out.println("請求ip："+ip);
 
            //*========資料庫紀錄檔=========*//
            Action action = new Action();
            action.setActionDes(getControllerMethodDescription(joinPoint));
            action.setActionType("0");
            action.setActionIp(ip);
            action.setUserId(user.getId());
            action.setActionTime(new Date());
            //儲存資料庫
            actionService.add(action);
 
        }catch (Exception e){
            //記錄本地異常紀錄檔
            logger.error("==前置通知異常==");
            logger.error("異常資訊：{}",e.getMessage());
        }
    }
 
    /**
     * @Description  異常通知 用於攔截service層記錄異常紀錄檔
     * @date 2018年9月3日 下午5:43
     */
    @AfterThrowing(pointcut = "serviceAspect()",throwing = "e")
    public void doAfterThrowing(JoinPoint joinPoint,Throwable e){
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        HttpSession session = request.getSession();
        //讀取session中的使用者
        User user = (User) session.getAttribute("user");
        //獲取請求ip
        String ip = IpUtils.getIpAddr(request);
        //獲取使用者請求方法的引數並序列化為JSON格式字串
        String params = "";
        if (joinPoint.getArgs()!=null&&joinPoint.getArgs().length>0){
            for (int i = 0; i < joinPoint.getArgs().length; i++) {
                params+= JsonUtils.objectToJson(joinPoint.getArgs()[i])+";";
            }
        }
        try{
            /*========控制檯輸出=========*/
            System.out.println("=====異常通知開始=====");
            System.out.println("異常程式碼:" + e.getClass().getName());
            System.out.println("異常資訊:" + e.getMessage());
            System.out.println("異常方法:" + (joinPoint.getTarget().getClass().getName() + "." + joinPoint.getSignature().getName() + "()"));
            System.out.println("方法描述:" + getServiceMethodDescription(joinPoint));
            System.out.println("請求人:" + user.getUsername());
            System.out.println("請求IP:" + ip);
            System.out.println("請求引數:" + params);
            /*==========資料庫紀錄檔=========*/
            Action action = new Action();
            action.setActionDes(getServiceMethodDescription(joinPoint));
            action.setActionType("1");
            action.setUserId(user.getId());
            action.setActionIp(ip);
            action.setActionTime(new Date());
            //儲存到資料庫
            actionService.add(action);
        }catch (Exception ex){
            //記錄本地異常紀錄檔
            logger.error("==異常通知異常==");
            logger.error("異常資訊:{}", ex.getMessage());
        }
    }
 
 
    /**
     * @Description  獲取註解中對方法的描述資訊 用於service層註解
     * @date 2018年9月3日 下午5:05
     */
    public static String getServiceMethodDescription(JoinPoint joinPoint)throws Exception{
        String targetName = joinPoint.getTarget().getClass().getName();
        String methodName = joinPoint.getSignature().getName();
        Object[] arguments = joinPoint.getArgs();
		Class targetClass = Class.forName(targetName);
        Method[] methods = targetClass.getMethods();
        String description = "";
        for (Method method:methods) {
            if (method.getName().equals(methodName)){
                Class[] clazzs = method.getParameterTypes();
                if (clazzs.length==arguments.length){
                    description = method.getAnnotation(SystemServiceLog.class).description();
                    break;
                }
            }
        }
        return description;
    }
 
 
 
    /**
     * @Description  獲取註解中對方法的描述資訊 用於Controller層註解
     * @date 2018年9月3日 上午12:01
     */
    public static String getControllerMethodDescription(JoinPoint joinPoint) throws Exception {
        String targetName = joinPoint.getTarget().getClass().getName();
        String methodName = joinPoint.getSignature().getName();//目標方法名
        Object[] arguments = joinPoint.getArgs();
        Class targetClass = Class.forName(targetName);
        Method[] methods = targetClass.getMethods();
        String description = "";
        for (Method method:methods) {
            if (method.getName().equals(methodName)){
                Class[] clazzs = method.getParameterTypes();
                if (clazzs.length==arguments.length){
                    description = method.getAnnotation(SystemControllerLog.class).description();
                    break;
                }
            }
        }
        return description;
    }
}