feat: count in query

XadillaX · Sep 30, 2016 · 2b72ede · 2b72ede
1 parent 87a6882
commit 2b72ede
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/lib/adapters/mysql.js b/lib/adapters/mysql.js
@@ -530,8 +530,13 @@ class MySQLAdapter extends Adapter {
     }
 
     makeSql(type, model, options) {
+        if(type === "count") {
+            options.count = true;
+        }
+
         switch(type) {
             case "find": return this.makeFind(model, options);
+            case "count": return this.makeFind(model, options);
             case "update": return this.makeUpdate(model, options);
             case "delete": return this.makeDelete(model, options);
             default: return this.makeFind(model, options);
@@ -679,6 +684,15 @@ class MySQLAdapter extends Adapter {
         return _options;
     }
 
+    count(query, callback) {
+        const _options = this.queryToOptions(query, {});
+        const sql = this.makeSet("count", query.model, _options);
+        this.execute(sql, function(err, rows) {
+            if(err) return callback(err, undefined, sql);
+            return callback(undefined, (rows || [ { "COUNT(0)": 0 } ])[0]["COUNT(0)"], sql);
+        });
+    }
+
     find(query, callback, options) {
         const _options = this.queryToOptions(query, options);
         if(!query.cache || options.noCache) {

diff --git a/lib/query.js b/lib/query.js
@@ -150,6 +150,12 @@ class ToshihikoQuery {
         return this;
     }
 
+    count(callback) {
+        this.adapter.count(this, function(err, count, extra) {
+            return callback(err, count, extra);
+        });
+    }
+
     find(callback, toJSON, options) {
         if(typeof toJSON === "object") {
             options = toJSON;