added; Query#slaveOk() support

Automattic · Jul 6, 2011 · ee80882 · ee80882
1 parent 8a8b026
commit ee80882
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 5 deletions.
diff --git a/lib/mongoose/query.js b/lib/mongoose/query.js
@@ -284,7 +284,7 @@ Query.prototype.where = function (path, val) {
 
 Query.prototype.notEqualTo = Query.prototype.ne;
 
-['mod', 'near'].forEach( function ($conditional) {
+;['mod', 'near'].forEach( function ($conditional) {
   Query.prototype['$' + $conditional] =
   Query.prototype[$conditional] = function (path, val) {
     if (arguments.length === 1) {
@@ -346,7 +346,7 @@ Query.prototype.elemMatch = function (path, criteria) {
   return this;
 };
 
-['maxscan'].forEach( function (method) {
+;['maxscan'].forEach( function (method) {
   Query.prototype[method] = function (v) {
     this.options[method] = v;
     return this;
@@ -366,7 +366,7 @@ Query.prototype.explain = function () {
 // TODO geoNear command
 
 // To be used idiomatically where Query#box and Query#center
-['wherein', '$wherein'].forEach(function (getter) {
+;['wherein', '$wherein'].forEach(function (getter) {
   Object.defineProperty(Query.prototype, getter, {
     get: function () {
       return this;
@@ -423,7 +423,7 @@ Query.prototype.fields = function () {
 };
 
 /**
- * Chainable method for adding the specified fields to the 
+ * Chainable method for adding the specified fields to the
  * object of fields to only include.
  *
  * Examples:
@@ -589,7 +589,7 @@ Query.prototype.desc = function () {
   return this;
 };
 
-['limit', 'skip', 'maxscan', 'snapshot'].forEach( function (method) {
+;['limit', 'skip', 'maxscan', 'snapshot'].forEach( function (method) {
   Query.prototype[method] = function (v) {
     this.options[method] = v;
     return this;
@@ -627,6 +627,21 @@ Query.prototype.hint = function (v, multi) {
   return this;
 };
 
+/**
+ * Sets slaveOk option
+ *
+ *   new Query().slaveOk() <== true
+ *   new Query().slaveOk(true)
+ *   new Query().slaveOk(false)
+ *
+ * @param {Boolean} v (defaults to true)
+ */
+
+Query.prototype.slaveOk = function (v) {
+  this.options.slaveOk = arguments.length ? !!v : true;
+  return this;
+};
+
 Query.prototype.execFind = function (callback) {
   var model = this.model
     , options = this._optionsForExec(model);

diff --git a/test/query.test.js b/test/query.test.js
@@ -688,6 +688,20 @@ module.exports = {
     query.options.maxscan.should.equal(100);
   },
 
+  'test Query#slaveOk': function () {
+    var query = new Query();
+    query.slaveOk();
+    query.options.slaveOk.should.be.true;
+
+    var query = new Query();
+    query.slaveOk(true);
+    query.options.slaveOk.should.be.true;
+
+    var query = new Query();
+    query.slaveOk(false);
+    query.options.slaveOk.should.be.false;
+  },
+
   'test Query#hint': function () {
     var query = new Query();
     query.hint('indexAttributeA', 1, 'indexAttributeB', -1);