Use of transactions in entity manager

Author: na2axl

src/LightQL/Entities/EntityManager.php

@@ -35,6 +35,7 @@
 use ElementaryFramework\Annotations\Annotations;
 use ElementaryFramework\LightQL\LightQL;
 use ElementaryFramework\LightQL\Persistence\PersistenceUnit;
+use ElementaryFramework\LightQL\Exceptions\EntityException;
 
 /**
  * Entity Manager
@@ -150,12 +151,21 @@ public function persist(Entity &$entity)
             }
         }
 
-        $this->_lightql
-            ->from($entityAnnotation[0]->table)
-            ->insert($fieldAndValues);
+        $this->_lightql->beginTransaction();
+        try {
+            $this->_lightql
+                ->from($entityAnnotation[0]->table)
+                ->insert($fieldAndValues);
+
+            if ($autoIncrementProperty !== null) {
+                $entity->$autoIncrementProperty = $this->_lightql->lastInsertID();
+            }
+
+            $this->_lightql->commit();
+        } catch (\Exception $e) {
+            $this->_lightql->rollback();
 
-        if ($autoIncrementProperty !== null) {
-            $entity->$autoIncrementProperty = $this->_lightql->lastInsertID();
+            throw new EntityException($e->getMessage());
         }
     }
 
@@ -184,10 +194,19 @@ public function merge(Entity &$entity)
             }
         }
 
-        $this->_lightql
-            ->from($entityAnnotation[0]->table)
-            ->where($where)
-            ->update($fieldAndValues);
+        $this->_lightql->beginTransaction();
+        try {
+            $this->_lightql
+                ->from($entityAnnotation[0]->table)
+                ->where($where)
+                ->update($fieldAndValues);
+
+            $this->_lightql->commit();
+        } catch (\Exception $e) {
+            $this->_lightql->rollback();
+
+            throw new EntityException($e->getMessage());
+        }
     }
 
     /**
@@ -217,19 +236,31 @@ public function delete(Entity &$entity)
             }
         }
 
-        $this->_lightql
-            ->from($entityAnnotation[0]->table)
-            ->where($where)
-            ->delete();
+        $this->_lightql->beginTransaction();
+        try {
+            $this->_lightql
+                ->from($entityAnnotation[0]->table)
+                ->where($where)
+                ->delete();
 
-        if (count($pk) > 0) {
-            foreach ($pk as $item) {
-                $entity->$item = null;
+            if (count($pk) > 0) {
+                foreach ($pk as $item) {
+                    $entity->$item = null;
+                }
             }
+
+            $this->_lightql->commit();
+        } catch (\Exception $e) {
+            $this->_lightql->rollback();
+
+            throw new EntityException($e->getMessage());
         }
     }
 
     /**
+     * Gets the LightQL instance associated
+     * to this entity manager.
+     * 
      * @return LightQL
      */
     public function getLightQL(): LightQL

src/LightQL/LightQL.php

@@ -986,6 +986,21 @@ public function quote($value): string
         return $this->_pdo->quote($value);
     }
 
+    public function beginTransaction(): bool
+    {
+        return $this->_pdo->beginTransaction();
+    }
+
+    public function commit(): bool
+    {
+        return $this->_pdo->commit();
+    }
+
+    public function rollback(): bool
+    {
+        return $this->_pdo->rollBack();
+    }
+
     /**
      * Converts a value to a string.
      *