Merge pull request #134 from kivudesign/version_boss

[ENH] add missing count method for entity model
kivudesign · Sep 17, 2023 · 654d5d5 · 654d5d5
2 parents e5fc13c + 0e38de1
commit 654d5d5
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/src/Core/Orm/EntityModel/Provider/Entity.php b/src/Core/Orm/EntityModel/Provider/Entity.php
@@ -22,6 +22,10 @@ abstract class Entity implements EntityInterface
      * @var array
      */
     private array $include_entity;
+    /**
+     * @var array|mixed
+     */
+    private array $param;
     use EntityReflexion;
 
     /**
@@ -31,6 +35,7 @@ public function __construct()
     {
         $this->db = DB::getInstance();
         $this->include_entity = [];
+        $this->param = [];
     }
 
     /**
@@ -200,7 +205,28 @@ public function findOne(): array
             return ['exception' => $ex->getMessage()];
         }
     }
+    /**
+     * @return array
+     */
+    public function count(): array
+    {
+        try {
+
+            $query = $this->db->count($this->getTableName());
+            if (isset($this->param['where'])) {
+                $query->where($this->param['where']);
+            }
 
+            $result = $query->result();
+            $this->param = [];
+            if ($this->db->error()) {
+                throw new \Exception($this->db->error());
+            }
+            return $result;
+        } catch (\Exception $ex) {
+            return ['exception' => $ex->getMessage()];
+        }
+    }
     /**
      * @param array $fields
      * @return $this