From 469507d494a03c5a8a39a538258f41e9b6512674 Mon Sep 17 00:00:00 2001
From: Paul Klimov <klimov.paul@gmail.com>
Date: Mon, 2 Dec 2013 17:26:30 +0200
Subject: [PATCH] Mongo index manipulation methods added.

---
 extensions/mongo/Collection.php                | 66 ++++++++++++++++++++++++++
 extensions/mongo/Database.php                  | 14 ++++++
 tests/unit/extensions/mongo/CollectionTest.php | 10 ++++
 3 files changed, 90 insertions(+)

diff --git a/extensions/mongo/Collection.php b/extensions/mongo/Collection.php
index a9e6f0a..497122c 100644
--- a/extensions/mongo/Collection.php
+++ b/extensions/mongo/Collection.php
@@ -33,6 +33,72 @@ class Collection extends Object
 	}
 
 	/**
+	 * Creates an index on the collection and the specified fields
+	 * @param array|string $columns column name or list of column names.
+	 * If array is given, each element in the array has as key the field name, and as
+	 * value either 1 for ascending sort, or -1 for descending sort.
+	 * You can specify field using native numeric key with the field name as a value,
+	 * in this case ascending sort will be used.
+	 * For example:
+	 * ~~~
+	 * [
+	 *     'name',
+	 *     'status' => -1,
+	 * ]
+	 * ~~~
+	 * @param array $options list of options in format: optionName => optionValue.
+	 * @throws Exception on failure.
+	 * @return boolean whether the operation successful.
+	 */
+	public function createIndex($columns, $options = [])
+	{
+		if (!is_array($columns)) {
+			$columns = [$columns];
+		}
+		$token = 'Creating index at ' . $this->mongoCollection->getName() . ' on ' . implode(',', $columns);
+		Yii::info($token, __METHOD__);
+		try {
+			Yii::beginProfile($token, __METHOD__);
+			$keys = [];
+			foreach ($columns as $key => $value) {
+				if (is_numeric($key)) {
+					$keys[$value] = \MongoCollection::ASCENDING;
+				} else {
+					$keys[$key] = $value;
+				}
+			}
+			$options = array_merge(['w' => 1], $options);
+			$result = $this->mongoCollection->ensureIndex($keys, $options);
+			$this->tryResultError($result);
+			Yii::endProfile($token, __METHOD__);
+			return true;
+		} catch (\Exception $e) {
+			Yii::endProfile($token, __METHOD__);
+			throw new Exception($e->getMessage(), (int)$e->getCode(), $e);
+		}
+	}
+
+	/**
+	 * Drop indexes for specified column(s).
+	 * @param string|array $columns column name or list of column names.
+	 * @return array result.
+	 */
+	public function dropIndex($columns)
+	{
+		return $this->mongoCollection->deleteIndex($columns);
+	}
+
+	/**
+	 * Drops all indexes for this collection
+	 * @return boolean whether the operation successful.
+	 */
+	public function dropAllIndexes()
+	{
+		$result = $this->mongoCollection->deleteIndexes();
+		return !empty($result['ok']);
+	}
+
+	/**
 	 * @param array $condition
 	 * @param array $fields
 	 * @return \MongoCursor
diff --git a/extensions/mongo/Database.php b/extensions/mongo/Database.php
index 4954a20..c6ecc74 100644
--- a/extensions/mongo/Database.php
+++ b/extensions/mongo/Database.php
@@ -61,4 +61,18 @@ class Database extends Object
 	{
 		$this->mongoDb->drop();
 	}
+
+	/**
+	 * Creates new collection.
+	 * Note: Mongo creates new collections automatically on the first demand,
+	 * this method makes sense only for the migration script or for the case
+	 * you need to create collection with the specific options.
+	 * @param string $name name of the collection
+	 * @param array $options collection options in format: "name" => "value"
+	 * @return \MongoCollection new mongo collection instance.
+	 */
+	public function createCollection($name, $options = [])
+	{
+		return $this->mongoDb->createCollection($name, $options);
+	}
 }
\ No newline at end of file
diff --git a/tests/unit/extensions/mongo/CollectionTest.php b/tests/unit/extensions/mongo/CollectionTest.php
index f7d8090..110e716 100644
--- a/tests/unit/extensions/mongo/CollectionTest.php
+++ b/tests/unit/extensions/mongo/CollectionTest.php
@@ -172,4 +172,14 @@ class CollectionTest extends MongoTestCase
 		$this->assertNotEmpty($result[0]['address']);
 		$this->assertNotEmpty($result[0]['items']);
 	}
+
+	public function testCreateIndex()
+	{
+		$collection = $this->getConnection()->getCollection('customer');
+		$columns = [
+			'name',
+			'status' => \MongoCollection::DESCENDING,
+		];
+		$this->assertTrue($collection->createIndex($columns));
+	}
 }
\ No newline at end of file