Cassandra Java驱动4.x可用负载均衡策略及旧策略兼容实现方法

4.x版本内置可用负载均衡策略 #

4.x版本对负载均衡模块做了架构重构，没有沿用3.x版本拆分单一职责策略的设计，核心内置策略如下：

• DefaultLoadBalancingPolicy：官方推荐生产使用的默认策略，原生整合数据中心感知、令牌感知、轮询调度、慢/故障节点自动规避能力，默认优先访问本地数据中心节点，仅在本地节点全部不可用时才会访问远程DC节点，完全适配DSE 6.8版本。
• DcInferringLoadBalancingPolicy：和默认策略能力一致，区别是不需要手动指定本地数据中心名称，会从首次成功连接的节点自动推断所属DC作为本地DC，适合部署环境DC不固定的场景。
• BasicLoadBalancingPolicy：最基础的轮询策略，不做数据中心感知、故障节点特殊逻辑，会把所有集群节点纳入轮询范围，适合需要自定义策略组合的场景。
  除此之外还提供了可用于包装基础策略的增强策略，包括实现令牌感知路由的TokenAwarePolicy、实现故障节点临时剔除的ErrorAwarePolicy、实现自定义节点过滤的NodeFilterPolicy，可以按需组合使用。

3.x版本RoundRobinPolicy、DCAwareRoundRobinPolicy兼容实现方案 #

4.x版本没有保留和3.x同名的这两个策略类，但是可以通过内置策略1:1还原原有行为，不需要自己从零实现逻辑：

• DCAwareRoundRobinPolicy 等价实现：直接实例化DefaultLoadBalancingPolicy，传入本地DC名称即可，默认行为和3.x的DCAwareRoundRobinPolicy完全一致，优先轮询本地DC节点，远程节点作为兜底。
• RoundRobinPolicy 等价实现：直接实例化BasicLoadBalancingPolicy即可，该策略会无视数据中心划分，对所有集群节点做轮询调度，和3.x的RoundRobinPolicy行为完全匹配。

原有代码改造示例 #

针对提供的3.x版本代码，改造后适配4.12版本的实现如下，注意导入包时要使用4.x版本的com.datastax.oss.driver开头的包路径，不要混入3.x的com.datastax.driver包：

import com.datastax.oss.driver.api.core.config.DriverExecutionProfile;
import com.datastax.oss.driver.api.core.loadbalancing.LoadBalancingPolicy;
import com.datastax.oss.driver.internal.core.loadbalancing.BasicLoadBalancingPolicy;
import com.datastax.oss.driver.internal.core.loadbalancing.DefaultLoadBalancingPolicy;
import com.datastax.oss.driver.internal.core.loadbalancing.policies.TokenAwarePolicy;

public static LoadBalancingPolicy getLoadBalancingPolicy(String loadBalanceStr, boolean isTokenAware, DriverExecutionProfile profile, String localDc) {
    LoadBalancingPolicy basePolicy = loadBalanceDataConvert(loadBalanceStr, profile, localDc);
    // 注意：DefaultLoadBalancingPolicy本身已经内置令牌感知能力，不需要重复包装
    if (isTokenAware && !(basePolicy instanceof DefaultLoadBalancingPolicy)) {
        return new TokenAwarePolicy(basePolicy);
    }
    return basePolicy;
}

private static LoadBalancingPolicy loadBalanceDataConvert(String loadBalanceStr, DriverExecutionProfile profile, String localDc) {
    if (CassandraConstants.CASSANDRACONNECTION_LOADBALANCEPOLICY_DC.equals(loadBalanceStr)) {
        // 等价3.x DCAwareRoundRobinPolicy
        return new DefaultLoadBalancingPolicy(profile, localDc);
    } else if (CassandraConstants.CASSANDRACONNECTION_LOADBALANCEPOLICY_ROUND.equals(loadBalanceStr)) {
        // 等价3.x RoundRobinPolicy
        return new BasicLoadBalancingPolicy(profile);
    }
    return null;
}

注意：4.x版本的负载均衡策略初始化必须传入DriverExecutionProfile实例，该实例可以在构建CqlSession时从会话的配置对象中直接获取，不需要单独手动构造。如果没有特殊的动态切换策略的需求，也可以直接在驱动配置文件中指定负载均衡策略实现类，不需要手动编码实例化。

内容的提问来源于stack exchange，提问作者prathyusha magam