Merge pull request #92 from doctrine/read-pref-methods

CollectionTest refactoring and get/setReadPreference() methods
doctrine · Mar 4, 2013 · bc59129 · bc59129
2 parents ec10f4d + 257fb45
commit bc59129
Show file tree

Hide file tree

Showing 9 changed files with 446 additions and 252 deletions.
diff --git a/lib/Doctrine/MongoDB/Collection.php b/lib/Doctrine/MongoDB/Collection.php
@@ -666,6 +666,20 @@ public function getSlaveOkay()
         return \MongoClient::RP_PRIMARY !== $readPref['type'];
     }
 
+    public function getReadPreference()
+    {
+        return $this->getMongoCollection()->getReadPreference();
+    }
+
+    public function setReadPreference($readPreference, array $tags = null)
+    {
+        if (isset($tags)) {
+            return $this->getMongoCollection()->setReadPreference($readPreference, $tags);
+        }
+
+        return $this->getMongoCollection()->setReadPreference($readPreference);
+    }
+
     public function validate($scanData = false)
     {
         return $this->getMongoCollection()->validate($scanData);

diff --git a/lib/Doctrine/MongoDB/Connection.php b/lib/Doctrine/MongoDB/Connection.php
@@ -258,6 +258,20 @@ public function selectDatabase($name)
         return $database;
     }
 
+    public function getReadPreference()
+    {
+        return $this->mongo->getReadPreference();
+    }
+
+    public function setReadPreference($readPreference, array $tags = null)
+    {
+        if (isset($tags)) {
+            return $this->mongo->setReadPreference($readPreference, $tags);
+        }
+
+        return $this->mongo->setReadPreference($readPreference);
+    }
+
     /**
      * Method which creates a Doctrine\MongoDB\Database instance.
      *

diff --git a/lib/Doctrine/MongoDB/Cursor.php b/lib/Doctrine/MongoDB/Cursor.php
@@ -387,6 +387,20 @@ public function getSingleResult()
         return $result;
     }
 
+    public function getReadPreference()
+    {
+        return $this->getMongoDB()->getReadPreference();
+    }
+
+    public function setReadPreference($readPreference, array $tags = null)
+    {
+        if (isset($tags)) {
+            return $this->mongoCursor->setReadPreference($readPreference, $tags);
+        }
+
+        return $this->mongoCursor->setReadPreference($readPreference);
+    }
+
     protected function retry(\Closure $retry, $recreate = false)
     {
         if ($this->numRetries) {

diff --git a/lib/Doctrine/MongoDB/Database.php b/lib/Doctrine/MongoDB/Database.php
@@ -250,6 +250,20 @@ public function getSlaveOkay()
         return \MongoClient::RP_PRIMARY !== $readPref['type'];
     }
 
+    public function getReadPreference()
+    {
+        return $this->getMongoDB()->getReadPreference();
+    }
+
+    public function setReadPreference($readPreference, array $tags = null)
+    {
+        if (isset($tags)) {
+            return $this->getMongoDB()->setReadPreference($readPreference, $tags);
+        }
+
+        return $this->getMongoDB()->setReadPreference($readPreference);
+    }
+
     public function getProfilingLevel()
     {
         return $this->getMongoDB()->getProfilingLevel();