/**

* Instance name

*

* @var string

*/

protected $_name;

/**

* Connected

*

* @var bool

*/

protected $_connected = FALSE;

/**

* Raw server connection

*

* @var Mongo

*/

protected $_connection;

/**

* Raw database connection

*

* @var Mongo Database

*/

protected $_db;

/**

* Local config

*

* @var array

*/

protected $_config;

/**

* @param $name

* @param array $config

*/

public function __construct($config_name = 'database')

{

$this->_config = Config::get('mongor::' . $config_name);

$this->connect();

return $this;

}

final public function __destruct()

{

$this->disconnect();

}

/**

* Connect to MongoDB, select database

*

* @return bool

*/

public function connect()

{

if ($this->_connection)

{

return;

}

/**

* Add required variables

* Clear the connection parameters for security

*/

$options = $this->_config;

unset($this->_config);

$hostname = array_shift($options);

$conn = 'mongodb://'. $hostname;

$this->_connection = new \Mongo($conn, $options);

/* Try connect */

try

{

$this->_connection->connect();

}

catch (MongoConnectionException $e)

{

throw new \Exception('Unable to connect to MongoDB server at ' . $hostname);

}

$this->_db = new \MongoDB($this->_connection, $options['db']);

return $this->_connected = TRUE;

}

/**

* Disconnect from MongoDB

*

* @returns null

*/

public function disconnect()

{

if ($this->_connection)

{

$this->_connection->close();

}

$this->_db = $this->_connection = NULL;

}

/* Database Management */

public function last_error()

{

return $this->_connected

? $this->_db->lastError()

: NULL;

}

public function prev_error()

{

return $this->_connected

? $this->_db->prevError()

: NULL;

}

public function reset_error()

{

return $this->_connected

? $this->_db->resetError()

: NULL;

}

public function command( array $data)

{

return $this->_call('command', array(), $data);

}

public function execute( $code, array $args = array() )

{

return $this->_call('execute', array(

'code' => $code,

'args' => $args

));

}

/* Collection management */

public function create_collection ( string $name, $capped= FALSE, $size= 0, $max= 0 )

{

return $this->_call('create_collection', array(

'name' => $name,

'capped' => $capped,

'size' => $size,

'max' => $max

));

}

public function drop_collection( $name )

{

return $this->_call('drop_collection', array(

'name' => $name

));

}

public function ensure_index ( $collection_name, $keys, $options = array())

{

return $this->_call('ensure_index', array(

'collection_name' => $collection_name,

'keys' => $keys,

'options' => $options

));

}

/* Data Management */

public function batch_insert ( $collection_name, array $a )

{

return $this->_call('batch_insert', array(

'collection_name' => $collection_name

), $a);

}

public function count( $collection_name, array $query = array() )

{

return $this->_call('count', array(

'collection_name' => $collection_name,

'query' => $query

));

}

public function find_one($collection_name, array $query = array(), array $fields = array())

{

return $this->_call('find_one', array(

'collection_name' => $collection_name,

'query' => $query,

'fields' => $fields

));

}

public function find($collection_name, array $query = array(), array $fields = array())

{

return $this->_call('find', array(

'collection_name' => $collection_name,

'query' => $query,

'fields' => $fields

));

}

public function group( $collection_name, $keys , array $initial , $reduce, array $condition= array() )

{

return $this->_call('group', array(

'collection_name' => $collection_name,

'keys' => $keys,

'initial' => $initial,

'reduce' => $reduce,

'condition' => $condition

));

}

public function update($collection_name, array $criteria, array $newObj, $options = array())

{

return $this->_call('update', array(

'collection_name' => $collection_name,

'criteria' => $criteria,

'options' => $options

), $newObj);

}

public function insert($collection_name, array $a, $options = array())

{

return $this->_call('insert', array(

'collection_name' => $collection_name,

'options' => $options

), $a);

}

public function remove($collection_name, array $criteria, $options = array())

{

return $this->_call('remove', array(

'collection_name' => $collection_name,

'criteria' => $criteria,

'options' => $options

));

}

public function save($collection_name, array $a, $options = array())

{

return $this->_call('save', array(

'collection_name' => $collection_name,

'options' => $options

), $a);

}

/* File management */

public function gridFS( $arg1 = NULL, $arg2 = NULL)

{

$this->_connected OR $this->connect();

if ( ! isset($arg1))

{

$arg1 = isset($this->_config['gridFS']['arg1'])

? $this->_config['gridFS']['arg1']

: 'fs';

}

if ( ! isset($arg2) && isset($this->_config['gridFS']['arg2']))

{

$arg2 = $this->_config['gridFS']['arg2'];

}

return $this->_db->getGridFS($arg1,$arg2);

}

public function get_file(array $criteria = array())

{

return $this->_call('get_file', array(

'criteria' => $criteria

));

}

public function get_files(array $query = array(), array $fields = array())

{

return $this->_call('get_files', array(

'query' => $query,

'fields' => $fields

));

}

public function set_file_bytes($bytes, array $extra = array(), array $options = array())

{

return $this->_call('set_file_bytes', array(

'bytes' => $bytes,

'extra' => $extra,

'options' => $options

));

}

public function set_file($filename, array $extra = array(), array $options = array())

{

return $this->_call('set_file', array(

'filename' => $filename,

'extra' => $extra,

'options' => $options

));

}

public function remove_file( array $criteria = array(), array $options = array())

{

return $this->_call('remove_file', array(

'criteria' => $criteria,

'options' => $options

));

}

/* Run Command */

protected function _call($command, array $arguments = array(), array $values = NULL)

{

$start = microtime(true);

$this->_connected OR $this->connect();

extract($arguments);

if (isset($collection_name))

{

$c = new \MongoCollection($this->_db, $collection_name);

}

switch ($command)

{

case 'ensure_index':

$r = $c->ensureIndex($keys, $options);

break;

case 'create_collection':

$r = $this->_db->createCollection($name, $capped, $size, $max);

break;

case 'drop_collection':

$r = $this->_db->dropCollection($name);

break;

case 'command':

$r = $this->_db->command($values);

break;

case 'execute':

$r = $this->_db->execute($code, $args);

break;

case 'batch_insert':

$r = $c->batchInsert($values);

break;

case 'count':

$r = $c->count($query);

break;

case 'find_one':

$r = $c->findOne($query, $fields);

break;

case 'find':

$r = $c->find($query, $fields);

break;

case 'group':

$r = $c->group($keys, $initial, $reduce, $condition);

break;

case 'update':

$r = $c->update($criteria, $values, $options);

break;

case 'insert':

$r = $c->insert($values, $options);

return $values;

break;

case 'remove':

$r = $c->remove($criteria, $options);

break;

case 'save':

$r = $c->save($values, $options);

break;

case 'get_file':

$r = $this->gridFS()->findOne($criteria);

break;

case 'get_files':

$r = $this->gridFS()->find($query, $fields);

break;

case 'set_file_bytes':

$r = $this->gridFS()->storeBytes($bytes, $extra, $options);

break;

case 'set_file':

$r = $this->gridFS()->storeFile($filename, $extra, $options);

break;

case 'remove_file':

$r = $this->gridFS()->remove($criteria, $options);

break;

}

$this->log($command, $start, $arguments);

return $r;

}

protected function log($command, $start, $arguments) {

$time = number_format((microtime(true) - $start) * 1000, 2);

\Laravel\Event::fire('laravel.mongoquery', array($this->_db, $command, $arguments, $time));

}