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脚本来实现分布式系统中的请求限流,这对于防止恶意请求、防止系统被暴力攻击等场景非常有用。

评论已关闭

推荐阅读

Vue中使用mind-map实现在线思维导图
2024年08月04日
VUE
Web前端最全Vue实现免密登录跳转的方式_vue怎么样不登录返回首页,最强技术实现
2024年08月04日
VUE
vue3 项目搭建教程(基于create-vue,vite,Vite + Vue)
2024年08月04日
VUE
Vue-颜色选择器实现方案——>Vue-Color( 实战*1+ Demo*7)
2024年08月04日
VUE
Vue项目卡顿慢加载?这些优化技巧告诉你!_vue数据多渲染卡顿
2024年08月04日
VUE
vue中的keep-alive详解与应用场景
2024年08月04日
VUE
Vue、React实现excel导出功能(三种实现方式保姆级讲解)
2024年08月04日
vue-office/docx插件实现docx文件预览
2024年08月04日
VUE
java调用js文件的两种方法(支持V8引擎)
2024年08月04日
JavaScript:解决计算精度问题/mathjs/bignumber.js/big.js/decimal.js
2024年08月04日
两周从爬虫小白变大神 _yjs_js_security_passport
2024年08月04日
JS笔记(对象、函数、数组)
2024年08月04日
Markdown.js:强大的纯JavaScript Markdown解析器
2024年08月04日
Vue项目:js模拟点击a标签下载文件并重命名,URL文件地址下载方法、请求接口下载文件方法总结。
2024年08月04日
vue 父组件怎么获取子组件里面的data数据
2024年08月04日
VUE
个人开发实现AI套壳网站快速搭建(Vue+elementUI+SpringBoot)
2024年08月04日
el-table 表格封装并改造实现单元格可编辑
2024年08月04日
none
nodejs环境下创建vue项目、SSH密钥登陆!!!
2024年08月04日
vue+quill+element-ui实现视频、图片上传及缩放保姆级教程,轻松使用富文本
2024年08月04日
【three.js】22. Imported Models导入模型
2024年08月04日