Java最全可以了,基于Redis和Lua实现分布式令牌桶限流,分布式架构+RPC+kafka+多线程

在这个示例中,我们将使用Redis和Lua脚本来实现一个分布式令牌桶限流器。这里的解决方案将使用Redis的EVAL命令来运行Lua脚本,以确保操作的原子性。




import redis.clients.jedis.Jedis;
 
public class RateLimiter {
 
    private static final String LUA_SCRIPT = 
        "local key = KEYS[1] " +
        "local limit = tonumber(ARGV[1]) " +
        "local current = tonumber(redis.call('get', key) or '0') " +
        "if current + 1 > limit then return 0 else " +
        "redis.call('INCRBY', key, '1') " +
        "redis.call('EXPIRE', key, '10') " +
        "return 1 end";
 
    private Jedis jedis;
    private String key;
    private int limit;
 
    public RateLimiter(Jedis jedis, String key, int limit) {
        this.jedis = jedis;
        this.key = key;
        this.limit = limit;
    }
 
    public boolean isAllowed() {
        Long isAllowed = (Long) jedis.eval(LUA_SCRIPT, 1, key, String.valueOf(limit));
        return isAllowed == 1L;
    }
 
    public static void main(String[] args) {
        Jedis jedis = new Jedis("localhost");
        RateLimiter rateLimiter = new RateLimiter(jedis, "rate_limit", 10);
 
        for (int i = 0; i < 20; i++) {
            if (rateLimiter.isAllowed()) {
                System.out.println("Request allowed");
            } else {
                System.out.println("Request not allowed, hit the limit");
            }
        }
 
        jedis.close();
    }
}

在这个Java代码示例中,我们定义了一个RateLimiter类,它有一个isAllowed方法,该方法使用了Redis的EVAL命令来运行Lua脚本。Lua脚本会检查当前令牌桶的令牌数是否超过限制,并相应地增加令牌或返回不允许的信号。

这个简单的例子展示了如何使用Redis和Lua脚本来实现分布式系统中的请求限流,这对于防止恶意请求、防止系统被暴力攻击等场景非常有用。

评论已关闭

推荐阅读

DDPG 模型解析,附Pytorch完整代码
2024年11月24日
DQN 模型解析,附Pytorch完整代码
2024年11月24日
AIGC实战——Transformer模型
2024年12月01日
Socket TCP 和 UDP 编程基础(Python)
2024年11月30日
python , tcp , udp
如何使用 ChatGPT 进行学术润色?你需要这些指令
2024年12月01日
AI
最新 Python 调用 OpenAi 详细教程实现问答、图像合成、图像理解、语音合成、语音识别(详细教程)
2024年11月24日
ChatGPT 和 DALL·E 2 配合生成故事绘本
2024年12月01日
omegaconf,一个超强的 Python 库!
2024年11月24日
【视觉AIGC识别】误差特征、人脸伪造检测、其他类型假图检测
2024年12月01日
[超级详细]如何在深度学习训练模型过程中使用 GPU 加速
2024年11月29日
Python 物理引擎pymunk最完整教程
2024年11月27日
MediaPipe 人体姿态与手指关键点检测教程
2024年11月27日
深入了解 Taipy:Python 打造 Web 应用的全面教程
2024年11月26日
基于Transformer的时间序列预测模型
2024年11月25日
Python在金融大数据分析中的AI应用(股价分析、量化交易)实战
2024年11月25日
AIGC Gradio系列学习教程之Components
2024年12月01日
Python3 `asyncio` — 异步 I/O,事件循环和并发工具
2024年11月30日
llama-factory SFT系列教程:大模型在自定义数据集 LoRA 训练与部署
2024年12月01日
Python 多线程和多进程用法
2024年11月24日
Python socket详解,全网最全教程
2024年11月27日