diff --git a/.github/workflows/performance-tests-pipeline-options/JDBC.txt b/.github/workflows/performance-tests-pipeline-options/JDBC.txt index 74fd85145c59..49055d30028a 100644 --- a/.github/workflows/performance-tests-pipeline-options/JDBC.txt +++ b/.github/workflows/performance-tests-pipeline-options/JDBC.txt @@ -26,4 +26,4 @@ --postgresSsl=false --postgresPort=5432 --autoscalingAlgorithm=NONE ---numWorkers=5 \ No newline at end of file +--numWorkers=5 diff --git a/sdks/java/io/jdbc/src/test/java/org/apache/beam/sdk/io/jdbc/JdbcIOIT.java b/sdks/java/io/jdbc/src/test/java/org/apache/beam/sdk/io/jdbc/JdbcIOIT.java index 5f53d3243001..07d94952ddb7 100644 --- a/sdks/java/io/jdbc/src/test/java/org/apache/beam/sdk/io/jdbc/JdbcIOIT.java +++ b/sdks/java/io/jdbc/src/test/java/org/apache/beam/sdk/io/jdbc/JdbcIOIT.java @@ -210,7 +210,7 @@ private PipelineResult runWrite() { .apply(ParDo.of(new TimeMonitor<>(NAMESPACE, "write_time"))) .apply( JdbcIO.write() - .withDataSourceConfiguration(JdbcIO.DataSourceConfiguration.create(dataSource)) + .withDataSourceConfiguration(JdbcIO.DataSourceConfiguration.create(dataSource).withMaxConnections(4)) .withStatement(String.format("insert into %s values(?, ?)", tableName)) .withPreparedStatementSetter(new JdbcTestHelper.PrepareStatementFromTestRow())); @@ -281,7 +281,7 @@ public void testWriteWithAutosharding() throws Exception { .apply(ParDo.of(new TimeMonitor<>(NAMESPACE, "write_time"))) .apply( JdbcIO.write() - .withDataSourceConfiguration(JdbcIO.DataSourceConfiguration.create(dataSource)) + .withDataSourceConfiguration(JdbcIO.DataSourceConfiguration.create(dataSource).withMaxConnections(4)) .withStatement(String.format("insert into %s values(?, ?)", firstTableName)) .withAutoSharding() .withPreparedStatementSetter(new JdbcTestHelper.PrepareStatementFromTestRow())); @@ -421,7 +421,7 @@ public void testWriteWithWriteResults() throws Exception { */ private static JdbcIO.Write> getJdbcWriteWithReturning(String tableName) { return JdbcIO.>write() - .withDataSourceConfiguration(JdbcIO.DataSourceConfiguration.create(dataSource)) + .withDataSourceConfiguration(JdbcIO.DataSourceConfiguration.create(dataSource).withMaxConnections(4)) .withStatement(String.format("insert into %s values(?, ?) returning *", tableName)) .withPreparedStatementSetter( (element, statement) -> {