Count feature in progress

1 files changed, 31 insertions, 0 deletions

diff --git a/src/Psecio/Gatekeeper/DataSource/Mysql.php b/src/Psecio/Gatekeeper/DataSource/Mysql.php
index 5a1403a..a254f67 100644
--- a/src/Psecio/Gatekeeper/DataSource/Mysql.php
+++ b/src/Psecio/Gatekeeper/DataSource/Mysql.php
@@ -228,6 +228,37 @@ class Mysql extends \Psecio\Gatekeeper\DataSource
     }
 
     /**
+     * Find count of entities by where conditions.
+     * All where conditions applied with AND
+     *
+     * @param \Modler\Model $model Model instance
+     * @param array $where Data to use in "where" statement
+     * @return array Fetched data
+     */
+    public function count(\Modler\Model $model, array $where = array())
+    {
+        $properties = $model->getProperties();
+        list($columns, $bind) = $this->setup($where);
+
+        $update = array();
+        foreach ($bind as $column => $name) {
+            // See if we keep to transfer it over to a column name
+            if (array_key_exists($column, $properties)) {
+                $column = $properties[$column]['column'];
+            }
+            $update[] = $column.' = '.$name;
+        }
+
+        $sql = 'select * from '.$model->getTableName();
+        if (!empty($update)) {
+            $sql .= ' where '.implode(' and ', $update);
+        }
+
+        $result = $this->fetch($sql, $where, true);
+        return $result;
+    }
+
+    /**
      * Execute the request (not a fetch)
      *
      * @param string $sql SQL statement to execute