Question

我正在尝试为连接到数据库的应用程序创建测试。 DataSource是一个连接池（Hikari）。

这是我的测试配置：

@Configuration
public class SqlTestConfig {

    @Bean
    DataSource dataSource() {
        HikariConfig config = new HikariConfig();
        config.setMaximumPoolSize(2);
        config.setDriverClassName("com.microsoft.sqlserver.jdbc.SQLServerDriver");
        config.setJdbcUrl("jdbc:sqlserver://serversql:1433;database=myDatabase");
        config.setUsername("user");
        config.setPassword("password");
        return new HikariDataSource(config);
    }
}

这是我的测试类：

@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest(classes = SqlTestConfig.class)
@Slf4j
@Sql(
        scripts = "/clearTables.sql",
        config = @SqlConfig(separator = "GO")
)
public class SqlTest {

    @Autowired
    DataSource dataSource;

    @Test
    public void test1() throws SQLException {
        log.info("catalog:" + dataSource.getConnection().getCatalog());
    }

    @Test
    public void test2() throws SQLException {
        log.info("catalog:" + dataSource.getConnection().getCatalog());
    }

    @Test
    public void test3() throws SQLException {
        log.info("catalog:" + dataSource.getConnection().getCatalog());
    }

    @Test
    public void test4() throws SQLException {
        log.info("catalog:" + dataSource.getConnection().getCatalog());
    }
}

请注意，MaximumPoolSize设置为2.当我运行测试类时，前两个测试成功完成，其余测试失败，因为池连接耗尽（连接超时）。

我认为问题是因为@Sql注释导致创建DataSourceInitializer -s来执行清理脚本，但连接永远不会返回到池中。

当我将MaximumPoolSize设置为4时，所有测试都成功完成。我无法判断我是否发生了配置错误，或者这是否是Spring中的错误。

Answer 1

getConnection从底层池获取连接。更改测试以正确关闭所获取的连接，如下所示：

@Test
public void test1() throws SQLException {
    try (Connection connection = dataSource.getConnection()) {
        log.info("catalog:" + connection.getCatalog());
    }
}

使用连接池

1 个答案: