fix swoft-cloud/swoft#277 (#96)

swoft-cloud · Jun 14, 2018 · 78dde00 · 78dde00
1 parent 75e5701
commit 78dde00
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/Executor.php b/src/Executor.php
@@ -17,6 +17,7 @@
 use Swoft\Db\Exception\MysqlException;
 use Swoft\Db\Validator\ValidatorInterface;
 use Swoft\Exception\ValidatorException;
+use Swoft\Helper\StringHelper;
 
 /**
  * Executor
@@ -36,7 +37,13 @@ public static function save($entity): ResultInterface
 
         $fields = $fields ?? [];
         $query  = Query::table($table)->selectInstance($instance);
-
+        // Set Primary Id to Entity
+        $query->addDecorator(function ($primaryId) use ($entity, $className) {
+            list(, , $idColumn) = self::getTable($className);
+            $setter = 'set' . StringHelper::camel($idColumn, false);
+            method_exists($entity, $setter) && $entity->$setter($primaryId);
+            return $primaryId;
+        });
         return $query->insert($fields);
     }