Soul 学习笔记之 rate-limiter(十九)

总系列目录地址

启动 soul-admin ， soul-bootstrap， soul-examples-http

soul-bootstrap 相关依赖：

 <!-- soul ratelimiter plugin start-->
 <dependency>
     <groupId>org.dromara</groupId>
     <artifactId>soul-spring-boot-starter-plugin-ratelimiter</artifactId>
     <version>${project.version}</version>
 </dependency>
 <!-- soul ratelimiter plugin end-->

前端 divide 已经注册了 /http 服务，需要增加 rate_limiter 相关选择器和规则

打开插件开关

使用 Redis 是为了实现全局的锁，限流对整个集群都生效。

使用 idea 里面一个插件（RestServices）进行测试

配置的容量是1，每次秒产生的是1个令牌

技术方案流程图

使用令牌通策略：
大小固定的令牌桶可自行以恒定的速率源源不断地产生令牌。如果令牌不被消耗，或者被消耗的速度小于产生的速度，令牌就会不断地增多，直到把桶填满。后面再产生的令牌就会从桶中溢出。最后桶中可以保存的最大令牌数永远不会超过桶的大小。

代码实现

脚本来源

public class RedisRateLimiter {
	public Mono<RateLimiterResponse> isAllowed(final String id, final double replenishRate, final double burstCapacity) {
	        ...
	        // 通过 Redis 执行 lua 脚本，实现全局限流
	        Flux<List<Long>> resultFlux = Singleton.INST.get(ReactiveRedisTemplate.class).execute(this.script, keys, scriptArgs);
	        return resultFlux.onErrorResume(throwable -> Flux.just(Arrays.asList(1L, -1L)))
	                .reduce(new ArrayList<Long>(), (longs, l) -> {
	                    longs.addAll(l);
	                    return longs;
	                }).map(results -> {
	                    boolean allowed = results.get(0) == 1L;
	                    Long tokensLeft = results.get(1);
	                    RateLimiterResponse rateLimiterResponse = new RateLimiterResponse(allowed, tokensLeft);
	                    ...
	                    return rateLimiterResponse;
	                })...
	    }
	    // 导入 lua 脚本
	    private RedisScript<List<Long>> redisScript() {
	        ...
	        redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("/META-INF/scripts/request_rate_limiter.lua")));
	        ...
	    }
}

总结

rate-limiter 插件使用令牌桶 + Redis lua 脚本的方式进行限流。 令牌桶算法固定可访问上限，每秒生成可以用令牌放到桶内直至桶满了。
Redis 执行 lua 脚本，可以做到整个集群公用一个令牌桶，做到集群服务的限流。