add transaction support to DatabaseHelper (#556)

airbytehq · Oct 13, 2020 · 8dd3933 · 8dd3933
1 parent 0fffd07
commit 8dd3933
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/airbyte-db/src/main/java/io/airbyte/db/DatabaseHelper.java b/airbyte-db/src/main/java/io/airbyte/db/DatabaseHelper.java
@@ -46,14 +46,24 @@ public static BasicDataSource getConnectionPool(String username,
     return connectionPool;
   }
 
-  public static <T> T query(BasicDataSource connectionPool, ContextQueryFunction<T> transform)
-      throws SQLException {
+  public static <T> T query(BasicDataSource connectionPool, ContextQueryFunction<T> transform) throws SQLException {
+
     try (Connection connection = connectionPool.getConnection()) {
       DSLContext context = getContext(connection);
       return transform.apply(context);
     }
   }
 
+  public static <T> T transaction(BasicDataSource connectionPool, ContextQueryFunction<T> transform) throws SQLException {
+    try (Connection connection = connectionPool.getConnection()) {
+      DSLContext context = getContext(connection);
+      return context.transactionResult(configuration -> {
+        DSLContext transactionContext = DSL.using(configuration);
+        return transform.apply(transactionContext);
+      });
+    }
+  }
+
   private static DSLContext getContext(Connection connection) {
     return DSL.using(connection, SQLDialect.POSTGRES);
   }