wpstudio
/
majestic
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
Browse Source

Mongo driver with separate CRUD commsnds, without profiler

master
Anton Grebnev 13 years ago
parent
3c5a5d9603
commit
488655b788
3 changed files with 149 additions and 11 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 54
      model/MongoDriver.php
  2. 3
      model/NoSqlDbDriver.php
  3. 103
      tests/model/MongoDriverTest.php

54
model/MongoDriver.php
View File

@ -13,30 +13,65 @@
class MongoDriver extends NoSqlDbDriver
{
public function find($data, $params = array(), $cache_key = null)
protected $last_inset_id = 0;
protected $db_name = 'admin';
protected function getCollection($name)
{
return $this->connection->selectCollection($this->db, $name);
}
public function find($collection, $condition = array(), $fields = array(), $cache_key = null)
{
return $this->getCollection($collection)->find($condition, $fields);
}
public function insert($table, $bind, $on_duplicate = array())
public function get($collection, $condition, $fields = array())
{
return parent::insert($table, $bind, $on_duplicate);
return $this->getCollection($collection)->findOne($condition, $fields);
}
public function update($table, $bind, $where = '')
public function insert($collection, $data, $safe = true)
{
return parent::update($table, $bind, $where);
$result = $this->getCollection($collection)->insert($data, array('safe' => $safe));
if (isset($result['ok']) && $result['ok'] == 1) {
$this->last_inset_id = (string) $data['_id'];
return true;
} else {
return false;
}
}
public function delete($table, $where = '')
public function update($collection, $data, $condition = array(), $upsert = false, $safe = true)
{
return parent::delete($table, $where);
$result = $this->getCollection($collection)->update($condition, $data, array('upsert' => $upsert, 'safe' => $safe));
if (isset($result['ok']) && $result['ok'] == 1) {
if(isset($result['updatedExisting']) && isset($result['upserted'])) {
$this->last_inset_id = (string) $result['upserted'];
}
return $result['n'];
} else {
return false;
}
}
public function delete($collection, $condition = array(), $safe = true)
{
$result = $this->getCollection($collection)->remove($condition, array('safe' => $safe));
if (isset($result['ok']) && $result['ok'] == 1) {
return $result['n'];
} else {
return false;
}
}
public function getInsertId($table = null, $key = null)
{
// TODO: Implement getInsertId() method.
return $this->last_inset_id;
}
public function isConnected()
@ -72,6 +107,7 @@ class MongoDriver extends NoSqlDbDriver
);
$this->connection = new Mongo('mongodb://' . $host . $port, $this->config);
$this->db = $this->connection->selectDB($this->config['db']);
}
}

3
model/NoSqlDbDriver.php
View File

@ -9,5 +9,6 @@
abstract class NoSqlDbDriver extends DbDriver
{
abstract function find($data, $params = array(), $cache_key = null);
abstract function find($collection, $condition = array(), $fields = array(), $cache_key = null);
}

103
tests/model/MongoDriverTest.php
View File

@ -28,7 +28,45 @@ class MongoDriverTest extends PHPUnit_Framework_TestCase
'password' => '1234',
'port' => 27017
);
$data = array(
array(
'name' => 'bread',
'price' => 3.2,
'quantity' => 10
),
array(
'name' => 'eggs',
'price' => 2.1,
'quantity' => 20
),
array(
'name' => 'fish',
'price' => 13.2,
'quantity' => 2
),
array(
'name' => 'milk',
'price' => 3.8,
'quantity' => 1
),
array(
'name' => 'eggs',
'price' => 2.3,
'quantity' => 5
)
);
$connection = new Mongo('mongodb://' . $this->conf['hostname'] . ':' . $this->conf['port']);
$db = $connection->selectDB($this->conf['database']);
$db->authenticate($this->conf['username'], $this->conf['password']);
$collection = 'items';
$db->dropCollection($collection);
$collection = $db->selectCollection($collection);
foreach($data as $document) {
$collection->insert($document);
}
}
/**
* @expectedException Exception
* @expectedExceptionMessage Configuration must have a "hostname".
@ -48,7 +86,7 @@ class MongoDriverTest extends PHPUnit_Framework_TestCase
{
$this->conf['password'] = 'nopass';
$mongo = new MongoDriver($this->conf);
$this->assertInstanceOf('Mongo', $mongo->getConnection());
$this->assertInstanceOf('MongoDB', $mongo->getConnection());
}
public function testGetConnection()
@ -58,8 +96,71 @@ class MongoDriverTest extends PHPUnit_Framework_TestCase
$this->assertFalse($mongo->isConnected());
$this->assertInstanceOf('Mongo', $mongo->getConnection());
$this->assertTrue($mongo->isConnected());
$mongo->getConnection();
$mongo->disconnect();
$this->assertFalse($mongo->isConnected());
}
public function testFind()
{
$mongo = new MongoDriver($this->conf);
$mongo->getConnection();
$this->assertEquals(1, $mongo->find('items', array('name' => 'bread'))->count());
$this->assertEquals(2, $mongo->find('items', array('name' => 'eggs'))->count());
$this->assertEquals(3, $mongo->find('items', array('price' => array('$lt' => 3.5)))->count());
$this->assertEquals(5, $mongo->find('items', array())->count());
}
public function testGet()
{
$mongo = new MongoDriver($this->conf);
$mongo->getConnection();
$bread = $mongo->get('items', array('name' => 'bread'));
$this->assertEquals(3.2, $bread['price']);
}
public function testRemove()
{
$mongo = new MongoDriver($this->conf);
$mongo->getConnection();
$this->assertEquals(1, $mongo->find('items', array('name' => 'bread'))->count());
$this->assertEquals(0, $mongo->delete('items', array('name' => 'esggs')));
$this->assertEquals(2, $mongo->delete('items', array('name' => 'eggs')));
$this->assertEquals(1, $mongo->delete('items', array('name' => 'bread')));
$this->assertEquals(0, $mongo->find('items', array('name' => 'bread'))->count());
}
public function testInsert()
{
$mongo = new MongoDriver($this->conf);
$mongo->getConnection();
$this->assertEquals(1, $mongo->find('items', array('name' => 'bread'))->count());
$this->assertTrue($mongo->insert('items', array('name' => 'bread')));
$this->assertNotEmpty($mongo->getInsertId());
$this->assertEquals(2, $mongo->find('items', array('name' => 'bread'))->count());
}
public function testUpdate()
{
$mongo = new MongoDriver($this->conf);
$mongo->getConnection();
$this->assertEquals(1, $mongo->find('items', array('name' => 'bread'))->count());
$mongo->update('items', array('$set' => array('price' => 200, 'date' => 'today')), array('name' => 'fish'));
$fish = $mongo->get('items', array('name' => 'fish'));
$this->assertEquals(200, $fish['price']);
$this->assertEquals('today', $fish['date']);
$this->assertEquals(0, $mongo->update('items', array('$set' => array('price' => 200, 'date' => 'today')), array('name' => 'ball')));
}
public function testUpsert()
{
$mongo = new MongoDriver($this->conf);
$mongo->getConnection();
$this->assertTrue($mongo->insert('items', array('name' => 'bread')));
$id = $mongo->getInsertId();
$this->assertNotEmpty($id);
$this->assertEquals(1, $mongo->update('items', array('$set' => array('price' => 200, 'date' => 'today')), array('name' => 'ball'), true));
$this->assertNotEquals($id, $mongo->getInsertId());
}
}
Write Preview
Loading…
Cancel
Save