【SpringBoot】Spring Boot 实现接口防刷

阳光穿透心脏的1/2处 2023-09-27 12:29 134阅读 0赞

AOP + 自定义注解 + Redis 实现接口防刷

自定义注解：

Documented
    @Target({
        ElementType.METHOD})
    @Retention(RetentionPolicy.RUNTIME)
    public @interface MyPrevent {
        
    
        /**
         * redis 缓存前缀，用来区分是哪个方法(有可能有多个方法被注解)
         */
        String prefix();
    
        /**
         * 过期时间
         */
        int expire() default 1;
    
        /**
         * 过期单位
         */
        TimeUnit timeUnit() default TimeUnit.SECONDS;
    
    }

AOP:

@Aspect
    @Configuration
    public class MyPreventAspect {
        
    
        @Autowired
        private StringRedisTemplate stringRedisTemplate;
    
        @Around("@annotation(com.zzc.hardcore.annotation.MyPrevent)")
        public Object interceptor(ProceedingJoinPoint pjp) {
        
            MethodSignature signature = (MethodSignature) pjp.getSignature();
            Method method = signature.getMethod();
            MyPrevent myPrevent = method.getAnnotation(MyPrevent.class);
    
            // 用户Id 是可以获取得到的
            String userId = "666666";
            String prefix = myPrevent.prefix();
            final String redisKey = prefix + "_" + userId;
            String s = stringRedisTemplate.opsForValue().get(redisKey);
            if (!StringUtils.isEmpty(s)) {
        
                Map<String, Object> result = new HashMap<>(2);
                result.put("code", "500");
                result.put("msg", "系统数据处理中...,请稍后");
                return result;
            }
            stringRedisTemplate.opsForValue().set(redisKey, userId, myPrevent.expire(), myPrevent.timeUnit());
            try {
        
                return pjp.proceed();
            } catch (Throwable throwable) {
        
                throw new RuntimeException("系统异常");
            }
        }
    
    }

控制器测试：

@RequestMapping("/myPrevent")
    @RestController
    public class MyPreventController {
        
    
        @MyPrevent(prefix = "test", expire = 3, timeUnit = TimeUnit.SECONDS)
        @RequestMapping("/test")
        public Object test() {
        
            return "test";
        }
    
    }