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 ( ) ;
}
}
}
用起来和guava的限流器差不多,而且支持多机环境下的全局限流。关于不足,我觉得就是它的令牌不是匀速发布的,感觉不够优雅。