Fix #380. Добавил триггеры на insert

Author: arutyunyan

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Change log
 
+## v7.3.14 - July 22, 2022
+
+- Добавил триггеры на добавление записей
+
+---
+
 ## v7.3.13 - June 24, 2022
 
 - Рефакторинг макросов

src/db/Container.php

@@ -13,6 +13,10 @@ class Container
      * @var array контейнер подключений
      */
     private static $items;
+    /**
+     * @var array контейнер триггеров
+     */
+    private static $triggers;
     /**
      * Создает контейнер с подключениями
      * @param array $dbList
@@ -38,4 +42,24 @@ public static function get($connection)
     {
         return isset(self::$items[$connection]) ? self::$items[$connection] : null;
     }
+    /**
+     * Добавляет триггер
+     * @param string $modelClass
+     * @param string $type
+     * @param \Closure $callback
+     */
+    public static function addTrigger($modelClass, $type, $callback)
+    {
+        self::$triggers[$modelClass][$type] = $callback;
+    }
+    /**
+     * Получает триггер
+     * @param string $modelClass
+     * @param string $type
+     * @return \Closure|null
+     */
+    public static function getTrigger($modelClass, $type)
+    {
+        return isset(self::$triggers[$modelClass][$type]) ? self::$triggers[$modelClass][$type] : null;
+    }
 }

src/manager/TableEntityManager.php

@@ -65,6 +65,7 @@ private function saveEntity($entity)
         $entity->init();
         $entity->addInsert($entity->query());
         $entity->getQuery($entity->query())->addInto($entity->table());
+        $row = [];
         foreach ($entity->getPublicProperties() as $property) {
             $propertyName = $property->getName();
             if (is_subclass_of($entity->$propertyName, ActiveRecord::class)) {
@@ -74,6 +75,7 @@ private function saveEntity($entity)
             }
             else {
                 $entity->getQuery($entity->query())->addValue($propertyName, $entity->$propertyName);
+                $row[$propertyName] = $entity->$propertyName;
             }
         }
 
@@ -91,6 +93,10 @@ private function saveEntity($entity)
 
         $entity->getDb()->getPdo()->exec($entity->getRawSql());
 
+        if ($entity->hasTrigger('insert')) {
+            $entity->runTrigger('insert', $row);
+        }
+
         $entity->setLastInsertId($entity->getDb()->getPdo()->lastInsertId());
     }
     /**

src/syntax/ActiveRecord.php

@@ -12,6 +12,7 @@
 use ReflectionProperty;
 use suql\core\Scheme;
 use suql\core\SmartDate;
+use suql\db\Container;
 use suql\manager\TableEntityManager;
 use suql\syntax\field\Field;
 use suql\syntax\field\Raw;
@@ -176,6 +177,38 @@ public function save()
         $entityManager->persist($this);
         $entityManager->run();
     }
+    /**
+     * Вешает триггер
+     * @param string $type
+     * @param \Closure $callback
+     */
+    public static function trigger($type, $callback)
+    {
+        Container::addTrigger(static::class, $type, $callback);
+    }
+    /**
+     * Проверяет есть ли какой-нибудь триггер
+     * @param string $type
+     * @return boolean
+     */
+    public function hasTrigger($type)
+    {
+        if (!is_null(Container::getTrigger(static::class, $type))) {
+            return true;
+        }
+
+        return false;
+    }
+    /**
+     * Запускает триггер
+     * @param string $type
+     * @param array $row
+     */
+    public function runTrigger($type, $row)
+    {
+        $trigger = Container::getTrigger(static::class, $type);
+        $trigger($row);
+    }
     /**
      * Выборка с начально заданной фильтрацией
      * @return self

tests/suql/TriggerTest.php

@@ -0,0 +1,43 @@
+<?php
+
+declare(strict_types=1);
+
+use PHPUnit\Framework\TestCase;
+use suql\db\Container;
+use suql\syntax\Query;
+use test\suql\models\Query19;
+
+final class TriggerTest extends TestCase
+{
+    public function setUp(): void
+    {
+        // Create a database
+        Container::create(require('config/db-null.php'));
+        Query::create('create database db_test')->setConnection('connection')->exec();
+        Container::add(require('config/db.php'));
+        Query::create('create table table_10(f1 int, f2 int, primary key (f1))')->setConnection('db_test')->exec();
+        Query::create('insert into table_10 (f1, f2) values (1, 1), (2, 2), (3, 3)')->setConnection('db_test')->exec();
+        Query::create('insert into table_10 (f1, f2) values (?, ?), (?, ?), (?, ?)')->setConnection('db_test')->exec([4, 4, 5, 5, 6, 6]);
+    }
+
+    public function tearDown(): void
+    {
+        // Drop the database
+        Query::create('drop table table_10')->setConnection('db_test')->exec();
+        Query::create('drop database db_test')->setConnection('db_test')->exec();
+    }
+
+    public function testTrigger(): void
+    {
+        $expected = ['c1' => 7, 'c2' => 'sagittaracc'];
+
+        Query19::trigger('insert', function ($actual) use ($expected) {
+            $this->assertEquals($expected, $actual);
+        });
+
+        $record = new Query19();
+        $record->c1 = 7;
+        $record->c2 = 'sagittaracc';
+        $record->save();
+    }
+}