SpringBoot自定义注解+AOP+自定义异常

本文来源于需要在多个方法中首先判断某个数据是否存在，存在就进行各自的操作，不存在就不执行各自的方法。

1. 编写自定义注解

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface SnValidation {
}

2. Aop切面类

import com.thrid.thrid_sdk.dto.req.SetModeRequest;
import com.thrid.thrid_sdk.exception.SnValidationException;
import com.thrid.thrid_sdk.service.DeviceService;
import lombok.RequiredArgsConstructor;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.springframework.stereotype.Component;

@Aspect
@Component
@RequiredArgsConstructor
public class SnValidationAspect {
    private final PersonService personService;

    @Before("@annotation(com.thrid.thrid_sdk.annotation.SnValidation)")//指定哪个注解
    public void validateSn(JoinPoint joinPoint) {//使用该注解的具体逻辑
        Object[] args = joinPoint.getArgs();
        for (Object arg : args) {
            if (arg instanceof Person) {
                Person person = (Person) arg;
                boolean isExist = personService.isExists(person.gteId());
                if (!isExist) {
                    throw new SnValidationException("用户不存在");
                }
            }
        }
    }
}

3. 方法上使用自定义注解

这个注解加在需要做判断的任何方法上

   @Override
    @SnValidation
    public String setDisChargeTimes(DisChargeTimeReq disChargeTimeReq, HttpServletRequest request) {
          //方法体
    }

4. 步骤2中的自定义异常SnValidationException

/**
 * @Author luoyi
 */
public class SnValidationException extends RuntimeException{
    public SnValidationException(String message) {
        super(message);
    }
}

这样就完成了这个需求。