feat: batch sql queries

mjeanroy · Nov 14, 2023 · 3466046 · 3466046
1 parent 379ca89
commit 3466046
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/src/main/java/com/github/mjeanroy/dbunit/core/jdbc/JdbcUtils.java b/src/main/java/com/github/mjeanroy/dbunit/core/jdbc/JdbcUtils.java
@@ -83,19 +83,17 @@ static <T> List<T> executeQuery(Connection connection, String query, ResultSetMa
 	}
 
 	static void executeUpdates(Connection connection, Collection<String> queries) {
-		log.debug("Executing queries: {}", queries);
-		for (String query : queries) {
-			executeUpdate(connection, query);
-		}
-	}
-
-	static void executeUpdate(Connection connection, String query) {
-		log.debug("Executing update: {}", query);
+		log.debug("Executing batched queries: {}", queries);
 		try (Statement statement = connection.createStatement()) {
-			statement.executeUpdate(query);
+			for (String query : queries) {
+				log.debug("Adding batched query: {}", query);
+				statement.addBatch(query);
+			}
+
+			statement.executeBatch();
 		}
 		catch (SQLException ex) {
-			throw new JdbcException("Cannot execute query: " + query, ex);
+			throw new JdbcException("Cannot execute queries: " + queries, ex);
 		}
 	}