Commit dfefd060 by Klimov Paul

Mongo index related methods fixed.

parent 469507d4
...@@ -25,11 +25,40 @@ class Collection extends Object ...@@ -25,11 +25,40 @@ class Collection extends Object
public $mongoCollection; public $mongoCollection;
/** /**
* @return string name of this collection.
*/
public function getName()
{
return $this->mongoCollection->getName();
}
/**
* @return string full name of this collection, including database name.
*/
public function getFullName()
{
return $this->mongoCollection->__toString();
}
/**
* Drops this collection. * Drops this collection.
* @throws Exception on failure.
* @return boolean whether the operation successful.
*/ */
public function drop() public function drop()
{ {
$this->mongoCollection->drop(); $token = 'Drop collection ' . $this->getFullName();
Yii::info($token, __METHOD__);
try {
Yii::beginProfile($token, __METHOD__);
$result = $this->mongoCollection->drop();
$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);
}
} }
/** /**
...@@ -55,18 +84,11 @@ class Collection extends Object ...@@ -55,18 +84,11 @@ class Collection extends Object
if (!is_array($columns)) { if (!is_array($columns)) {
$columns = [$columns]; $columns = [$columns];
} }
$token = 'Creating index at ' . $this->mongoCollection->getName() . ' on ' . implode(',', $columns); $token = 'Creating index at ' . $this->getFullName() . ' on ' . implode(',', $columns);
Yii::info($token, __METHOD__); Yii::info($token, __METHOD__);
try { try {
Yii::beginProfile($token, __METHOD__); Yii::beginProfile($token, __METHOD__);
$keys = []; $keys = $this->normalizeIndexKeys($columns);
foreach ($columns as $key => $value) {
if (is_numeric($key)) {
$keys[$value] = \MongoCollection::ASCENDING;
} else {
$keys[$key] = $value;
}
}
$options = array_merge(['w' => 1], $options); $options = array_merge(['w' => 1], $options);
$result = $this->mongoCollection->ensureIndex($keys, $options); $result = $this->mongoCollection->ensureIndex($keys, $options);
$this->tryResultError($result); $this->tryResultError($result);
...@@ -81,21 +103,73 @@ class Collection extends Object ...@@ -81,21 +103,73 @@ class Collection extends Object
/** /**
* Drop indexes for specified column(s). * Drop indexes for specified column(s).
* @param string|array $columns column name or list of column names. * @param string|array $columns column name or list of column names.
* @return array result. * 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,
* ]
* ~~~
* @throws Exception on failure.
* @return boolean whether the operation successful.
*/ */
public function dropIndex($columns) public function dropIndex($columns)
{ {
return $this->mongoCollection->deleteIndex($columns); if (!is_array($columns)) {
$columns = [$columns];
}
$token = 'Drop index at ' . $this->getFullName() . ' on ' . implode(',', $columns);
Yii::info($token, __METHOD__);
try {
$keys = $this->normalizeIndexKeys($columns);
$result = $this->mongoCollection->deleteIndex($keys);
$this->tryResultError($result);
return true;
} catch (\Exception $e) {
Yii::endProfile($token, __METHOD__);
throw new Exception($e->getMessage(), (int)$e->getCode(), $e);
}
} }
/** /**
* Drops all indexes for this collection * Compose index keys from given columns/keys list.
* @return boolean whether the operation successful. * @param array $columns raw columns/keys list.
* @return array normalizes index keys array.
*/
protected function normalizeIndexKeys($columns)
{
$keys = [];
foreach ($columns as $key => $value) {
if (is_numeric($key)) {
$keys[$value] = \MongoCollection::ASCENDING;
} else {
$keys[$key] = $value;
}
}
return $keys;
}
/**
* Drops all indexes for this collection.
* @throws Exception on failure.
* @return integer count of dropped indexes.
*/ */
public function dropAllIndexes() public function dropAllIndexes()
{ {
$result = $this->mongoCollection->deleteIndexes(); $token = 'Drop ALL indexes at ' . $this->getFullName();
return !empty($result['ok']); Yii::info($token, __METHOD__);
try {
$result = $this->mongoCollection->deleteIndexes();
$this->tryResultError($result);
return $result['nIndexesWas'] - 1;
} catch (\Exception $e) {
Yii::endProfile($token, __METHOD__);
throw new Exception($e->getMessage(), (int)$e->getCode(), $e);
}
} }
/** /**
...@@ -132,7 +206,7 @@ class Collection extends Object ...@@ -132,7 +206,7 @@ class Collection extends Object
*/ */
public function insert($data, $options = []) public function insert($data, $options = [])
{ {
$token = 'Inserting data into ' . $this->mongoCollection->getName(); $token = 'Inserting data into ' . $this->getFullName();
Yii::info($token, __METHOD__); Yii::info($token, __METHOD__);
try { try {
Yii::beginProfile($token, __METHOD__); Yii::beginProfile($token, __METHOD__);
...@@ -155,7 +229,7 @@ class Collection extends Object ...@@ -155,7 +229,7 @@ class Collection extends Object
*/ */
public function batchInsert($rows, $options = []) public function batchInsert($rows, $options = [])
{ {
$token = 'Inserting batch data into ' . $this->mongoCollection->getName(); $token = 'Inserting batch data into ' . $this->getFullName();
Yii::info($token, __METHOD__); Yii::info($token, __METHOD__);
try { try {
Yii::beginProfile($token, __METHOD__); Yii::beginProfile($token, __METHOD__);
...@@ -179,7 +253,7 @@ class Collection extends Object ...@@ -179,7 +253,7 @@ class Collection extends Object
*/ */
public function update($condition, $newData, $options = []) public function update($condition, $newData, $options = [])
{ {
$token = 'Updating data in ' . $this->mongoCollection->getName(); $token = 'Updating data in ' . $this->getFullName();
Yii::info($token, __METHOD__); Yii::info($token, __METHOD__);
try { try {
Yii::beginProfile($token, __METHOD__); Yii::beginProfile($token, __METHOD__);
...@@ -214,7 +288,7 @@ class Collection extends Object ...@@ -214,7 +288,7 @@ class Collection extends Object
*/ */
public function save($data, $options = []) public function save($data, $options = [])
{ {
$token = 'Saving data into ' . $this->mongoCollection->getName(); $token = 'Saving data into ' . $this->getFullName();
Yii::info($token, __METHOD__); Yii::info($token, __METHOD__);
try { try {
Yii::beginProfile($token, __METHOD__); Yii::beginProfile($token, __METHOD__);
...@@ -237,7 +311,7 @@ class Collection extends Object ...@@ -237,7 +311,7 @@ class Collection extends Object
*/ */
public function remove($condition = [], $options = []) public function remove($condition = [], $options = [])
{ {
$token = 'Removing data from ' . $this->mongoCollection->getName(); $token = 'Removing data from ' . $this->getFullName();
Yii::info($token, __METHOD__); Yii::info($token, __METHOD__);
try { try {
Yii::beginProfile($token, __METHOD__); Yii::beginProfile($token, __METHOD__);
...@@ -264,7 +338,12 @@ class Collection extends Object ...@@ -264,7 +338,12 @@ class Collection extends Object
*/ */
public function distinct($column, $condition = []) public function distinct($column, $condition = [])
{ {
return $this->mongoCollection->distinct($column, $this->buildCondition($condition)); $token = 'Get distinct ' . $column . ' from ' . $this->getFullName();
Yii::info($token, __METHOD__);
Yii::beginProfile($token, __METHOD__);
$result = $this->mongoCollection->distinct($column, $this->buildCondition($condition));
Yii::endProfile($token, __METHOD__);
return $result;
} }
/** /**
...@@ -276,8 +355,13 @@ class Collection extends Object ...@@ -276,8 +355,13 @@ class Collection extends Object
*/ */
public function aggregate($pipeline, $pipelineOperator = []) public function aggregate($pipeline, $pipelineOperator = [])
{ {
$token = 'Aggregating from ' . $this->getFullName();
Yii::info($token, __METHOD__);
Yii::beginProfile($token, __METHOD__);
$args = func_get_args(); $args = func_get_args();
return call_user_func_array([$this->mongoCollection, 'aggregate'], $args); $result = call_user_func_array([$this->mongoCollection, 'aggregate'], $args);
Yii::endProfile($token, __METHOD__);
return $result;
} }
/** /**
...@@ -297,6 +381,10 @@ class Collection extends Object ...@@ -297,6 +381,10 @@ class Collection extends Object
*/ */
public function mapReduce($keys, $initial, $reduce, $options = []) public function mapReduce($keys, $initial, $reduce, $options = [])
{ {
$token = 'Map reduce from ' . $this->getFullName();
Yii::info($token, __METHOD__);
Yii::beginProfile($token, __METHOD__);
if (!($reduce instanceof \MongoCode)) { if (!($reduce instanceof \MongoCode)) {
$reduce = new \MongoCode((string)$reduce); $reduce = new \MongoCode((string)$reduce);
} }
...@@ -314,6 +402,8 @@ class Collection extends Object ...@@ -314,6 +402,8 @@ class Collection extends Object
} else { } else {
$result = $this->mongoCollection->group($keys, $initial, $reduce, $options); $result = $this->mongoCollection->group($keys, $initial, $reduce, $options);
} }
Yii::endProfile($token, __METHOD__);
if (array_key_exists('retval', $result)) { if (array_key_exists('retval', $result)) {
return $result['retval']; return $result['retval'];
} else { } else {
...@@ -329,8 +419,20 @@ class Collection extends Object ...@@ -329,8 +419,20 @@ class Collection extends Object
protected function tryResultError($result) protected function tryResultError($result)
{ {
if (is_array($result)) { if (is_array($result)) {
if (!empty($result['err'])) { if (!empty($result['errmsg'])) {
throw new Exception($result['errmsg'], (int)$result['code']); $errorMessage = $result['errmsg'];
} elseif (!empty($result['err'])) {
$errorMessage = $result['err'];
}
if (isset($errorMessage)) {
if (array_key_exists('code', $result)) {
$errorCode = (int)$result['code'];
} elseif (array_key_exists('ok', $result)) {
$errorCode = (int)$result['ok'];
} else {
$errorCode = 0;
}
throw new Exception($errorMessage, $errorCode);
} }
} elseif (!$result) { } elseif (!$result) {
throw new Exception('Unknown error, use "w=1" option to enable error tracking'); throw new Exception('Unknown error, use "w=1" option to enable error tracking');
......
...@@ -15,6 +15,14 @@ class CollectionTest extends MongoTestCase ...@@ -15,6 +15,14 @@ class CollectionTest extends MongoTestCase
// Tests : // Tests :
public function testGetName()
{
$collectionName = 'customer';
$collection = $this->getConnection()->getCollection($collectionName);
$this->assertEquals($collectionName, $collection->getName());
$this->assertEquals($this->mongoConfig['defaultDatabaseName'] . '.' . $collectionName, $collection->getFullName());
}
public function testFind() public function testFind()
{ {
$collection = $this->getConnection()->getCollection('customer'); $collection = $this->getConnection()->getCollection('customer');
...@@ -181,5 +189,35 @@ class CollectionTest extends MongoTestCase ...@@ -181,5 +189,35 @@ class CollectionTest extends MongoTestCase
'status' => \MongoCollection::DESCENDING, 'status' => \MongoCollection::DESCENDING,
]; ];
$this->assertTrue($collection->createIndex($columns)); $this->assertTrue($collection->createIndex($columns));
$indexInfo = $collection->mongoCollection->getIndexInfo();
$this->assertEquals(2, count($indexInfo));
}
/**
* @depends testCreateIndex
*/
public function testDropIndex()
{
$collection = $this->getConnection()->getCollection('customer');
$collection->createIndex('name');
$this->assertTrue($collection->dropIndex('name'));
$indexInfo = $collection->mongoCollection->getIndexInfo();
$this->assertEquals(1, count($indexInfo));
$this->setExpectedException('\yii\mongo\Exception');
$collection->dropIndex('name');
}
/**
* @depends testCreateIndex
*/
public function testDropAllIndexes()
{
$collection = $this->getConnection()->getCollection('customer');
$collection->createIndex('name');
$this->assertEquals(1, $collection->dropAllIndexes());
$indexInfo = $collection->mongoCollection->getIndexInfo();
$this->assertEquals(1, count($indexInfo));
} }
} }
\ No newline at end of file
...@@ -5,6 +5,7 @@ namespace yiiunit\extensions\mongo; ...@@ -5,6 +5,7 @@ namespace yiiunit\extensions\mongo;
use yii\helpers\FileHelper; use yii\helpers\FileHelper;
use yii\mongo\Connection; use yii\mongo\Connection;
use Yii; use Yii;
use yii\mongo\Exception;
use yiiunit\TestCase; use yiiunit\TestCase;
class MongoTestCase extends TestCase class MongoTestCase extends TestCase
...@@ -95,7 +96,11 @@ class MongoTestCase extends TestCase ...@@ -95,7 +96,11 @@ class MongoTestCase extends TestCase
protected function dropCollection($name) protected function dropCollection($name)
{ {
if ($this->mongo) { if ($this->mongo) {
$this->mongo->getCollection($name)->drop(); try {
$this->mongo->getCollection($name)->drop();
} catch (Exception $e) {
// shut down exception
}
} }
} }
} }
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment