拥有800个自动化测试用例，如何减少数据库连接的启停次数？

嘿，这个问题我之前在优化自动化测试框架的时候也踩过类似的坑！频繁创建销毁数据库连接不仅拖慢测试执行速度，还会给数据库服务器带来不必要的负载。咱们先聊聊你之前遇到的“一次连接跑所有用例出问题”的可能原因，再给出几个可行的优化方案：

先分析下“单连接跑全量用例”出问题的常见原因 #

大概率是用例之间的状态污染或者连接生命周期问题：

• 某个用例修改了数据（比如插入/更新）但没回滚，导致后续用例的测试数据不符合预期
• 会话级别的变量（比如时区、隔离级别）被修改后没重置，影响后续用例
• 数据库服务器长时间无操作自动断开连接（超时设置）
• 用例执行中持有锁未释放，阻塞后续用例

可行的优化方案 #

1. 用数据库连接池（最推荐） #

连接池是解决这类问题的标准方案：初始化时创建一批连接放在池中，每个测试用例从池里获取连接，执行完查询后归还连接（不是关闭），避免重复创建销毁的开销。

实现思路：

• 在测试框架的全局初始化阶段（比如所有用例执行前）初始化连接池
• 每个用例执行时调用pool.getConnection()获取连接
• 用例执行完成后调用conn.close()（注意：连接池的close()其实是把连接放回池里，不是真的关闭）
• 所有用例执行完后，关闭连接池释放所有连接

伪代码示例：

// 全局初始化连接池
private static DataSource dataSource;

@BeforeAll
public static void initPool() {
    HikariConfig config = new HikariConfig();
    config.setJdbcUrl("your_db_url");
    config.setUsername("user");
    config.setPassword("pass");
    config.setMaximumPoolSize(10); // 根据测试并发数调整
    dataSource = new HikariDataSource(config);
}

@Test
public void testCase1() {
    try (Connection conn = dataSource.getConnection()) {
        // 执行查询、断言逻辑
    } catch (SQLException e) {
        // 异常处理
    }
}

@AfterAll
public static void closePool() {
    ((HikariDataSource) dataSource).close();
}

2. 测试类级别的连接复用 #

如果不想引入连接池，可以利用测试框架的生命周期钩子，让同一个测试类里的所有用例共享一个连接：

• 在@BeforeClass（JUnit 4）或@BeforeAll（JUnit 5）中创建连接
• 在每个用例执行后清理连接状态：比如回滚事务、重置会话变量
• 在@AfterClass/@AfterAll中关闭连接

注意事项：

• 每个用例必须在独立的事务中执行，执行完立即回滚，避免数据残留
• 如果用例需要提交数据（比如测试持久化逻辑），可以在@AfterEach里手动清理数据（比如删除插入的记录）

示例：

private static Connection conn;

@BeforeAll
public static void initConnection() throws SQLException {
    conn = DriverManager.getConnection("url", "user", "pass");
    conn.setAutoCommit(false); // 关闭自动提交，手动控制事务
}

@AfterEach
public void cleanUp() throws SQLException {
    conn.rollback(); // 回滚当前事务，清除用例产生的数据
    // 重置会话变量，比如：
    conn.execute("SET SESSION time_zone = '+00:00'");
}

@Test
public void testCase1() throws SQLException {
    // 执行查询、修改操作
}

@AfterAll
public static void closeConnection() throws SQLException {
    conn.close();
}

3. 分批次复用连接 #

如果全量用例共享一个连接风险太高，可以把800个用例分成若干批次（比如每50-100个一批），每个批次共用一个连接：

• 批次开始时创建连接
• 批次内每个用例执行后清理状态（回滚/重置）
• 批次结束后关闭连接

这种方案既能大幅减少连接创建次数，又能降低单连接长时间运行的风险（比如超时、状态污染累积）。

4. 优化连接参数（针对单连接超时问题） #

如果之前单连接跑全量用例是因为超时被断开，可以调整数据库连接的参数：

• 设置autoReconnect=true（MySQL为例），让连接断开时自动重连
• 增加连接超时时间（比如connectTimeout、socketTimeout）
• 在每个用例执行前检查连接是否有效，无效则重新创建

总结 #

优先推荐数据库连接池方案，它既能高效复用连接，又能自动处理连接的生命周期和状态管理。如果暂时不想引入连接池，用测试类级别的连接复用+事务回滚也能解决问题，核心是要保证每个用例的执行环境是隔离的，避免状态污染。

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