自研网关:限流功能的开发
时间:2022-07-28
本文章向大家介绍自研网关:限流功能的开发,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。
自研网关系统已开源,求star 项目地址:
网关系统,如果没有限流功能,感觉就没有了灵魂。 所以一直想把限流功能加上。 经查资料,实现功能如下:
限流功能配置
测试限流功能
具体实现如下:
@Slf4j
@Component
public class GatewayRedisRateLimiter implements InitializingBean {
@Autowired
@Lazy
private RedisTemplate stringRedisTemplate;
private RedisScript<List<Long>> script;
private AtomicBoolean initialized = new AtomicBoolean(false);
/**
* This uses a basic token bucket algorithm and relies on the fact that Redis scripts
* execute atomically. No other operations can run between fetching the count and
* writing the new count.
*
* @param key is rule id
* @param replenishRate replenishRate
* @param burstCapacity burstCapacity
* @return {@code Mono<Response>} to indicate when request processing is complete
*/
public Boolean isAllowed(final String key, final double replenishRate, final double burstCapacity) {
if (!this.initialized.get()) {
throw new IllegalStateException("RedisRateLimiter is not initialized");
}
List<String> keys = getKeys(key);
String[] scriptArgs = new String[]{replenishRate + "", burstCapacity + "", Instant.now().getEpochSecond() + "", "1"};
List<Long> resultFlux = (List<Long>) stringRedisTemplate.execute(this.script, keys, scriptArgs);
if (CollectionUtils.isEmpty(resultFlux)) {
resultFlux = Arrays.asList(1L, -1L);
}
boolean allowed = resultFlux.get(0) == 1L;
Long tokensLeft = resultFlux.get(1);
log.info("RateLimiter key:{},allowed:{},tokensLeft:{}", key, allowed, tokensLeft);
return allowed;
}
private static List<String> getKeys(final String id) {
String prefix = "request_rate_limiter.{" + id;
String tokenKey = prefix + "}.tokens";
String timestampKey = prefix + "}.timestamp";
return Arrays.asList(tokenKey, timestampKey);
}
@SuppressWarnings("unchecked")
private RedisScript<List<Long>> redisScript() {
DefaultRedisScript redisScript = new DefaultRedisScript<>();
redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("/META-INF/scripts/request_rate_limiter.lua")));
redisScript.setResultType(List.class);
return redisScript;
}
@Override
public void afterPropertiesSet() {
this.script = redisScript();
initialized.compareAndSet(false, true);
}
}
这是主要的核心代码,可用于普通的spring mvc项目 具体代码大家可以去我的开源项目中看,欢迎提ISSUE及需求。 因为只有一个人,也欢迎大家参与进来.
网关系统暂时一段落,后面只修改缺陷不新增功能,主要是人不够,只有我一个人 其实网关还有很多我想做的。
- 协议泛化功能,将dubbo转http,将webservice转http
- 特定的日志查询与跟踪。需要将部分的日志做特殊处理,好做数据分析及跟踪
- 引入GraphQl,将普通的Rest接口转成GraphQl接口 可惜我一个人忙不过来 下期计划: 流程引擎(暂未开源) 1。增加机器人节点,可考虑做RPA的相关事情 2。优化流程图设计器,这方面涉及前端,不太擅长
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法