【RPC】一步一步实现基于netty+zookeeper的RPC框架（五）

上一篇实现了服务的链路追踪,本篇带来限流。

关于服务限流，比较流行的是
1：令牌桶算法：桶算法的升级版，实现简单，应对热点请求效果更理想。
2：动态限流：根据实时的统计当前时间段请求响应时间来动态调整限流数量，实现复杂，但应对各种情况效果更好。

这里还是贴出github代码地址,想直接看代码的可以直接下载运行:https://github.com/whiteBX/wrpc

本篇带来令牌桶实现限流的实现，核心代码很简单，主要就下面一个类:

public class ConsumerLimiter {
   

    /**
     * 存放限流器
     */
    private ConcurrentMap<String, RateLimiter> rateLimiterMap = new ConcurrentHashMap<String, RateLimiter>();

    /**
     * 限流
     * @param appCode
     * @return
     */
    public boolean limit(String appCode) {
   
        RateLimiter rateLimiter = rateLimiterMap.get(appCode);
        if (rateLimiter == null) {
   
            rateLimiterMap.putIfAbsent(appCode, RateLimiter.create(1));
        }
        rateLimiter = rateLimiterMap.get(appCode);
        // 默认等待500mS
        return rateLimiter.tryAcquire(500, TimeUnit.NANOSECONDS);
    }
}

这里核心就是用到了guava包的RateLimiter，它内部实现了令牌桶，默认情况下RateLimiter.create(1)表示每秒向令牌桶中投放1个令牌，当然它也支持自己传入时间单位。

家下来改造RPCConsumer类的getServer方法：

/**
 * 获取服务
 * @param appCode
 * @return
 */
private String get