使用策略模式实现 Spring 分布式和单机限流

编码行者

已于 2024-04-17 17:05:52 修改

阅读量720

点赞数 5

CC 4.0 BY-SA版权

文章标签：策略模式 spring 分布式

于 2024-04-17 17:02:55 首次发布

本文链接：https://blog.csdn.net/mbh12333/article/details/137880117

本文介绍了如何使用策略模式实现单机和分布式限流，通过RateLimitStrategy接口解耦算法和策略，提高代码的可扩展性和可维护性。AOP切面根据配置动态选择限流策略，简化了使用和未来的扩展.

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我们可以使用策略模式来统一单机限流和分布式限流的实现,提高代码的可扩展性和可维护性。

思路是定义一个 RateLimitStrategy 接口,并分别实现单机限流策略 LocalRateLimitStrategy 和分布式限流策略 DistributedRateLimitStrategy。在 AOP 切面中,根据配置决定使用哪种限流策略。

定义策略接口

public interface RateLimitStrategy {
   
   
    boolean tryAcquire(String key, double qps, long timeout, TimeUnit timeUnit);
}

实现单机限流策略

import com.google.common.util.concurrent.RateLimiter;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.TimeUnit;

public class LocalRateLimitStrategy implements RateLimitStrategy {
   
   

    private final Map<String, RateLimiter> rateLimiters = new ConcurrentHashMap<>();

    @Override
    public boolean tryAcquire(String key, double qps, long timeout, TimeUnit timeUnit) {
   
   
        RateLimiter limiter = rateLimiters.computeIfAbsent(key, k -> RateLimiter.create(qps));
        if (timeout > 0