sharding-jdbc源码之load-balance-algorithm-type

最新推荐文章于 2023-08-08 11:30:12 发布

kong-kong

最新推荐文章于 2023-08-08 11:30:12 发布

阅读量3.3k

点赞数

CC 4.0 BY-SA版权

分类专栏： shardingsphere

本文链接：https://blog.csdn.net/kq1983/article/details/102586945

shardingsphere 专栏收录该内容

3 篇文章

订阅专栏

本文深入解析了两种常见的负载均衡算法：轮询(round_robin)和随机(random)，并提供了这两种算法的具体实现方式，包括它们在主从数据库场景中的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


load-balance-algorithm-type: round_robin
load-balance-algorithm-type: random


# 算法接口
io.shardingjdbc.core.api.algorithm.masterslave.MasterSlaveLoadBalanceAlgorithm

# random 实现类
RandomMasterSlaveLoadBalanceAlgorithm


# round_robin 实现类
RoundRobinMasterSlaveLoadBalanceAlgorithm

MasterSlaveLoadBalanceAlgorithmType


# 默认是ROUND_ROBIN
public enum MasterSlaveLoadBalanceAlgorithmType {
    
    ROUND_ROBIN(new RoundRobinMasterSlaveLoadBalanceAlgorithm()),
    RANDOM(new RandomMasterSlaveLoadBalanceAlgorithm());
    
    private final MasterSlaveLoadBalanceAlgorithm algorithm;
    
    /**
     * Get default master-slave database load-balance algorithm type.
     * 
     * @return default master-slave database load-balance algorithm type
     */
    public static MasterSlaveLoadBalanceAlgorithmType getDefaultAlgorithmType() {
        return ROUND_ROBIN;
    }
}

RoundRobinMasterSlaveLoadBalanceAlgorithm



public final class RoundRobinMasterSlaveLoadBalanceAlgorithm implements MasterSlaveLoadBalanceAlgorithm {
    
    private static final ConcurrentHashMap<String, AtomicInteger> COUNT_MAP = new ConcurrentHashMap<>();
    
    @Override
    public String getDataSource(final String name, final String masterDataSourceName, final List<String> slaveDataSourceNames) {
        AtomicInteger count = COUNT_MAP.containsKey(name) ? COUNT_MAP.get(name) : new AtomicInteger(0); //默认从0开始
        COUNT_MAP.putIfAbsent(name, count);
        count.compareAndSet(slaveDataSourceNames.size(), 0); // slave如果有多个 到slave.size=count  设置为0  比如slave 有3个 则count=3的时候 设置为0
        return slaveDataSourceNames.get(Math.abs(count.getAndIncrement()) % slaveDataSourceNames.size()); // count每次都+1 slave从0开始取，到slave.size-1
    }
}

RandomMasterSlaveLoadBalanceAlgorithm



public final class RandomMasterSlaveLoadBalanceAlgorithm implements MasterSlaveLoadBalanceAlgorithm {
    
    @Override
    public String getDataSource(final String name, final String masterDataSourceName, final List<String> slaveDataSourceNames) {
        return slaveDataSourceNames.get(new Random().nextInt(slaveDataSourceNames.size()));
    }
}