From 651d294783e8f6fc3bdf1013cc76983a6f85ab7d Mon Sep 17 00:00:00 2001
From: Leonid Meleshin <hello@leon0399.ru>
Date: Fri, 27 Sep 2024 15:42:04 +0200
Subject: [PATCH] feat: dispatch transactions events

cc #96
---
 src/Driver/Driver.php                  |  9 +++++++++
 src/Driver/Postgres/PostgresDriver.php |  3 +++
 src/Event/AbstractDriverEvent.php      | 13 +++++++++++++
 src/Event/QueryExecuted.php            |  5 +++--
 src/Event/TransactionBeginning.php     | 10 ++++++++++
 src/Event/TransactionCommited.php      |  8 ++++++++
 src/Event/TransactionRolledBack.php    |  8 ++++++++
 src/EventDispatcherAwareTrait.php      |  2 +-
 8 files changed, 55 insertions(+), 3 deletions(-)
 create mode 100644 src/Event/AbstractDriverEvent.php
 create mode 100644 src/Event/TransactionBeginning.php
 create mode 100644 src/Event/TransactionCommited.php
 create mode 100644 src/Event/TransactionRolledBack.php

diff --git a/src/Driver/Driver.php b/src/Driver/Driver.php
index 533fb6f7..379041fc 100644
--- a/src/Driver/Driver.php
+++ b/src/Driver/Driver.php
@@ -16,6 +16,9 @@
 use Cycle\Database\Config\PDOConnectionConfig;
 use Cycle\Database\Config\ProvidesSourceString;
 use Cycle\Database\Event\QueryExecuted;
+use Cycle\Database\Event\TransactionBeginning;
+use Cycle\Database\Event\TransactionCommited;
+use Cycle\Database\Event\TransactionRolledBack;
 use Cycle\Database\EventDispatcherAwareInterface;
 use Cycle\Database\EventDispatcherAwareTrait;
 use Cycle\Database\Exception\DriverException;
@@ -358,6 +361,8 @@ public function beginTransaction(string $isolationLevel = null): bool
 
         $this->createSavepoint($this->transactionLevel);
 
+        $this->eventDispatcher?->dispatch(new TransactionBeginning($this));
+
         return true;
     }
 
@@ -399,6 +404,8 @@ public function commitTransaction(): bool
 
         $this->releaseSavepoint($this->transactionLevel + 1);
 
+        $this->eventDispatcher?->dispatch(new TransactionCommited($this));
+
         return true;
     }
 
@@ -436,6 +443,8 @@ public function rollbackTransaction(): bool
 
         $this->rollbackSavepoint($this->transactionLevel + 1);
 
+        $this->eventDispatcher?->dispatch(new TransactionRolledBack($this));
+
         return true;
     }
 
diff --git a/src/Driver/Postgres/PostgresDriver.php b/src/Driver/Postgres/PostgresDriver.php
index be60e157..d823b8da 100644
--- a/src/Driver/Postgres/PostgresDriver.php
+++ b/src/Driver/Postgres/PostgresDriver.php
@@ -19,6 +19,7 @@
 use Cycle\Database\Driver\Postgres\Query\PostgresInsertQuery;
 use Cycle\Database\Driver\Postgres\Query\PostgresSelectQuery;
 use Cycle\Database\Driver\Postgres\Query\PostgresUpdateQuery;
+use Cycle\Database\Event\TransactionBeginning;
 use Cycle\Database\Exception\DriverException;
 use Cycle\Database\Exception\StatementException;
 use Cycle\Database\Query\QueryBuilder;
@@ -180,6 +181,8 @@ public function beginTransaction(string $isolationLevel = null): bool
 
         $this->createSavepoint($this->transactionLevel);
 
+        $this->eventDispatcher?->dispatch(new TransactionBeginning($this));
+
         return true;
     }
 
diff --git a/src/Event/AbstractDriverEvent.php b/src/Event/AbstractDriverEvent.php
new file mode 100644
index 00000000..10fc6bd1
--- /dev/null
+++ b/src/Event/AbstractDriverEvent.php
@@ -0,0 +1,13 @@
+<?php
+
+namespace Cycle\Database\Event;
+
+use Cycle\Database\Driver\DriverInterface;
+
+abstract class AbstractDriverEvent
+{
+    public function __construct(
+        public readonly DriverInterface $driver,
+    ) {
+    }
+}
diff --git a/src/Event/QueryExecuted.php b/src/Event/QueryExecuted.php
index 292db933..a2f21c9c 100644
--- a/src/Event/QueryExecuted.php
+++ b/src/Event/QueryExecuted.php
@@ -4,14 +4,15 @@
 
 use Cycle\Database\Driver\DriverInterface;
 
-class QueryExecuted
+final class QueryExecuted extends AbstractDriverEvent
 {
     public function __construct(
         public readonly string $query,
         public readonly array $params,
         public readonly float $queryStart,
         public readonly float $queryEnd,
-        public readonly DriverInterface $driver,
+        DriverInterface $driver,
     ) {
+        parent::__construct($driver);
     }
 }
diff --git a/src/Event/TransactionBeginning.php b/src/Event/TransactionBeginning.php
new file mode 100644
index 00000000..446c6fd9
--- /dev/null
+++ b/src/Event/TransactionBeginning.php
@@ -0,0 +1,10 @@
+<?php
+
+namespace Cycle\Database\Event;
+
+use Cycle\Database\Event\AbstractDriverEvent;
+
+class TransactionBeginning extends AbstractDriverEvent
+{
+
+}
diff --git a/src/Event/TransactionCommited.php b/src/Event/TransactionCommited.php
new file mode 100644
index 00000000..07140361
--- /dev/null
+++ b/src/Event/TransactionCommited.php
@@ -0,0 +1,8 @@
+<?php
+
+namespace Cycle\Database\Event;
+
+class TransactionCommited extends AbstractDriverEvent
+{
+
+}
diff --git a/src/Event/TransactionRolledBack.php b/src/Event/TransactionRolledBack.php
new file mode 100644
index 00000000..b1740d35
--- /dev/null
+++ b/src/Event/TransactionRolledBack.php
@@ -0,0 +1,8 @@
+<?php
+
+namespace Cycle\Database\Event;
+
+class TransactionRolledBack extends AbstractDriverEvent
+{
+
+}
diff --git a/src/EventDispatcherAwareTrait.php b/src/EventDispatcherAwareTrait.php
index b9e1b6fc..5580ae81 100644
--- a/src/EventDispatcherAwareTrait.php
+++ b/src/EventDispatcherAwareTrait.php
@@ -6,7 +6,7 @@
 
 trait EventDispatcherAwareTrait
 {
-    private ?EventDispatcherInterface $eventDispatcher = null;
+    protected ?EventDispatcherInterface $eventDispatcher = null;
 
     public function setEventDispatcher(EventDispatcherInterface $eventDispatcher): void
     {