Java限流器的小栗子

JDK原生类-Semaphore

@Slf4j
public class JDKLimiter {

    public static void main(String[] args) {
        Semaphore semaphore = new Semaphore(2);
        for (int i = 0; i < 10; i++) {
            new Thread(() -> {
                semaphore.acquireUninterruptibly();
                log.debug("do something");
            }).start();
        }
    }
}

这个看起来简单粗暴，例如上面的示例，初始化的时候，我们设置2个令牌，那就只有2个令牌，消耗掉之后不会继续产生，需要手动调用release()方法来补充一个或多个令牌，才能保证后面的线程继续能拿到令牌。

Guava工具类-RateLimiter

@Slf4j
public class GuavaLimiter {

    public static void main(String[] args) {
        RateLimiter limiter = RateLimiter.create(2.0);
        for (int i = 0; i < 10; i++) {
            new Thread(() -> {
                limiter.acquire();
                log.debug("do something");
            }).start();
        }
    }
}

这个比较简单，而且我觉得也不粗暴。初始化的时候，设置好速率即可。如上面的示例，每秒产生2个令牌，而且桶中最大也只能存在2个令牌。

我最喜欢的一点，就是它产生令牌的速度是均匀的，看起来很优雅。例如我们设置的2，那么他就会每500ms产生一个令牌。缺点就是，它是一个单机限流器，多机环境下不能做全局限流。还有一点我感到奇怪，就是guava现在都28.2了，为什么诞生于13.0的限流器竟然还是Beta？

Redisson工具类-RRateLimiter

@Slf4j
public class RedisLimiter {

    public static void main(String[] args) {
        Config config = new Config();
        config.useSingleServer().setAddress("redis://xxx.xxx.xxx.xxx:6379").setPassword("******").setDatabase(0);
        RedissonClient client = Redisson.create(config);
        RRateLimiter limiter = client.getRateLimiter("req_limiter");
        limiter.trySetRate(RateType.OVERALL, 2, 1, RateIntervalUnit.SECONDS);
        for (int i = 0; i < 10; i++) {
            new Thread(() -> {
                limiter.acquire();
                log.debug("do something");
            }).start();
        }
    }
}