Extract DropDatabase operation class

Author: jmikola

src/Client.php

@@ -8,6 +8,7 @@
 use MongoDB\Driver\ReadPreference;
 use MongoDB\Driver\WriteConcern;
 use MongoDB\Model\DatabaseInfoIterator;
+use MongoDB\Operation\DropDatabase;
 use MongoDB\Operation\ListDatabases;
 
 class Client
@@ -36,17 +37,15 @@ public function __construct($uri, array $options = array(), array $driverOptions
     /**
      * Drop a database.
      *
-     * @see http://docs.mongodb.org/manual/reference/command/dropDatabase/
      * @param string $databaseName
-     * @return Cursor
+     * @return object Command result document
      */
     public function dropDatabase($databaseName)
     {
-        $databaseName = (string) $databaseName;
-        $command = new Command(array('dropDatabase' => 1));
-        $readPreference = new ReadPreference(ReadPreference::RP_PRIMARY);
+        $operation = new DropDatabase($databaseName);
+        $server = $this->manager->selectServer(new ReadPreference(ReadPreference::RP_PRIMARY));
 
-        return $this->manager->executeCommand($databaseName, $command, $readPreference);
+        return $operation->execute($server);
     }
 
     /**

src/Database.php

@@ -12,6 +12,7 @@
 use MongoDB\Driver\WriteConcern;
 use MongoDB\Exception\InvalidArgumentException;
 use MongoDB\Model\CollectionInfoIterator;
+use MongoDB\Operation\DropDatabase;
 use MongoDB\Operation\ListCollections;
 
 class Database
@@ -71,15 +72,14 @@ public function createCollection($collectionName, array $options = array())
     /**
      * Drop this database.
      *
-     * @see http://docs.mongodb.org/manual/reference/command/dropDatabase/
-     * @return Cursor
+     * @return object Command result document
      */
     public function drop()
     {
-        $command = new Command(array('dropDatabase' => 1));
-        $readPreference = new ReadPreference(ReadPreference::RP_PRIMARY);
+        $operation = new DropDatabase($this->databaseName);
+        $server = $this->manager->selectServer(new ReadPreference(ReadPreference::RP_PRIMARY));
 
-        return $this->manager->executeCommand($this->databaseName, $command, $readPreference);
+        return $operation->execute($server);
     }
 
     /**

src/Operation/DropDatabase.php

@@ -0,0 +1,50 @@
+<?php
+
+namespace MongoDB\Operation;
+
+use MongoDB\Driver\Command;
+use MongoDB\Driver\Server;
+use MongoDB\Exception\RuntimeException;
+
+/**
+ * Operation for the dropDatabase command.
+ *
+ * @api
+ * @see MongoDB\Client::dropDatabase()
+ * @see MongoDB\Database::drop()
+ * @see http://docs.mongodb.org/manual/reference/command/dropDatabase/
+ */
+class DropDatabase implements Executable
+{
+    private $databaseName;
+
+    /**
+     * Constructs a dropDatabase command.
+     *
+     * @param string $databaseName   Database name
+     * @param string $collectionName Collection name
+     */
+    public function __construct($databaseName)
+    {
+        $this->databaseName = (string) $databaseName;
+    }
+
+    /**
+     * Execute the operation.
+     *
+     * @see Executable::execute()
+     * @param Server $server
+     * @return object Command result document
+     */
+    public function execute(Server $server)
+    {
+        $cursor = $server->executeCommand($this->databaseName, new Command(array('dropDatabase' => 1)));
+        $result = current($cursor->toArray());
+
+        if (empty($result['ok'])) {
+            throw new RuntimeException(isset($result['errmsg']) ? $result['errmsg'] : 'Unknown error');
+        }
+
+        return $result;
+    }
+}