You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

并发环境下轮询负载均衡器实现失败排查求助

问题分析与解决方案

核心问题1:测试逻辑错误

你错误假设并发任务的执行顺序与提交顺序完全一致,但线程池中的任务是并行调度执行的,任务完成顺序不依赖于提交顺序。比如你按顺序提交30个任务,实际可能先执行第3个、再执行第2个,导致Future队列的结果顺序并非严格的1→2→3循环,直接触发断言失败。

核心问题2:RoundRobinStrategy的潜在严谨性问题(测试未触发,但需优化)

虽然get()方法持有读锁,保证策略执行期间实例列表大小不会变化,但currentIndex.getAndUpdate()的lambda表达式中直接访问instances.size(),逻辑上可以优化为提前缓存size值,避免在原子操作逻辑中重复访问列表,让代码更严谨。


修复方案

1. 修改测试逻辑,验证轮询的统计特性

替换原有的顺序断言,改为统计每个实例的返回次数,或验证结果序列符合轮询的循环规律:

@RepeatedTest(100)
void should_keep_round_robin_pattern_in_concurrent_environment() throws InterruptedException, ExecutionException {
    // given
    loadBalancerRoundRobin.register("1");
    loadBalancerRoundRobin.register("2");
    loadBalancerRoundRobin.register("3");
    Queue<Future<String>> objects = new ArrayBlockingQueue<>(30);

    int threads = Runtime.getRuntime().availableProcessors();
    var executorService = Executors.newScheduledThreadPool(threads);
    try {
        for (int i = 0; i < 30; i++) {
            Future<String> submit = executorService.submit(() -> loadBalancerRoundRobin.get());
            objects.add(submit);
        }
    } finally {
        executorService.shutdown();
        executorService.awaitTermination(1, TimeUnit.MINUTES);
    }

    // 统计每个实例的返回次数
    Map<String, Integer> countMap = new HashMap<>();
    while (!objects.isEmpty()) {
        String result = objects.remove().get();
        countMap.put(result, countMap.getOrDefault(result, 0) + 1);
    }

    // 验证每个实例被返回10次(30/3)
    assertEquals(10, countMap.get("1"));
    assertEquals(10, countMap.get("2"));
    assertEquals(10, countMap.get("3"));
}

2. 优化RoundRobinStrategy的索引计算

提前缓存实例列表大小,避免在原子更新逻辑中直接访问列表:

class RoundRobinStrategy implements LoadBalancerStrategy {

    private final AtomicInteger currentIndex = new AtomicInteger(0);

    @Override
    public String get(List<String> instances) {
        if (instances.isEmpty()) {
            return null;
        }
        int size = instances.size();
        int index = currentIndex.getAndUpdate(i -> (i + 1) % size);
        return instances.get(index);
    }
}

额外说明

你的LoadBalancer锁机制是正确的:读写锁保证了注册(写操作)与获取(读操作)的线程安全,AtomicInteger也确保了currentIndex的原子更新。测试失败的核心原因是错误假设了并发任务的执行顺序与提交顺序一致,这在多线程环境下不成立。

内容的提问来源于stack exchange,提问作者Anonymous9991

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.06.18 17:35:19