[ENH] Add support to query transactions

Author: bim-g

composer.json

@@ -16,7 +16,8 @@
         }
     ],
     "require": {
-        "php": ">=7.4"
+        "php": ">=7.4",
+      "ext-pdo": "*"
     },
     "require-dev": {
         "phpunit/phpunit": "^9.5"

src/DB.php

@@ -20,11 +20,7 @@ class DB
     private \PDO $pdo;
     private string $_action="";
     private int $_count;
-    private array $option=[
-        \PDO::MYSQL_ATTR_INIT_COMMAND=>"SET NAMES utf8",
-        \PDO::ATTR_EMULATE_PREPARES=>false,
-        \PDO::ATTR_ERRMODE=>\PDO::ERRMODE_EXCEPTION
-    ]  ;
+    private string $db_name;
     use BuildQuery;
     private function __construct(string $host="",string $db_name="",string $user_name="",string $password="")
     {
@@ -33,8 +29,13 @@ private function __construct(string $host="",string $db_name="",string $user_nam
             $this->_error = null;
             $this->_lastID = -1;
             $this->_count = 0;
+            $this->db_name = $db_name;
             //
-            $this->pdo = new \PDO("mysql:host=" . $host . ";dbname=" . $db_name.";charset=utf8mb4", $user_name,$password,$this->option);
+            $this->pdo = new \PDO("mysql:host=" . $host . ";dbname=" . $db_name.";charset=utf8mb4", $user_name,$password);
+            $this->pdo->setAttribute(\PDO::MYSQL_ATTR_INIT_COMMAND,'SET NAMES utf8');
+            $this->pdo->setAttribute(\PDO::ATTR_EMULATE_PREPARES,false);
+            $this->pdo->setAttribute(\PDO::ATTR_PERSISTENT,true);
+            $this->pdo->setAttribute(\PDO::ATTR_ERRMODE,\PDO::ERRMODE_EXCEPTION);
         } catch (\PDOException $ex) {
             echo $ex->getMessage();
             die();
@@ -181,4 +182,31 @@ function rowCount(){
         }
         return $this->_count;
     }
+
+    function beginTransaction(): bool
+    {
+        return $this->pdo->beginTransaction();
+    }
+    function commit(){
+        return $this->pdo->commit();
+    }
+    function rollBack(){
+        return $this->pdo->rollBack();
+    }
+
+    /**
+     * @throws Exception
+     */
+    function transaction(\Closure $callable){
+        try{
+            $this->pdo->beginTransaction();
+            $callable($this);
+            $this->pdo->commit();
+        }catch (\Exception $ex){
+            if ($this->pdo->inTransaction()) {
+                $this->pdo->rollBack();
+            }
+            throw $ex;
+        }
+    }
 }

src/DB_Insert.php

@@ -71,8 +71,8 @@ private function query(string $sql, array $params = [])
     {
         $q = $this->executeQuery($this->pdo, $sql, $params);
         $this->_results = $q['result'];
+        $this->_error = $q['error']??"";
         $this->lastID = $q['lastID']??0;
-        $this->_error = $q['error'];
     }
 
     /**
@@ -88,10 +88,10 @@ private function insert()
     }
 
     /**
-     * @return bool
+     * @return array
      * return result after a request select
      */
-    function result()
+    function result(): array
     {
         $this->insert();
         return $this->_results;

test/index.php

@@ -8,8 +8,9 @@
     "password"=>""
 ];
 $db = DB::getInstance($config);
- include("./test/insert.php");
+// include("./test/insert.php");
 //include("./test/select.php");
 // include("./test/delete.php");
 // include("./test/query.php");
-// include("./test/update.php");
+// include("./test/update.php");
+ include("./test/transaction.php");

test/insert.php

@@ -1,5 +1,5 @@
 <?php
-$db=$db??[];
+$db = $db??[];
 $field = [
     "userid" => 1,
     "message" => "hello from wepesi",

test/transaction.php

@@ -0,0 +1,34 @@
+<?php
+$db = $db??[];
+$user = [
+    "fullname" => "Celestin Doe",
+    "username" => "John Doe",
+    "password" => md5("12345678"),
+    "datecreated" => Date("Y-m-d H:i:s",time())
+];
+
+$message = [
+    "message" => "Hello Celestin",
+    'datecreated' => Date('Y-m-d H:i:s', time())
+];
+try {
+    $db->transaction(function($db) use ($user,$message){
+        $db->insert('users')->field($user)->result();
+        if ($db->error()) {
+            throw new \Exception($db->error());
+        }
+        $user_id = $db->lastId();
+        $user['id'] = $user_id;
+        $message['userid'] = $user_id;
+        $db->insert('message')->field($message)->result();
+        if ($db->error()) {
+            throw new \Exception($db->error());
+        }
+        $message['id'] = $db->lastId();
+        $user['messages'] = $message;
+
+        print_r($user);
+    });
+} catch (\Exception $ex) {
+    var_dump($ex);
+}