feat: call the database clean within a transaction

pact-foundation · Dec 2, 2021 · 408c84e · 408c84e
1 parent 34c989c
commit 408c84e
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/lib/pact_broker/db/clean_incremental.rb b/lib/pact_broker/db/clean_incremental.rb
@@ -66,13 +66,15 @@ def call
         if dry_run?
           dry_run_results
         else
-          before_counts = current_counts
-          PactBroker::Domain::Version.where(id: resolve_ids(version_ids_to_delete)).delete
-          delete_orphan_pact_versions
-          after_counts = current_counts
-
-          TABLES.each_with_object({}) do | table_name, comparison_counts |
-            comparison_counts[table_name.to_s] = { "deleted" => before_counts[table_name] - after_counts[table_name], "kept" => after_counts[table_name] }
+          db.transaction do
+            before_counts = current_counts
+            PactBroker::Domain::Version.where(id: resolve_ids(version_ids_to_delete)).delete
+            delete_orphan_pact_versions
+            after_counts = current_counts
+
+            TABLES.each_with_object({}) do | table_name, comparison_counts |
+              comparison_counts[table_name.to_s] = { "deleted" => before_counts[table_name] - after_counts[table_name], "kept" => after_counts[table_name] }
+            end
           end
         end
       end