谈谈架构中的限流与计数器的实现方式

1. 先谈谈Nginx分流

对,要考虑限流先得假设访问量达到了一定得程度,再高并发得前提下,请求过多很有可能导致某天服务器承受不了导致死机。

在这个前提下,我相信你最先想到的一定是nginx,使用nginx分流让所有的请求不要直接到达某一个服务器,当并发量继续上升的时候,我提供更多的服务器视乎就能解决问题了,这想法看上去很对,而且很多行业就是这么做的,比如下面的nginx配置文件,很简单,我相信你一看就能懂

#user  nobody;

worker_processes  1;

events {
    worker_connections  1024;
}

http {

    upstream enjoy{
        server 127.0.0.1:8080;        
        server 127.0.0.1:8081;      

    }
    server {
            listen 80;
            location / {
                    proxy_pass http://enjoy;
            }
    }

}

在上面案例中我配置了个代理,当高并发的请求过来的时候,会把请求分发给8080与8081两个端口的服务器。

但现在我们假设一个极端的情况,这个时候由于业务有个秒杀要求,请求过大一下就让这两个服务器爆了,这个时候在我继续增加服务器之前这整个秒杀业务几乎都处在瘫痪状态,而这突然的访问量是你始料未及的,也就是说你根本就没法事先准备好足够的服务器来解决这种情况。

这个时候我相信你已经看出了如果仅仅做分流依然会出现很严重的问题,怎么办呢?这个时候你就需要限流了。

2. 再说下限流

限流依然是再高并发的前提下,如果某个服务器承受不了数目过多的请求量的一种限制机制,不同的是分流是让请求分发的其他服务器,而限流是达到某个阈值后直接不让你访问了

如果想完成限流的功能其实是有一些解决方案的(算法),比如来说,基于令牌桶的,程序计数器以及漏桶算法;

今天挑一个最简单的计数器方式来讲讲限流

3. 解决方案

计数器的解决方式是最简单最容易实现的一种解决方案,假设有一个接口,要求1分钟的访问量不能超过10次

这样当有任何请求过来,我可以让计数器+1;如果这个计数器的值大于10,而且和第一次的请求相比,时间间隔在1分钟以内,那么久能说明该请求访问过多。

如果这个请求与第一次请求的访问时间之间的间隔超过了1分钟,那么该计数器的值久还是限流范围之内,接下来久只要重置计数器就好;

import java.util.concurrent.ExecutorService;

import java.util.concurrent.Executors;

import java.util.concurrent.atomic.AtomicInteger;

public class EnjoyCountLimit {

    private int limtCount = 60;// 限制最大访问的容量

    AtomicInteger atomicInteger = new AtomicInteger(0); // 每秒钟 实际请求的数量

    private long start = System.currentTimeMillis();// 获取当前系统时间

    private int interval = 60*1000;// 间隔时间60秒

    public boolean acquire() {

        long newTime = System.currentTimeMillis();

        if (newTime > (start + interval)) {

            // 判断是否是一个周期

            start = newTime;

            atomicInteger.set(0); // 清理为0

            return true;

        }

        atomicInteger.incrementAndGet();// i++;

        return atomicInteger.get() <= limtCount;

    }

    static EnjoyCountLimit limitService = new EnjoyCountLimit();

    public static void main(String[] args) {

        ExecutorService newCachedThreadPool = Executors.newCachedThreadPool();

        for (int i = 1; i < 100; i++) {

            final int tempI = i;

            newCachedThreadPool.execute(new Runnable() {

                public void run() {

                    if (limitService.acquire()) {

                        System.out.println("你没有被限流,可以正常访问逻辑 i:" + tempI);

                    } else {

                        System.out.println("你已经被限流呢  i:" + tempI);

                    }

                }

            });

        }

    }

}

这个计数器的限流方式很简单吧,但这样问题吗?好好想想……

还是以60运行访问10次请求为例,在第一次0-58秒之内,没有访问请求,在59秒之内突然来了10次请求,这个时候会做什么,由于已经到了1分钟计数器会重置。

这个时候第二次的1秒内(1分0秒)又了10请求,这个时候是不是就在2秒之内有20个请求被放行了呢?(59秒,1分0秒),如果某个服务器的访问量只能是10次请求,那这种限流方式已经导致服务器挂了;

4. 滑动窗口计数器

前面已经知道简单的计数器的实现方式,也知道他会出现的一些问题,虽然这些问题举得有些极端,但还是有更好得解决方案,这方案就是使用滑动窗口计数器

滑动窗口计数器得原理是在没错请求过来得时候,先判断前面N个单位内得总访问量是否操过得阈值,并且在当前得时间单位得请求数上+1

举例来说,要求1分钟的访问量不能超过10次

可以把1分钟看成是6个10秒钟的时间,0-9秒的访问数记录到第一个格子,10-19秒的访问数记录数记录到第二个格子以此内推,每次统计将6个格子里面的数据求和,如果超过了10次就不允许访问。

import java.util.concurrent.atomic.AtomicInteger;

public class EnjoySlidingWindow {

    private AtomicInteger[] timeSlices;

    /* 队列的总长度  */

    private final int timeSliceSize;

    /* 每个时间片的时长 */

    private final long timeMillisPerSlice;

    /* 窗口长度 */

    private final int windowSize;

    /* 当前所使用的时间片位置 */

    private AtomicInteger cursor = new AtomicInteger(0);

    public static enum Time {

        MILLISECONDS(1),

        SECONDS(1000),

        MINUTES(SECONDS.getMillis() * 60),

        HOURS(MINUTES.getMillis() * 60),

        DAYS(HOURS.getMillis() * 24),

        WEEKS(DAYS.getMillis() * 7);

        private long millis;

        Time(long millis) {

            this.millis = millis;

        }

        public long getMillis() {

            return millis;

        }

    }

    public EnjoySlidingWindow(int windowSize, Time timeSlice) {

        this.timeMillisPerSlice = timeSlice.millis;

        this.windowSize = windowSize;

        // 保证存储在至少两个window

        this.timeSliceSize = windowSize * 2 + 1;

        init();

    }

    /**

     * 初始化

     */

    private void init() {

        AtomicInteger[] localTimeSlices = new AtomicInteger[timeSliceSize];

        for (int i = 0; i < timeSliceSize; i++) {

            localTimeSlices[i] = new AtomicInteger(0);

        }

        timeSlices = localTimeSlices;

    }

    private int locationIndex() {

        long time = System.currentTimeMillis();

        return (int) ((time / timeMillisPerSlice) % timeSliceSize);

    }

    /**

     * <p>对时间片计数+1,并返回窗口中所有的计数总和

     * <p>该方法只要调用就一定会对某个时间片进行+1

     * @return

     */

    public int incrementAndSum() {

        int index = locationIndex();

        int sum = 0;

        // cursor等于index,返回true

        // cursor不等于index,返回false,并会将cursor设置为index

        int oldCursor = cursor.getAndSet(index);

        if (oldCursor == index) {

            // 在当前时间片里继续+1

            sum += timeSlices[index].incrementAndGet();

        } else {

            //轮到新的时间片,置0,可能有其它线程也置了该值,容许

            timeSlices[index].set(0);

            // 清零,访问量不大时会有时间片跳跃的情况

            clearBetween(oldCursor, index);

            sum += timeSlices[index].incrementAndGet();

        }

        for (int i = 1; i < windowSize; i++) {

            sum += timeSlices[(index - i + timeSliceSize) % timeSliceSize].get();

        }

        return sum;

    }

    /**

     * 判断是否允许进行访问,未超过阈值的话才会对某个时间片+1

     * @param threshold

     * @return

     */

    public boolean allow(int threshold) {

        int index = locationIndex();

        int sum = 0;

        int oldCursor = cursor.getAndSet(index);

        if (oldCursor != index) {

            timeSlices[index].set(0);

            clearBetween(oldCursor, index);

        }

        for (int i = 0; i < windowSize; i++) {

            sum += timeSlices[(index - i + timeSliceSize) % timeSliceSize].get();

        }

        // 阈值判断

        if (sum < threshold) {

            // 未超过阈值才+1

            timeSlices[index].incrementAndGet();

            return true;

        }

        return false;

    }

    /**

     * <p>将fromIndex~toIndex之间的时间片计数都清零

     * <p>极端情况下,当循环队列已经走了超过1个timeSliceSize以上,这里的清零并不能如期望的进行

     * @param fromIndex 不包含

     * @param toIndex 不包含

     */

    private void clearBetween(int fromIndex, int toIndex) {

        for (int index = (fromIndex + 1) % timeSliceSize; index != toIndex; index = (index + 1) % timeSliceSize) {

            timeSlices[index].set(0);

        }

    }

    public static void main(String[] args) {

        EnjoySlidingWindow window = new EnjoySlidingWindow(5, Time.MILLISECONDS);

        for (int i = 0; i < 10; i++) {

            System.out.println(window.allow(7));

        }

    }

}

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,445评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,889评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,047评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,760评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,745评论 5 367
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,638评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,011评论 3 398
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,669评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,923评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,655评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,740评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,406评论 4 320
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,995评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,961评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,197评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,023评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,483评论 2 342

推荐阅读更多精彩内容