DataStax Java Driver 4.x如何实现TokenAware负载均衡策略

• 不需要额外编写代码实现令牌感知能力，直接使用默认的DefaultLoadBalancingPolicy就会自动处理令牌感知的负载均衡逻辑。

版本设计差异 #

3.x版本的驱动采用装饰器模式拆分负载均衡能力，你之前代码里用的TokenAwarePolicy就是典型的装饰器，用来给底层的轮询、DC感知策略附加令牌路由能力。4.x版本对负载均衡模块做了重构，没有再提供独立的TokenAware策略类，所有生产环境常用的能力都直接内置整合到了默认策略中。

默认策略的实际行为 #

你之前3.x代码里实现的两种策略组合，DefaultLoadBalancingPolicy已经全部覆盖：

• 内置数据中心感知能力：默认优先将请求路由到本地数据中心的节点，不会随意跨DC发请求，和你之前使用的DCAwareRoundRobinPolicy行为一致，本地DC如果不手动指定，驱动也会通过接触点节点自动识别。
• 令牌感知默认强制开启：执行CQL语句时会自动计算分区键对应的token值，优先将请求直接发往持有对应数据副本的节点，减少节点间的转发开销，和你之前手动包装TokenAwarePolicy的效果完全一致。
• 兜底逻辑采用同DC节点轮询：如果遇到副本节点故障、或者执行的是不带分区键的批量/聚合类语句无法路由到特定副本时，会自动在同DC的健康节点间轮询分发请求，和你之前的RoundRobinPolicy兜底逻辑一致。

4.x版本对应实现参考 #

你之前的自定义负载均衡逻辑在4.x里可以大幅简化，如果没有特殊定制需求，甚至不需要手动配置负载均衡策略，驱动初始化时默认加载的就是带DC感知、令牌感知的DefaultLoadBalancingPolicy。如果要保留你之前的开关逻辑（生产环境不建议关闭令牌感知），可以参考如下写法：

import com.datastax.oss.driver.api.core.CqlSessionBuilder;
import com.datastax.oss.driver.api.core.config.DefaultDriverOption;
import com.datastax.oss.driver.api.core.config.DriverConfigLoader;
import com.datastax.oss.driver.internal.core.loadbalancing.DefaultLoadBalancingPolicy;

public static void configLoadBalancingPolicy(CqlSessionBuilder sessionBuilder, String loadBalanceStr, boolean isTokenAware) {
    DriverConfigLoader configLoader = DriverConfigLoader.programmaticBuilder()
            // 令牌感知默认即为开启状态，仅当需要关闭时才配置该参数
            .withBoolean(DefaultDriverOption.LOAD_BALANCING_POLICY_TOKEN_AWARE, isTokenAware)
            // 多DC部署场景建议手动指定本地DC名，避免驱动自动识别出错
            // .withString(DefaultDriverOption.LOAD_BALANCING_LOCAL_DATACENTER, "your-local-dc-name")
            .withClass(DefaultDriverOption.LOAD_BALANCING_POLICY_CLASS, DefaultLoadBalancingPolicy.class)
            .build();
    sessionBuilder.withConfigLoader(configLoader);
}

补充提示：令牌感知特性仅对携带完整分区键的语句做路由优化，不会影响无路由键语句的正常执行，开启后只有性能收益，没有额外副作用，不建议手动关闭。

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