展开菜单
首页 精品内容 本月促销 装机必备 Windows PDF教程 macOS软件 IOS软件 Android AI 专题
全部分类

当前位置:

首页 > 编程开发 > Java常见的限流算法怎么实现

Java常见的限流算法怎么实现

为什么要限流在保证可用的情况下尽可能多增加进入的人数,其余的人在排队等待,或者返回友好提示,保证里面的进行系统的用户可以正常使用,防止系统雪崩。限流算法限流算法很多,常见的有三类,分别是计数器算法、漏桶算法、令牌桶算法。(1)计数器:在一段时间间隔内,处理请求的最大数量固定,超过部分不做处理。(2)漏桶:漏桶大小固定,处理速度固定,但请求进入速度不固定(在突发情况请求过多时,会丢弃过多的请求)。(3)令牌桶:令牌桶的大小固定,令牌的产生速度固定,但是消耗令牌(即请求)速度不固定(可以应对一些某些时间请求过

本文目录

    为什么要限流

    在保证可用的情况下尽可能多增加进入的人数,其余的人在排队等待,或者返回友好提示,保证里面的进行系统的用户可以正常使用,防止系统雪崩。

    限流算法

    限流算法很多,常见的有三类,分别是 计数器算法 、漏桶算法、令牌桶算法 。

    (1)计数器:

              在一段时间间隔内,处理请求的最大数量固定,超过部分不做处理。

    (2)漏桶:

              漏桶大小固定,处理速度固定,但请求进入速度不固定(在突发情况请求过多时,会丢弃过多的请求)。

    (3)令牌桶:

              令牌桶的大小固定,令牌的产生速度固定,但是消耗令牌(即请求)速度不固定(可以应对一些某些时间请求过多的情况);每个请求都会从令牌桶中取出令牌,如果没有令牌则丢弃该次请求。

    计数器限流

    在一段时间间隔内,处理请求的最大数量固定,超过部分不做处理。

    举个例子,比如我们规定对于A接口,我们1分钟的访问次数不能超过100次。

    那么我们可以这么做:

    在一开 始的时候,我们可以设置一个计数器counter,每当一个请求过来的时候,counter就加1,如果counter的值大于100并且该请求与第一个请求的间隔时间还在1分钟之内,那么说明请求数过多,拒绝访问;

    如果该请求与第一个请求的间隔时间大于1分钟,且counter的值还在限流范围内,那么就重置 counter,就是这么简单粗暴。

    Java常见的限流算法怎么实现

    代码实现: 

    import java.util.concurrent.CountDownLatch;
    import java.util.concurrent.ExecutorService;
    import java.util.concurrent.Executors;
    import java.util.concurrent.atomic.AtomicInteger;
    import java.util.concurrent.atomic.AtomicLong;
    
    //计数器 限流
    public class CounterLimiter {
    
        //起始时间
        private static long startTime = System.currentTimeMillis();
    
        //时间间隔1000ms
        private static long interval = 1000;
    
        //每个时间间隔内,限制数量
        private static long limit = 3;
    
        //累加器
        private static AtomicLong accumulator = new AtomicLong();
    
        /**
         * true 代表放行,请求可已通过
         * false 代表限制,不让请求通过
         */
        public static boolean tryAcquire() {
            long nowTime = System.currentTimeMillis();
            //判断是否在上一个时间间隔内
            if (nowTime < startTime + interval) {
                //如果还在上个时间间隔内
                long count = accumulator.incrementAndGet();
                if (count <= limit) {
                    return true;
                } else {
                    return false;
                }
            } else {
                //如果不在上一个时间间隔内
                synchronized (CounterLimiter.class) {
                    //防止重复初始化
                    if (nowTime > startTime + interval) {
                        startTime = nowTime;
                        accumulator.set(0);
                    }
                }
                //再次进行判断
                long count = accumulator.incrementAndGet();
                if (count <= limit) {
                    return true;
                } else {
                    return false;
                }
            }
        }
    
    
        // 测试
        public static void main(String[] args) {
    
            //线程池,用于多线程模拟测试
            ExecutorService pool = Executors.newFixedThreadPool(10);
            // 被限制的次数
            AtomicInteger limited = new AtomicInteger(0);
            // 线程数
            final int threads = 2;
            // 每条线程的执行轮数
            final int turns = 20;
            // 同步器
            CountDownLatch countDownLatch = new CountDownLatch(threads);
            long start = System.currentTimeMillis();
            for (int i = 0; i < threads; i++) {
                pool.submit(() ->
                {
                    try {
    
                        for (int j = 0; j < turns; j++) {
    
                            boolean flag = tryAcquire();
                            if (!flag) {
                                // 被限制的次数累积
                                limited.getAndIncrement();
                            }
                            Thread.sleep(200);
                        }
    
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                    //等待所有线程结束
                    countDownLatch.countDown();
                });
            }
            try {
                countDownLatch.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            float time = (System.currentTimeMillis() - start) / 1000F;
            //输出统计结果
            System.out.println("限制的次数为:" + limited.get() +
                    ",通过的次数为:" + (threads * turns - limited.get()));
            System.out.println("限制的比例为:" + (float) limited.get() / (float) (threads * turns));
            System.out.println("运行的时长为:" + time + "s");
        }
    
    }

    计数器限流的不足: 

    这个算法虽然简单,但是存在临界问题,我们看下图:

    Java常见的限流算法怎么实现

    从上图中我们可以看到,假设有一个恶意用户,他在0:59时,瞬间发送了100个请求,并且1:00又瞬间发送了100个请求,那么其实这个用户在 1秒里面,瞬间发送了200个请求。

    我们刚才规定的是1分钟最多100个请求(规划的吞吐量),也就是每秒钟最多1.7个请求,用户通过在时间窗口的重置节点处突发请求, 可以瞬间超过我们的速率限制。

    用户有可能通过算法的这个漏洞,瞬间压垮我们的应用。

    漏桶限流

    漏桶算法限流的基本原理为:水(对应请求)从进水口进入到漏桶里,漏桶以一定的速度出水(请求放行),当水流入速度过大,桶内的总水量大于桶容量会直接溢出,请求被拒绝。

    大致的漏桶限流规则如下:

    (1)进水口(对应客户端请求)以任意速率流入进入漏桶。

    (2)漏桶的容量是固定的,出水(放行)速率也是固定的。

    (3)漏桶容量是不变的,如果处理速度太慢,桶内水量会超出了桶的容量,则后面流入的水滴会溢出,表示请求拒绝。

    Java常见的限流算法怎么实现

    ⭐漏桶算法其实很简单,可以粗略的认为就是注水漏水过程,往桶中以任意速率流入水,以一定速率流出水,当水超过桶容量(capacity)则丢弃,因为桶容量是不变的,保证了整体的速率。

    以一定速率流出水,

    Java常见的限流算法怎么实现

    削峰: 有大量流量进入时,会发生溢出,从而限流保护服务可用

    缓冲: 不至于直接请求到服务器, 缓冲压力

    代码实现: 

    import java.util.concurrent.CountDownLatch;
    import java.util.concurrent.ExecutorService;
    import java.util.concurrent.Executors;
    import java.util.concurrent.atomic.AtomicInteger;
    import java.util.concurrent.atomic.AtomicLong;
    
    //漏斗限流
    public class LeakBucketLimiter {
    
        //桶的大小
        private static long capacity = 10;
        //流出速率,每秒两个
        private static long rate = 2;
        //开始时间
        private static long startTime = System.currentTimeMillis();
        //桶中剩余的水
        private static AtomicLong water = new AtomicLong();
    
        /**
         * true 代表放行,请求可已通过
         * false 代表限制,不让请求通过
         */
        public synchronized static boolean tryAcquire() {
            //如果桶的余量问0,直接放行
            if (water.get() == 0) {
                startTime = System.currentTimeMillis();
                water.set(1);
                return true;
            }
            //计算从当前时间到开始时间流出的水,和现在桶中剩余的水
            //桶中剩余的水
            water.set(water.get() - (System.currentTimeMillis() - startTime) / 1000 * rate);
            //防止出现<0的情况
            water.set(Math.max(0, water.get()));
            //设置新的开始时间
            startTime += (System.currentTimeMillis() - startTime) / 1000 * 1000;
            //如果当前水小于容量,表示可以放行
            if (water.get() < capacity) {
                water.incrementAndGet();
                return true;
            } else {
                return false;
            }
        }
    
    
        // 测试
        public static void main(String[] args) {
    
            //线程池,用于多线程模拟测试
            ExecutorService pool = Executors.newFixedThreadPool(10);
            // 被限制的次数
            AtomicInteger limited = new AtomicInteger(0);
            // 线程数
            final int threads = 2;
            // 每条线程的执行轮数
            final int turns = 20;
            // 同步器
            CountDownLatch countDownLatch = new CountDownLatch(threads);
            long start = System.currentTimeMillis();
            for (int i = 0; i < threads; i++) {
                pool.submit(() ->
                {
                    try {
    
                        for (int j = 0; j < turns; j++) {
    
                            boolean flag = tryAcquire();
                            if (!flag) {
                                // 被限制的次数累积
                                limited.getAndIncrement();
                            }
                            Thread.sleep(200);
                        }
    
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                    //等待所有线程结束
                    countDownLatch.countDown();
                });
            }
            try {
                countDownLatch.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            float time = (System.currentTimeMillis() - start) / 1000F;
            //输出统计结果
            System.out.println("限制的次数为:" + limited.get() +
                    ",通过的次数为:" + (threads * turns - limited.get()));
            System.out.println("限制的比例为:" + (float) limited.get() / (float) (threads * turns));
            System.out.println("运行的时长为:" + time + "s");
        }
    
    }

    漏桶的不足: 

    漏桶的出水速度固定,也就是请求放行速度是固定的。

    漏桶出口的速度固定,不能灵活的应对后端能力提升。比如,通过动态扩容,后端流量从1000QPS提升到1WQPS,漏桶没有办法。

    令牌桶限流

    令牌桶算法中新请求到来时会从桶里拿走一个令牌,如果桶内没有令牌可拿,就拒绝服务。 当然,令牌的数量也是有上限的。令牌的数量与时间和发放速率强相关,时间流逝的时间越长,会不断往桶里加入越多的令牌,如果令牌发放的速度比申请速度快,令牌桶会放满令牌,直到令牌占满整个令牌桶。

    令牌桶限流大致的规则如下:

    (1)进水口按照某个速度,向桶中放入令牌。

    (2)令牌的容量是固定的,但是放行的速度不是固定的,只要桶中还有剩余令牌,一旦请求过来就能申请成功,然后放行。

    (3)如果令牌的发放速度,慢于请求到来速度,桶内就无牌可领,请求就会被拒绝。

    总之,令牌的发送速率可以设置,从而可以对突发的出口流量进行有效的应对。

    Java常见的限流算法怎么实现

    代码实现: 

    import java.util.concurrent.CountDownLatch;
    import java.util.concurrent.ExecutorService;
    import java.util.concurrent.Executors;
    import java.util.concurrent.atomic.AtomicInteger;
    import java.util.concurrent.atomic.AtomicLong;
    
    //令牌桶
    public class TokenBucketLimiter {
        //桶的容量
        private static long capacity = 10;
        //放入令牌的速率,每秒2个
        private static long rate = 2;
        //上次放置令牌的时间
        private static long lastTime = System.currentTimeMillis();
        //桶中令牌的余量
        private static AtomicLong tokenNum = new AtomicLong();
    
        /**
         * true 代表放行,请求可已通过
         * false 代表限制,不让请求通过
         */
        public synchronized static boolean tryAcquire() {
            //更新桶中剩余令牌的数量
            long now = System.currentTimeMillis();
            tokenNum.addAndGet((now - lastTime) / 1000 * rate);
            tokenNum.set(Math.min(capacity, tokenNum.get()));
            //更新时间
            lastTime += (now - lastTime) / 1000 * 1000;
            //桶中还有令牌就放行
            if (tokenNum.get() > 0) {
                tokenNum.decrementAndGet();
                return true;
            } else {
                return false;
            }
        }
    
    
        //测试
        public static void main(String[] args) {
    
            //线程池,用于多线程模拟测试
            ExecutorService pool = Executors.newFixedThreadPool(10);
            // 被限制的次数
            AtomicInteger limited = new AtomicInteger(0);
            // 线程数
            final int threads = 2;
            // 每条线程的执行轮数
            final int turns = 20;
            // 同步器
            CountDownLatch countDownLatch = new CountDownLatch(threads);
            long start = System.currentTimeMillis();
            for (int i = 0; i < threads; i++) {
                pool.submit(() ->
                {
                    try {
    
                        for (int j = 0; j < turns; j++) {
    
                            boolean flag = tryAcquire();
                            if (!flag) {
                                // 被限制的次数累积
                                limited.getAndIncrement();
                            }
                            Thread.sleep(200);
                        }
    
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                    //等待所有线程结束
                    countDownLatch.countDown();
                });
            }
            try {
                countDownLatch.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            float time = (System.currentTimeMillis() - start) / 1000F;
            //输出统计结果
            System.out.println("限制的次数为:" + limited.get() +
                    ",通过的次数为:" + (threads * turns - limited.get()));
            System.out.println("限制的比例为:" + (float) limited.get() / (float) (threads * turns));
            System.out.println("运行的时长为:" + time + "s");
        }
    
    }

    令牌桶的好处: 

    令牌桶的好处之一就是可以方便地应对 突发出口流量(后端能力的提升)。

    比如,可以改变令牌的发放速度,算法能按照新的发送速率调大令牌的发放数量,使得出口突发流量能被处理。

    本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系bd@zhengruan.com
    作者最新文章
    编程开发 Java
    相关文章 更多
    精品专题 更多
    装机必备

    正软商城装机必备专区,精选办公、浏览器、安全防护、影音播放、压缩解压、设计创作和系统工具等电脑常用正版软件,帮助用户快速完成新电脑软件配置。

    Windows

    正软商城Windows软件专区,汇集适用于Windows电脑的办公、设计、安全防护、影音播放、开发工具和系统优化软件,提供软件介绍、系统要求、正版授权及购买下载服务。

    PDF教程

    正软商城PDF教程频道提供PDF编辑、转换、合并、拆分、压缩及格式处理方法,同时介绍常用PDF软件和工具的使用技巧。

    macOS软件

    正软商城macOS软件专区,精选适用于Mac电脑的办公、设计、影音、效率、开发和系统工具,提供软件功能介绍、macOS兼容版本、正版授权及购买下载服务。

    IOS软件

    正软商城iOS软件专区,精选适用于iPhone和iPad的办公、学习、影音、设计、效率及AI应用,提供功能介绍、适用设备、系统要求和正版获取方式等信息。

    AI

    正软商城AI软件专区,汇集AI写作、AI绘画、AI视频、AI办公、AI编程、AI翻译、智能客服和数据分析等人工智能工具,提供功能介绍、适用平台、收费方式及正版购买信息。

    Mac软件 更多
    Aperty macOS版
    Aperty macOS版

    Aperty 是 Skylum 推出的 AI 人像修图软件,支持自然肤质修饰、瑕疵处理、面部与身体调整、预设及批量同步。

    火山引擎 网页版
    火山引擎 网页版

    火山引擎是字节跳动旗下企业级云与AI服务平台。聚焦大模型核心能力,依托字节自研豆包多模态大模型,凭借火山方舟一站式平台,提供模型精调、推理部署、智能应用搭建等全链路服务,结合高性能AI算力,助力各行业快速落地大模型数字化与智能化应用。

    立刻MV 网页版
    立刻MV 网页版

    立刻MV 面向音乐人、内容创作者和品牌团队,帮助用户低成本完成歌曲 MV、歌词视频、AI 分镜、图生视频与最终合成。

    蛙蛙写作 网页版
    蛙蛙写作 网页版

    蛙蛙写作 AI 创作工具,覆盖小说、剧本、文案,一键生成文稿,打通文字转视频全流程,高效解决创作难题。

    讯飞智作 网页版
    讯飞智作 网页版

    讯飞智作是一款基于语音合成技术、虚拟人驱动技术与数字人形象声音构建能力的 AI 数字人音视频创作平台。

    超级简历WonderCV 网页版
    超级简历WonderCV 网页版

    超级简历WonderCV是面向应届生和AI小白的智能求职平台,覆盖简历、面试与职业规划。

    音述AI 网页版
    音述AI 网页版

    音述AI,是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区。

    AionClaw macOS版
    AionClaw macOS版

    AionClaw 是可在桌面端运行的任务型 AI 助手,通过数字专家与技能市场组合办公、创作、数据和开发能力。

    Loomy macOS版
    Loomy macOS版

    Loomy 是科大讯飞推出的桌面 AI 助手,可通过对话、任务与工作目录协助处理办公、内容创作和资料整理。

    AirBuddy 3 macOS版
    AirBuddy 3 macOS版

    AirBuddy 3 是面向 macOS 的蓝牙外设增强工具,可集中查看 AirPods、Beats 与苹果设备电量,提供快速连接、播放控制、电量提醒、桌面小组件和多台 Mac 外设接力等功能。

    AirServer macOS版
    AirServer macOS版

    AirServer macOS版可把 Mac 转换为无线投屏接收端,通过 AirPlay 或 Google Cast 接收手机、平板和电脑画面,并支持多设备同时连接、高清镜像及会话录制,适合家庭娱乐、课堂教学、会议演示和应用展示。

    AnyDesk macOS版
    AnyDesk macOS版

    AnyDesk macOS版是一款适用于远程办公、设备维护和客户支持的远程桌面工具,可连接并操作异地计算机,支持文件传输、无人值守访问、会话权限控制及跨平台协作。

    WINDOWS 更多
    讯飞绘文 网页版
    讯飞绘文 网页版

    讯飞绘文网页版是科大讯飞推出的AI图文创作与内容运营工具,覆盖热点选题、文章生成、稿件校对、智能配图、排版发布和数据复盘等环节,适合自媒体创作者及内容团队在线使用。

    Maya(Autodesk Maya) Windows版
    Maya(Autodesk Maya) Windows版

    Autodesk Maya 是面向影视、电视、动画和游戏行业的专业三维制作软件,集多边形与 NURBS 建模、角色绑定、关键帧和动作捕捉动画、Bifrost 特效模拟、材质开发及 Arnold 渲染于一体,

    Aperty Windows版
    Aperty Windows版

    Aperty 是 Skylum 推出的 AI 人像修图软件,支持自然肤质修饰、瑕疵处理、面部与身体调整、预设及批量同步。

    火山引擎 网页版
    火山引擎 网页版

    火山引擎是字节跳动旗下企业级云与AI服务平台。聚焦大模型核心能力,依托字节自研豆包多模态大模型,凭借火山方舟一站式平台,提供模型精调、推理部署、智能应用搭建等全链路服务,结合高性能AI算力,助力各行业快速落地大模型数字化与智能化应用。

    立刻MV 网页版
    立刻MV 网页版

    立刻MV 面向音乐人、内容创作者和品牌团队,帮助用户低成本完成歌曲 MV、歌词视频、AI 分镜、图生视频与最终合成。

    蛙蛙写作 网页版
    蛙蛙写作 网页版

    蛙蛙写作 AI 创作工具,覆盖小说、剧本、文案,一键生成文稿,打通文字转视频全流程,高效解决创作难题。

    讯飞智作 网页版
    讯飞智作 网页版

    讯飞智作是一款基于语音合成技术、虚拟人驱动技术与数字人形象声音构建能力的 AI 数字人音视频创作平台。

    超级简历WonderCV 网页版
    超级简历WonderCV 网页版

    超级简历WonderCV是面向应届生和AI小白的智能求职平台,覆盖简历、面试与职业规划。

    音述AI 网页版
    音述AI 网页版

    音述AI,是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区。

    AionClaw Windows版
    AionClaw Windows版

    AionClaw 是可在桌面端运行的任务型 AI 助手,通过数字专家与技能市场组合办公、创作、数据和开发能力。

    Loomy Windows版
    Loomy Windows版

    Loomy 是科大讯飞推出的桌面 AI 助手,可通过对话、任务与工作目录协助处理办公、内容创作和资料整理。

    AirServer Windows版
    AirServer Windows版

    AirServer Windows版可把电脑变成AirPlay、Google Cast和Miracast无线投屏接收端,适合会议演示、课堂教学及家庭大屏分享。发送设备通常无需另装客户端,并可同时接收和智能排列多个镜像会话。