Add CombinedTransaction::try()

Author: Composite PHP

src/CombinedTransaction.php

@@ -54,6 +54,19 @@ public function delete(AbstractTable $table, AbstractEntity &$entity): void
         });
     }
 
+    /**
+     * @throws Exceptions\DbException
+     */
+    public function try(callable $callback): void
+    {
+        try {
+            $callback();
+        } catch (\Throwable $e) {
+            $this->rollback();
+            throw new Exceptions\DbException($e->getMessage(), 500, $e);
+        }
+    }
+
     public function rollback(): void
     {
         foreach ($this->transactions as $connection) {

tests/Table/CombinedTransactionTest.php

@@ -130,6 +130,20 @@ public function test_failedDelete(): void
         $this->assertNotNull($compositeTable->findOne($cEntity->user_id, $cEntity->post_id));
     }
 
+    public function test_try(): void
+    {
+        $compositeTable = new Tables\TestCompositeTable();
+        $entity = new Entities\TestCompositeEntity(user_id: mt_rand(1, 1000), post_id: mt_rand(1, 1000), message: 'Bar');;
+
+        try {
+            $transaction = new CombinedTransaction();
+            $transaction->save($compositeTable, $entity);
+            $transaction->try(fn() => throw new \Exception('test'));
+            $transaction->commit();
+        } catch (DbException) {}
+        $this->assertNull($compositeTable->findOne($entity->user_id, $entity->post_id));
+    }
+
     public function test_lockFailed(): void
     {
         $cache = new Helpers\FalseCache();