Implement count for other MR stores

geddy · Oct 17, 2013 · c830a30 · c830a30
1 parent 667f0f9
commit c830a30
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/lib/adapters/level/index.js b/lib/adapters/level/index.js
@@ -113,6 +113,11 @@ utils.mixin(Adapter.prototype, new (function () {
               res = res.slice(0, limit);
             }
           }
+
+          if (query.opts.count) {
+            res = res.length;
+          }
+
           callback(null, res);
         });
     }

diff --git a/lib/adapters/riak/index.js b/lib/adapters/riak/index.js
@@ -176,6 +176,11 @@ utils.mixin(Adapter.prototype, new (function () {
           if (query.opts.limit == 1) {
             res = res[0];
           }
+
+          if (query.opts.count) {
+            res = res.length;
+          }
+
           callback(null, res);
         }
       });