支持selectForUpdate

easy-swoole · Sep 12, 2024 · ca0fc4a · ca0fc4a
1 parent 4d234df
commit ca0fc4a
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 4 deletions.
diff --git a/src/AbstractInterface/AbstractEntity.php b/src/AbstractInterface/AbstractEntity.php
@@ -97,7 +97,7 @@ function setData(array $data,bool $mergeCompare = false):static
         return $this;
     }
 
-    function all():ListResult
+    function all( bool $selectForUpdate = false ):ListResult
     {
         $query = $this->queryLimit()->__getQueryBuilder();
 
@@ -108,6 +108,10 @@ function all():ListResult
             $returnAsArray = $this->queryLimit()->getFields()['returnAsArray'];
         }
 
+        if($selectForUpdate){
+            $query->selectForUpdate();
+        }
+
         $query->get($this->tableName(),null,$fields);
         $ret = FastDb::getInstance()->query($query);
         $total = null;
@@ -622,7 +626,11 @@ public function find(array|string|int $queryLimit = null): ?static
         return null;
     }
 
-    public static function findRecord(callable|array|string|int $queryLimit, string $tableName = null): ?static
+    public static function findRecord(
+        callable|array|string|int $queryLimit,
+        string $tableName = null,
+        bool $selectForUpdate = false
+    ): ?static
     {
         $entity = new static();
         if (empty($tableName)) {
@@ -647,7 +655,9 @@ public static function findRecord(callable|array|string|int $queryLimit, string
             }
             $query->where($pk, $queryLimit);
         }
-
+        if($selectForUpdate){
+            $query->selectForUpdate();
+        }
         $query->get($tableName, 1);
         $ret = FastDb::getInstance()->query($query)->getResult();
         if (!empty($ret[0])) {
@@ -657,7 +667,12 @@ public static function findRecord(callable|array|string|int $queryLimit, string
         return null;
     }
 
-    public static function findAll(array|callable|string $queryLimit = null, string $tableName = null, bool $returnAsArray = false):mixed
+    public static function findAll(
+        array|callable|string $queryLimit = null,
+        string $tableName = null,
+        bool $returnAsArray = false,
+        bool $selectForUpdate = false
+    ):mixed
     {
         $entity = new static();
         if (empty($tableName)) {
@@ -692,6 +707,9 @@ public static function findAll(array|callable|string $queryLimit = null, string
                 }
             }
         }
+        if($selectForUpdate){
+            $query->selectForUpdate();
+        }
 
         $query->get($tableName);
         $result = FastDb::getInstance()->query($query)->getResult();

diff --git a/src/Beans/Query.php b/src/Beans/Query.php
@@ -99,6 +99,12 @@ function func(callable $func):Query
         return $this;
     }
 
+    function selectForUpdate():Query
+    {
+        $this->queryBuilder->selectForUpdate();
+        return $this;
+    }
+
     function returnEntity():AbstractEntity
     {
         return $this->entity;