guava限流

发表于 分类于 阅读次数： 阅读次数： Changyan： 本文字数： 1.5k 阅读时长 ≈ 1 分钟
请添加个描述吧
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
package com.example.login;

import com.google.common.util.concurrent.RateLimiter;

import java.util.Date;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 */
public class TestGuava {
    public static void main(String[] args) {
        RateLimiter limiter = RateLimiter.create(3);
        ExecutorService pool = Executors.newFixedThreadPool(10);
        for(int i =0; i <20; i++) {
            // 限流器限流
            limiter.acquire();
            pool.execute(() -> {
                System.out.println(new Date().toLocaleString());
            });
        }
        pool.shutdown();
    }
}
public class GuavaRateLimiter {
    public static ConcurrentHashMap<String, RateLimiter> resourceRateLimiter = new ConcurrentHashMap<String, RateLimiter>();

    //初始化限流工具RateLimiter
    static {
        createResourceRateLimiter("order", 5);
    }

    public static void createResourceRateLimiter(String resource, double qps) {
        if (resourceRateLimiter.contains(resource)) {
            resourceRateLimiter.get(resource).setRate(qps);
        } else {
            //创建限流工具，每秒发出50个令牌指令
            RateLimiter rateLimiter = RateLimiter.create(qps);
            resourceRateLimiter.putIfAbsent(resource, rateLimiter);

        }

    }

    public static void main(String[] args) {
        for (int i = 0; i < 50; i++) {
            new Thread(new Runnable() {
                @Override
                public void run() {
                    resourceRateLimiter.get("order").acquire();
                    //如果获得令牌指令，则执行业务逻辑
                    System.out.println("执行业务逻辑"+new Date().toLocaleString());
                }
            }).start();
        }

    }
}