帮我写一个springboot接口防刷的切面实现

怼烎@ 2024-03-26 08:40 43阅读 0赞

好的，我可以为您提供一个示例。这是一个使用Spring AOP的接口防刷的切面实现：

import java.util.concurrent.TimeUnit;
    
    import org.aspectj.lang.ProceedingJoinPoint;
    import org.aspectj.lang.annotation.Around;
    import org.aspectj.lang.annotation.Aspect;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.data.redis.core.RedisTemplate;
    import org.springframework.stereotype.Component;
    
    @Aspect
    @Component
    public class AntiBrushAspect {
    
        @Autowired
        private RedisTemplate<String, Integer> redisTemplate;
    
        @Around("@annotation(AntiBrush)")
        public Object antiBrush(ProceedingJoinPoint joinPoint) throws Throwable {
            // 获取目标方法的参数
            String userId = (String) joinPoint.getArgs()[0];
            // 从Redis中获取访问次数
            Integer count = redisTemplate.opsForValue().get(userId);
            if (count == null) {
                // 如果没有访问过，则初始化访问次数为1
                count = 1;
            } else {
                // 如果已经访问过，则加1
                count++;
            }
            // 如果访问次数超过了限制，则抛出异常
            if (count > 10) {
                throw new Exception("访问次数超过了限制");
            }
            // 将访问次数存入Redis，并设置过期时间为1小时
            redisTemplate.opsForValue().set(userId, count, 1, TimeUnit.HOURS);
            // 执行目标方法
            return joinPoint.proceed();
        }
    }

然后，您可以在需要防刷的接口方法上使用`@AntiBrush`注解，来触发这个切面。

\`\`\`java import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.RestController;

@RestController public class ExampleController \{