added countReturn which takes limit and and skip into consideration

SERVER-75

1 files changed, 12 insertions, 0 deletions

diff --git a/shell/query.js b/shell/query.js
index e152c669913..2c108af42ee 100644
--- a/shell/query.js
+++ b/shell/query.js
@@ -116,6 +116,18 @@ DBQuery.prototype.count = function(){
     throw "count failed: " + tojson( res );
 }
 
+DBQuery.prototype.countReturn = function(){
+    var c = this.count();
+
+    if ( this._skip )
+        c = c - this._skip;
+
+    if ( this._limit > 0 && this._limit < c )
+        return this._limit;
+    
+    return c;
+}
+
 /**
 * iterative count - only for testing
 */