From 051962b54d9a647adc8c09fb8ef33db2ac9b659a Mon Sep 17 00:00:00 2001
From: Christopher Baines <mail@cbaines.net>
Date: Sun, 5 May 2019 13:35:17 +0100
Subject: Add a with-postgresql-transaction procedure

To help with running tests.
---
 guix-data-service/database.scm | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/guix-data-service/database.scm b/guix-data-service/database.scm
index ba7cd64..ab6758b 100644
--- a/guix-data-service/database.scm
+++ b/guix-data-service/database.scm
@@ -17,7 +17,8 @@
 
 (define-module (guix-data-service database)
   #:use-module (squee)
-  #:export (with-postgresql-connection))
+  #:export (with-postgresql-connection
+            with-postgresql-transaction))
 
 ;; TODO This isn't exported for some reason
 (define pg-conn-finish
@@ -38,3 +39,16 @@
       (lambda (key . args)
         (pg-conn-finish conn)))))
 
+(define* (with-postgresql-transaction conn f
+                                      #:key always-rollback?)
+  (exec-query conn "BEGIN;")
+
+  (with-throw-handler #t
+    (lambda ()
+      (let ((result (f conn)))
+        (exec-query conn (if always-rollback?
+                             "ROLLBACK;"
+                             "COMMIT;"))
+        result))
+    (lambda (key . args)
+      (exec-query conn "ROLLBACK;"))))
-- 
cgit v1.2.3