soul 源码分析 —— 插件解析之ratelimiter插件

最新推荐文章于 2024-10-30 10:20:22 发布

cj271230

最新推荐文章于 2024-10-30 10:20:22 发布

阅读量118

点赞数

分类专栏： soul 源码分析文章标签： java

本文链接：https://blog.csdn.net/cj271230/article/details/113725728

版权

soul 源码分析专栏收录该内容

14 篇文章 0 订阅

订阅专栏

说明

ratelimiter 是用来限流的

使用

网关引入：

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

后台配置：
在这里插入图片描述

源码分析

请求进入到RateLimiterPlugin的doExecute方法：

protected Mono<Void> doExecute(final ServerWebExchange exchange, final SoulPluginChain chain, final SelectorData selector, final RuleData rule) {
        final String handle = rule.getHandle();
        final RateLimiterHandle limiterHandle = GsonUtils.getInstance().fromJson(handle, RateLimiterHandle.class);
        return redisRateLimiter.isAllowed(rule.getId(), limiterHandle.getReplenishRate(), limiterHandle.getBurstCapacity())
                .flatMap(response -> {
                    if (!response.isAllowed()) {
                        exchange.getResponse().setStatusCode(HttpStatus.TOO_MANY_REQUESTS);
                        Object error = SoulResultWrap.error(SoulResultEnum.TOO_MANY_REQUESTS.getCode(), SoulResultEnum.TOO_MANY_REQUESTS.getMsg(), null);
                        return WebFluxResultUtils.result(exchange, error);
                    }
                    return chain.execute(exchange);
                });
    }