TestNG数据并行场景下HashMap测试数据被并行线程重复取用问题咨询

HashMap本身属于非线程安全容器，你没有对数据的取用、标记逻辑做并发控制，多个线程同时筛选未使用数据时会触发竞态条件：两个线程可能同时判断同一条数据为未使用状态，先后获取执行，最终导致用例重复执行失败。

方案1：用并发队列实现无锁取数（最推荐） #

直接把HashMap中的所有测试数据提前存入JDK自带的并发队列，队列的poll()方法会自动移除取出的元素，天然保证不会重复分配，不需要额外加锁，性能最优。
示例代码：

import java.util.concurrent.ConcurrentLinkedQueue;

// 初始化阶段执行，把HashMap的value全部存入队列
ConcurrentLinkedQueue<TestData> testDataQueue = new ConcurrentLinkedQueue<>(testHashMap.values());

// 测试方法中取数逻辑
public void yourTestMethod() {
    TestData availableData = testDataQueue.poll();
    if (availableData == null) {
        // 所有数据已分配完成，可直接结束测试或返回提示
        return;
    }
    // 用availableData执行测试逻辑
}

方案2：保留HashMap结构+同步控制 #

如果必须保留HashMap存储结构，可给测试数据新增使用状态标记，对取数逻辑加同步锁，保证同一时间只有一个线程可以筛选、标记数据，避免重复获取。
示例代码：

// 测试数据类新增已使用标记
class TestData {
    // 原有业务字段
    private boolean isUsed = false;
    // 对应get、set方法
}

// 新增同步取数方法
private synchronized TestData getUnusedData() {
    for (TestData data : testHashMap.values()) {
        if (!data.isUsed()) {
            data.setUsed(true);
            return data;
        }
    }
    // 无可用数据返回空
    return null;
}

// 测试方法中调用取数
public void yourTestMethod() {
    TestData availableData = getUnusedData();
    if (availableData == null) return;
    // 执行测试逻辑
}

注意：该方案因为加了方法级同步锁，并发性能会低于方案1，仅适合数据量较小的场景使用。

方案3：复用TestNG原生DataProvider并行能力 #

TestNG自带的DataProvider支持并行数据分发，你只需要把测试数据导入DataProvider，框架会自动给每个线程分配不重复的测试数据，不需要手动实现取数逻辑，维护成本最低。
示例代码：

// 定义并行DataProvider
@DataProvider(name = "testDataProvider", parallel = true)
public Object[][] getTestData() {
    Object[][] dataArr = new Object[testHashMap.size()][1];
    int index = 0;
    for (TestData data : testHashMap.values()) {
        dataArr[index++][0] = data;
    }
    return dataArr;
}

// 测试方法绑定DataProvider
@Test(dataProvider = "testDataProvider")
public void yourTestMethod(TestData data) {
    // 直接使用传入的data执行测试，框架保证不会重复分配
}

需要同步调整TestNG套件配置，将并行模式设置为methods即可生效。

• 不要在无同步控制的场景下并发修改HashMap，除了数据重复外，还可能触发HashMap扩容死循环、数据丢失等更严重的问题。
• 如果需要留存已使用的数据，不要直接删除HashMap中的数据，可在取数后单独存储已使用数据，或者保留使用标记做后续校验。

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