Ribbon LoadBalancerStat中定时统计服务提供者请求状态信息的源码分析

最新推荐文章于 2024-06-23 17:00:51 发布

XMHANG

最新推荐文章于 2024-06-23 17:00:51 发布

阅读量565

点赞数

分类专栏： java 文章标签： ribbon java

本文链接：https://blog.csdn.net/weixin_40414302/article/details/123459211

版权

java 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

从RibbonClientConfifiguration配置类中默认加载ZoneAwareLoadBalancer类看起

若不知道为啥从这个配置类的加载开始看起的话，我再补一篇从RestTemplate调用getForObject发起http请求开始看的好了

@Bean
    @ConditionalOnMissingBean
    public ILoadBalancer ribbonLoadBalancer(IClientConfig config, ServerList<Server> serverList, ServerListFilter<Server> serverListFilter, IRule rule, IPing ping, ServerListUpdater serverListUpdater) {
        return (ILoadBalancer)(this.propertiesFactory.isSet(ILoadBalancer.class, this.name) ? (ILoadBalancer)this.propertiesFactory.get(ILoadBalancer.class, config, this.name) : new ZoneAwareLoadBalancer(config, rule, ping, serverList, serverListFilter, serverListUpdater));
    }

在初始化ZoneAwareLoadBalancer后会调用父类DynamicServerListLoadBalancer
在这里插入图片描述

进入restOfInit方法

void restOfInit(IClientConfig clientConfig) {
    boolean primeConnection = this.isEnablePrimingConnections();
    // turn this off to avoid duplicated asynchronous priming done in BaseLoadBalancer.setServerList()
    this.setEnablePrimingConnections(false);
    enableAndInitLearnNewServersFeature();

    updateListOfServers();
    if (primeConnection && this.getPrimeConnections() != null) {
        this.getPrimeConnections()
                .primeConnections(getReachableServers());
    }
    this.setEnablePrimingConnections(primeConnection);
    LOGGER.info("DynamicServerListLoadBalancer for client {} initialized: {}", clientConfig.getClientName(), this.toString());
}

其中的updateListOfServers();即步入定时统计服务提供者请求状态信息的任务的入口

@VisibleForTesting
public void updateListOfServers() {
    List<T> servers = new ArrayList<T>();
    if (serverListImpl != null) {
        servers = serverListImpl.getUpdatedListOfServers();
        LOGGER.debug("List of Servers for {} obtained from Discovery client: {}",
                getIdentifier(), servers);

        if (filter != null) {
            servers = filter.getFilteredListOfServers(servers);
            LOGGER.debug("Filtered List of Servers for {} obtained from Discovery client: {}",
                    getIdentifier(), servers);
        }
    }
    updateAllServerList(servers);
}

此方法最后会执行updateAllServerList(servers);¹

之后进入父类BaseLoadBalancer的forceQuickPing()方法中执行启动runPinger()定时统计服务提供者请求状态信息的任务，这是初始化后强制执行更新的一次操作

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Ug4geI4J-1647352907552)(C:\Users\HASEE\AppData\Roaming\Typora\typora-user-images\image-20220315214648497.png)]$

public void forceQuickPing() {
    if (canSkipPing()) {
        return;
    }
    logger.debug("LoadBalancer [{}]:  forceQuickPing invoking", name);
    
    try {
       new Pinger(pingStrategy).runPinger();
    } catch (Exception e) {
        logger.error("LoadBalancer [{}]: Error running forceQuickPing()", name, e);
    }
}

public void runPinger() throws Exception {
    if (!pingInProgress.compareAndSet(false, true)) { 
        return; // Ping in progress - nothing to do
    }
    
    // we are "in" - we get to Ping

    Server[] allServers = null;
    boolean[] results = null;

    Lock allLock = null;
    Lock upLock = null;

    try {
        /*
         * The readLock should be free unless an addServer operation is
         * going on...
         */
        allLock = allServerLock.readLock();
        allLock.lock();
        allServers = allServerList.toArray(new Server[allServerList.size()]);
        allLock.unlock();

        int numCandidates = allServers.length;
        results = pingerStrategy.pingServers(ping, allServers);

        final List<Server> newUpList = new ArrayList<Server>();
        final List<Server> changedServers = new ArrayList<Server>();

        for (int i = 0; i < numCandidates; i++) {
            boolean isAlive = results[i];
            Server svr = allServers[i];
            boolean oldIsAlive = svr.isAlive();

            svr.setAlive(isAlive);

            if (oldIsAlive != isAlive) {
                changedServers.add(svr);
                logger.debug("LoadBalancer [{}]:  Server [{}] status changed to {}", 
                  name, svr.getId(), (isAlive ? "ALIVE" : "DEAD"));
            }

            if (isAlive) {
                newUpList.add(svr);
            }
        }
        upLock = upServerLock.writeLock();
        upLock.lock();
        upServerList = newUpList;
        upLock.unlock();

        notifyServerStatusChangeListener(changedServers);
    } finally {
        pingInProgress.set(false);
    }
}

此外在BaseLoadBalancer类中有个属性

protected Timer lbTimer = null;

Timer由setupPingTask()方法实例化并启动加载定时任务

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-kv7FGxPZ-1647352907553)(C:\Users\HASEE\AppData\Roaming\Typora\typora-user-images\image-20220315204936487.png)]$

在new ShutdownEnabledTimer()的第二个参数即设置是否为守护进程，默认为true

之后设置定时任务中传入定时定时统计服务提供者请求状态信息的任务runPinger方法即定时执行并统计服务状态的更新

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-3CpSn9rD-1647352907554)(C:\Users\HASEE\AppData\Roaming\Typora\typora-user-images\image-20220315205137109.png)]$