如何为并行执行的Cucumber线程分配独立PostgreSQL数据库？

1. 用ThreadLocal管理线程专属资源 #

用ThreadLocal存储每个线程对应的PostgreSQL容器实例和数据源配置，从根源避免线程间资源干扰：

public class ThreadLocalDbContainer {
    private static final ThreadLocal<PostgreSQLContainer<?>> CONTAINER_HOLDER = new ThreadLocal<>();
    private static final ThreadLocal<String> DB_URL_HOLDER = new ThreadLocal<>();

    public static PostgreSQLContainer<?> getContainer() {
        return CONTAINER_HOLDER.get();
    }

    public static void setContainer(PostgreSQLContainer<?> container) {
        CONTAINER_HOLDER.set(container);
    }

    public static String getDbUrl() {
        return DB_URL_HOLDER.get();
    }

    public static void setDbUrl(String dbUrl) {
        DB_URL_HOLDER.set(dbUrl);
    }

    public static void clear() {
        PostgreSQLContainer<?> container = CONTAINER_HOLDER.get();
        if (container != null && container.isRunning()) {
            container.stop();
        }
        CONTAINER_HOLDER.remove();
        DB_URL_HOLDER.remove();
    }
}

2. 动态创建线程专属数据库容器 #

通过Cucumber Hook为每个线程初始化独立容器，绑定随机端口避免冲突：

public class DbHooks {
    @BeforeAll
    public static void initThreadDb() {
        if (ThreadLocalDbContainer.getContainer() == null) {
            PostgreSQLContainer<?> container = new PostgreSQLContainer<>("postgres:15-alpine")
                    .withDatabaseName("test_db_" + Thread.currentThread().getId())
                    .withUsername("test_user")
                    .withPassword("test_pass");
            container.start();
            
            ThreadLocalDbContainer.setContainer(container);
            String dbUrl = String.format("jdbc:postgresql://%s:%d/%s",
                    container.getHost(), container.getFirstMappedPort(), container.getDatabaseName());
            ThreadLocalDbContainer.setDbUrl(dbUrl);
            
            // 覆盖Spring数据源配置
            System.setProperty("spring.datasource.url", dbUrl);
            System.setProperty("spring.datasource.username", container.getUsername());
            System.setProperty("spring.datasource.password", container.getPassword());
        }
    }

    @AfterAll
    public static void cleanThreadDb() {
        ThreadLocalDbContainer.clear();
    }
}

如果测试基于Spring Boot，也可以用自定义上下文初始化器替换数据源配置：

public class ThreadDbInitializer implements ApplicationContextInitializer<ConfigurableApplicationContext> {
    @Override
    public void initialize(ConfigurableApplicationContext applicationContext) {
        if (ThreadLocalDbContainer.getContainer() == null) {
            PostgreSQLContainer<?> container = new PostgreSQLContainer<>("postgres:15-alpine")
                    .withDatabaseName("test_db_" + Thread.currentThread().getId())
                    .withUsername("test_user")
                    .withPassword("test_pass");
            container.start();
            
            ThreadLocalDbContainer.setContainer(container);
            String dbUrl = String.format("jdbc:postgresql://%s:%d/%s",
                    container.getHost(), container.getFirstMappedPort(), container.getDatabaseName());
            ThreadLocalDbContainer.setDbUrl(dbUrl);
            
            ConfigurableEnvironment environment = applicationContext.getEnvironment();
            Map<String, Object> props = new HashMap<>();
            props.put("spring.datasource.url", dbUrl);
            props.put("spring.datasource.username", container.getUsername());
            props.put("spring.datasource.password", container.getPassword());
            environment.getPropertySources().addFirst(new MapPropertySource("threadDbProps", props));
        }
    }
}

在测试配置类上添加：

@ContextConfiguration(initializers = ThreadDbInitializer.class)

3. 调整Cucumber并行配置强化线程隔离 #

保留现有并行配置，新增上下文隔离策略，确保每个线程使用独立Spring上下文：

@CucumberContextConfiguration
@SpringBootTest
@ConfigurationParameter(key = Constants.PARALLEL_EXECUTION_ENABLED_PROPERTY_NAME, value = "true")
@ConfigurationParameter(key = Constants.EXECUTION_MODE_FEATURE_PROPERTY_NAME, value = "same_thread")
@ConfigurationParameter(key = Constants.PARALLEL_CONFIG_FIXED_PARALLELISM_PROPERTY_NAME, value = "4")
@ConfigurationParameter(key = Constants.GLUE_EXECUTION_CONTEXT_STRATEGY_PROPERTY_NAME, value = "THREAD")
public class CucumberTestConfig {
}

4. 测试数据隔离补充 #

每个线程的数据库独立，可按需在测试前清空数据：

@Before
public void cleanTestData() {
    DataSource dataSource = ApplicationContextProvider.getApplicationContext().getBean(DataSource.class);
    try (Connection conn = dataSource.getConnection()) {
        String cleanSql = "TRUNCATE TABLE your_table RESTART IDENTITY CASCADE;";
        conn.createStatement().execute(cleanSql);
    } catch (SQLException e) {
        throw new RuntimeException("Failed to clean test data", e);
    }
}

注意事项 #

• 测试基类不要硬编码数据源，依赖Spring注入的实例
• 控制并行数（当前设置为4），避免过多容器占用系统资源
• 确保Docker资源分配足够（CPU、内存），避免容器启动失败

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