如何测试ConcurrentHashMap使用的线程安全性？代码存疑

你的代码确实存在线程安全问题，因为putIfAbsent之后的get、判断自增、replace这一系列操作并非原子性的。多个线程并发执行时，会出现竞态条件：比如两个线程同时读取到同一个val值，各自完成自增后，后续的replace操作会互相覆盖，直接丢失一次自增更新。

1. 基础并发测试代码 #

用线程池模拟多线程并发调用，配合CountDownLatch控制线程执行节奏，验证最终结果是否符合预期：

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class ConcurrentMapTest {
    private static ConcurrentHashMap<String, Integer> somemap = new ConcurrentHashMap<>();
    private static final int total_count = 1000;
    private static final int CONCURRENT_THREADS = 500; // 模拟500个并发线程

    public static boolean some_method() {
        // 补充原代码缺失的初始值定义，假设初始为0
        somemap.putIfAbsent("somekey", 0);
        final int val = somemap.get("somekey");
        int newVal = val;
        if (val < total_count) {
            newVal++;
        }
        somemap.replace("somekey", newVal);
        return true;
    }

    public static void main(String[] args) throws InterruptedException {
        ExecutorService executor = Executors.newFixedThreadPool(CONCURRENT_THREADS);
        CountDownLatch latch = new CountDownLatch(CONCURRENT_THREADS);

        for (int i = 0; i < CONCURRENT_THREADS; i++) {
            executor.submit(() -> {
                try {
                    some_method();
                } finally {
                    latch.countDown();
                }
            });
        }

        latch.await(); // 等待所有线程执行完成
        executor.shutdown();

        System.out.println("实际最终值: " + somemap.get("somekey"));
        System.out.println("预期最终值: " + CONCURRENT_THREADS);
    }
}

2. 结果判断 #

运行后如果实际最终值小于CONCURRENT_THREADS（比如500），就说明线程安全问题确实存在——多个线程的更新操作被互相覆盖了。

3. 进阶强化测试 #

如果想更严格验证，可以让每个线程循环调用some_method多次，比如每个线程调用100次，总调用次数为500*100=50000（确保total_count大于这个数）。如果最终结果远小于50000，就能明确证明竞态条件的影响。

要解决这个问题，需要把get-判断-更新的流程变成原子操作，推荐使用ConcurrentHashMap自带的原子方法：

// 用compute实现原子更新逻辑
somemap.compute("somekey", (key, oldVal) -> {
    if (oldVal == null) {
        return 0; // 设置初始值
    }
    return oldVal < total_count ? oldVal + 1 : oldVal;
});

内容的提问来源于stack exchange，提问作者Sourav gupta