Added new traits.

nomadicjosh · nomadicjosh · commit 71ec1ee1b10e · 2025-08-13T13:40:47.000-07:00
Signed-off-by: Joshua Parker &lt;joshua@joshuaparker.dev&gt;
diff --git a/Support/Traits/ContainerAware.php b/Support/Traits/ContainerAware.php
@@ -0,0 +1,15 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Codefy\Framework\Support\Traits;
+
+use Codefy\Framework\Codefy;
+
+trait ContainerAware
+{
+    public static function make(): static
+    {
+        return Codefy::$PHP->make(name: static::class);
+    }
+}
diff --git a/Support/Traits/DbTransactionsAware.php b/Support/Traits/DbTransactionsAware.php
@@ -0,0 +1,68 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Codefy\Framework\Support\Traits;
+
+use Codefy\Framework\Codefy;
+use Qubus\Exception\Exception;
+
+trait DbTransactionsAware
+{
+    /**
+     * Determines whether class uses transaction.
+     */
+    protected bool $useTransaction = false;
+
+    /**
+     * Enable transaction in pipeline.
+     */
+    public function withTransaction(): static
+    {
+        $this->useTransaction = true;
+
+        return $this;
+    }
+
+    /**
+     * Begin the transaction if enabled.
+     *
+     * @throws Exception
+     */
+    protected function beginTransaction(): void
+    {
+        if (!$this->useTransaction) {
+            return;
+        }
+
+        Codefy::$PHP->getDB()->beginTransaction();
+    }
+
+    /**
+     * Commit the transaction if enabled.
+     *
+     * @throws Exception
+     */
+    protected function commitTransaction(): void
+    {
+        if (!$this->useTransaction) {
+            return;
+        }
+
+        Codefy::$PHP->getDB()->commit();
+    }
+
+    /**
+     * Rollback the transaction if enabled.
+     *
+     * @throws Exception
+     */
+    protected function rollbackTransaction(): void
+    {
+        if (!$this->useTransaction) {
+            return;
+        }
+
+        Codefy::$PHP->getDB()->rollback();
+    }
+}
