improving countable

zemit-cms · Feb 29, 2024 · 7531f51 · 7531f51
1 parent 2c58c8d
commit 7531f51
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/Mvc/Controller/Rest/Actions/CountAction.php b/src/Mvc/Controller/Rest/Actions/CountAction.php
@@ -31,7 +31,7 @@ public function countAction(): ResponseInterface
         $model = $this->getModelClassName();
 
         $countResult = $model::count($this->getFindCount($this->getFind()));
-        $count = is_array($countResult) ? count($countResult) : $countResult;
+        $count = is_countable($countResult) ? count($countResult) : (int)$countResult;
 
         $this->view->setVar('count', $count);
         return $this->setRestResponse(true);

diff --git a/src/Mvc/Controller/Rest/Actions/GetListAction.php b/src/Mvc/Controller/Rest/Actions/GetListAction.php
@@ -37,7 +37,7 @@ public function getListAction(): ResponseInterface
         $find = $this->getFind() ?: [];
 
         $totalCount = $model::count($this->getFindCount($find));
-        $totalCount = is_int($totalCount) ? $totalCount : count($totalCount);
+        $totalCount = is_countable($totalCount)? count($totalCount) : (int)$totalCount;
         $this->view->setVars([
             'list' => $this->listExpose($model::findWith($with, $find)),
             'totalCount' => $totalCount,