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 LimitAccess {

    /**
     * 限制訪問(wèn)的key
     * @return
     */
    String key();

    /**
     * 限制訪問(wèn)次數(shù)
     * @return
     */
    int times();

    /**
     * 時(shí)間段
     * @return
     */
    int duration();
}

import com.hezy.annotation.LimitAccess;
import lombok.extern.log4j.Log4j2;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import java.util.concurrent.TimeUnit;

/**
 * AOP類（通知類）
 */
@Component
@Aspect
@Log4j2
public class LimitAspect {

    @Value("${access.enable:false}")
    private boolean enable;

    @Autowired
    private RedisTemplate redisTemplate;

    @Pointcut("@annotation(com.hezy.annotation.LimitAccess)")
    public void pt(){};

    @Around("pt()")
    public Object aopAround(ProceedingJoinPoint pjp) throws Throwable {
        // 設(shè)置一個(gè)開關(guān)，控制是否執(zhí)行
        if (!enable) {
            return pjp.proceed();
        }
        // 獲取切入點(diǎn)上面的自定義注解
        Signature signature = pjp.getSignature();
        MethodSignature methodSignature = (MethodSignature) signature;

        // 獲取方法上面的注解
        LimitAccess limitAccess = methodSignature.getMethod().getAnnotation(LimitAccess.class);
        // 獲取注解上面的屬性值
        int limit = limitAccess.times();
        String key = limitAccess.key();
        int duration = limitAccess.duration();

        // 遞增鍵的值，如果鍵不存在則初始化為1
        Long currentCount = redisTemplate.opsForValue().increment(key, 1);

        // 如果鍵是新創(chuàng)建的，設(shè)置過(guò)期時(shí)間
        if (currentCount != null && currentCount == 1) {
            redisTemplate.expire(key, duration, TimeUnit.SECONDS);
        }

        // 檢查是否超過(guò)限制
        if (currentCount != null && currentCount > limit) {
            log.info("訪問(wèn)過(guò)于頻繁: " + pjp.toLongString());
            throw new RuntimeException("訪問(wèn)過(guò)于頻繁");
        }
        return pjp.proceed();
    }
}

    @LimitAccess(key = "test", times = 3, duration = 1)
    @GetMapping
    public String test() {
        return demoService.test();
    }

access:
  enable: true

    @LimitAccess(key = "deleteUserById", times = 3, duration = 1)
    @Transactional
    @Override
    public void deleteUserById(Integer id) {
        // 刪除用戶
        userMapper.deleteUserById(id);
        
        int i = 1 / 0;

        // 刪除用戶對(duì)應(yīng)的角色
        userMapper.deleteUserRoleMapper(id);
    }

	String key = limitAccess.key() + 用戶標(biāo)識(shí)（用戶名、用戶id）;