feat: add promise support in query count

XadillaX · Oct 19, 2016 · 008bcc8 · 008bcc8
1 parent 656c4c5
commit 008bcc8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/query.js b/lib/query.js
@@ -152,9 +152,11 @@ class ToshihikoQuery {
     }
 
     count(callback) {
+        callback = common.promisify(callback);
         this.adapter.count(this, function(err, count, extra) {
             return callback(err, count, extra);
         });
+        return callback.promise;
     }
 
     find(callback, toJSON, options) {