wip

umbrellio · Oct 28, 2024 · 47bd675 · 47bd675
1 parent 63d2ba3
commit 47bd675
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/lib/umbrellio_utils/control.rb b/lib/umbrellio_utils/control.rb
@@ -24,7 +24,13 @@ def retry_on_unique_violation(
       checked_constraints: [],
       &block
     )
-      return yield if !retry_on_all_constraints && checked_constraints.empty?
+      if !retry_on_all_constraints && checked_constraints.empty?
+        begin
+          return yield
+        rescue Sequel::UniqueConstraintViolation => e
+          raise UniqueConstraintViolation, e.message
+        end
+      end
 
       retry_on(Sequel::UniqueConstraintViolation, times:) do
         DB.transaction(savepoint: true, &block)