共计 1717 个字符,预计需要花费 5 分钟才能阅读完成。
需求:单位时间内,比如 1s,限制某个接口的调用次数。
RateLimiter
RateLimiter 基于令牌桶算法,有一个令牌桶,单位时间内令牌会以恒定的数量(即令牌的加入速度)加入到令牌桶中,所有请求都需要获取令牌才可正常访问。当令牌桶中没有令牌可取的时候,则拒绝请求。
定义一个接口:
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface RateLimit {
// 每秒限制次数
double limit() default 1;}
定义一个切面:
@Slf4j
@Component
@Aspect
public class RateLimitAspect {private static final ConcurrentMap<String, RateLimiter> rateLimiterCache = new ConcurrentHashMap<>();
@Pointcut("@annotation(com.example.demo.RateLimit)")
public void rateLimit() {}
@Around("rateLimit()")
public Object pointcut(ProceedingJoinPoint joinPoint) throws Throwable {
Object obj = null;
MethodSignature signature = (MethodSignature) joinPoint.getSignature();
Method method = signature.getMethod();
RateLimit rateLimit = AnnotationUtils.findAnnotation(method, RateLimit.class);
if (rateLimit != null) {if (rateLimiterCache.get(method.getName()) == null) {rateLimiterCache.put(method.getName(), RateLimiter.create(rateLimit.limit()));
}
RateLimiter rateLimiter = rateLimiterCache.get(method.getName());
if (rateLimiter != null) {if (!rateLimiter.tryAcquire()) {log.info("===== 完了没抢到...=====");
} else {log.info("===== 我抢到了!!=====");
obj = joinPoint.proceed();}
}
}
return obj;
}
}
controller:
@Slf4j
@RestController
public class TestController {@RateLimit(limit = 5)
@GetMapping("/test")
public String test() {return "请求成功";}
}
Redis
利用了 Redis 的自增和过期时间
@Slf4j
@RestController
public class TestController {
private int countLimit = 5;
private int timeLimit = 1;
@Autowired
StringRedisTemplate stringRedisTemplate;
@GetMapping("/test")
public String test(String key) {if (stringRedisTemplate.hasKey(key)) {long count = stringRedisTemplate.boundValueOps(key).increment(1);
if (count > countLimit) {log.info("请求过于繁忙");
}
} else {stringRedisTemplate.opsForValue().set(key, "1", timeLimit, TimeUnit.SECONDS);
}
return "请求成功";
}
}
正文完