在别人基础上修改的mongodb操作类，适合mongodb2.x和mongodb3.x
<?php
/*** Mongodb类**
examples:
* $mongo = new HMongodb("127.0.0.1:11223");
* $mongo->selectDb("test_db");
* 创建索引
* $mongo->ensureIndex("test_table", array("id"=>1), array('unique'=>true));
* 获取表的记录
* $mongo->count("test_table");
* 插入记录
* $mongo->insert("test_table", array("id"=>2, "title"=>"asdqw"));
* 更新记录 array('$set'=>$newdata) 只修改指定字段
* $mongo->update("test_table", array("id"=>1),array("id"=>1,"title"=>"bbb"));
* 更新记录-存在时更新，不存在时添加-相当于set
* $mongo->update("test_table", array("id"=>1),array("id"=>1,"title"=>"bbb"),array("upsert"=>1));
* 查找记录
* $mongo->find("test_table", array("title"=>"asdqw"), array("start"=>2,"limit"=>2,"sort"=>array("id"=>1)))
* 查找一条记录
* $mongo->findOne("test_table", array("id"=>1));
* 删除记录
* $mongo->remove("ttt", array("title"=>"bbb"));
* 仅删除一条记录
* $mongo->remove("ttt", array("title"=>"bbb"), array("justOne"=>1));
* 获取Mongo操作的错误信息
* $mongo->getError();
*/
class HMongodb {
public $_idType=1;//主键类型 1 string 0 obj
//Mongodb连接
var $mongo;
var $curr_db_name;
var $curr_table_name;
var $error;
/**
* 构造函数
* 支持传入多个mongo_server(1.一个出问题时连接其它的server 2.自动将查询均匀分发到不同server)
*
* 参数：
* $mongo_server:数组或字符串-array("127.0.0.1:1111", "127.0.0.1:2222")-"127.0.0.1:1111"
* $connect:初始化mongo对象时是否连接，默认连接
* $auto_balance:是否自动做负载均衡，默认是
*
* 返回值：
* 成功：mongo object
* 失败：false
*/
function __construct($mongo_server, $connect=true, $auto_balance=true)
{
if (is_array($mongo_server))
{
$mongo_server_num = count($mongo_server);
if ($mongo_server_num > 1 && $auto_balance)
{
$prior_server_num = rand(1, $mongo_server_num);
$rand_keys = array_rand($mongo_server,$mongo_server_num);
$mongo_server_str = $mongo_server[$prior_server_num-1];
foreach ($rand_keys as $key)
{
if ($key != $prior_server_num - 1)
{
$mongo_server_str .= ',' . $mongo_server[$key];
}
}
}
else
{
$mongo_server_str = implode(',', $mongo_server);
} }
else
{
$mongo_server_str = trim($mongo_server);
//$config_arr=explode('/',$mongo_server);
//$this->curr_db_name = array_pop($config_arr);
$last_index=strrpos($mongo_server_str,'/');
$db_name=substr($mongo_server_str,$last_index+1,strlen($mongo_server_str));
$this->curr_db_name = $db_name;//获取dbname
//print_r($db_name);exit;
}
try {
$this->mongo = new MongoClient($mongo_server, array('connect'=>$connect));
}
catch (MongoConnectionException $e)
{
$this->error = $e->getMessage();
return false;
}
}
function getInstance($mongo_server, $flag=array())
{
static $mongodb_arr;
if (empty($flag['tag']))
{
$flag['tag'] = 'default'; }
if (isset($flag['force']) && $flag['force'] == true)
{
$mongo = new HMongodb($mongo_server);
if (empty($mongodb_arr[$flag['tag']]))
{
$mongodb_arr[$flag['tag']] = $mongo;
}
return $mongo;
}
else if (isset($mongodb_arr[$flag['tag']]) && is_resource($mongodb_arr[$flag['tag']]))
{
return $mongodb_arr[$flag['tag']];
}
else
{
$mongo = new HMongodb($mongo_server);
$mongodb_arr[$flag['tag']] = $mongo;
return $mongo; } }
/**
* 连接mongodb server
*
* 参数：无
*
* 返回值：
* 成功：true
* 失败：false
*/
function connect()
{
try {
$this->mongo->connect();
return true;
}
catch (MongoConnectionException $e)
{
$this->error = $e->getMessage();
return false;
}
}
/**
* select db
*
* 参数：$dbname
*
* 返回值：无
*/
function selectDb($dbname)
{
$this->curr_db_name = $dbname;
}
/**
* 创建索引：如索引已存在，则返回。
*
* 参数：
* $table_name:表名
* $index:索引-array("id"=>1)-在id字段建立升序索引
* $index_param:其它条件-是否唯一索引等
*
* 返回值：
* 成功：true
* 失败：false
*/
function ensureIndex($table_name, $index, $index_param=array())
{
$dbname = $this->curr_db_name;
$index_param['safe'] = 1;
try {
$this->mongo->$dbname->$table_name->ensureIndex($index, $index_param);
return true;
}
catch (MongoCursorException $e)
{
$this->error = $e->getMessage();
return false;
}
}
/**
* 插入记录
*
* 参数：
* $table_name:表名
* $record:记录
*
* 返回值：
* 成功：true
* 失败：false
*/
function insert($table_name, $record)
{
$dbname = $this->curr_db_name;
try {
$this->mongo->$dbname->$table_name->insert($record, array('safe'=>true));
return true;
}
catch (MongoCursorException $e)
{
$this->error = $e->getMessage();
return false;
}
}
/**
* 查询表的记录数
*
* 参数：
* $table_name:表名
*
* 返回值：表的记录数
*/
function count($table_name,$query_condition=array())
{
$dbname = $this->curr_db_name;
return $this->mongo->$dbname->$table_name->find($query_condition)->count();
}
//$newdata存在有_id则更新 不存在则插入
function save($table_name, $newdata) {
$dbname = $this->curr_db_name;
if($this->_idType && $newdata['_id']){
$newdata['_id']=new MongoId($newdata['_id']);
}
return $this->mongo->$dbname->$table_name->save($newdata);
}
/**
* 更新记录
*
* 参数：
* $table_name:表名
* $condition:更新条件
* $newdata:新的数据记录
* $options:更新选择-upsert/multiple
*
* 返回值：
* 成功：true
* 失败：false
*/
function update($table_name, $condition, $newdata, $options=array())
{ if($this->_idType && $condition['_id']){
$condition['_id']=new MongoId($condition['_id']);
}
$dbname = $this->curr_db_name;
$options['safe'] = 1;
if (!isset($options['multiple']))
{
$options['multiple'] = 0; }
try {
$this->mongo->$dbname->$table_name->update($condition, $newdata, $options);
return true;
}
catch (MongoCursorException $e)
{
$this->error = $e->getMessage();
return false;
} }
/**
* 删除记录
*
* 参数：
* $table_name:表名
* $condition:删除条件
* $options:删除选择-justOne
*
* 返回值：
* 成功：true
* 失败：false
*/
function remove($table_name, $condition, $options=array())
{
if($this->_idType && $condition['_id']){
$condition['_id']=new MongoId($condition['_id']);
}
$dbname = $this->curr_db_name;
$options['safe'] = 1;
try {
$this->mongo->$dbname->$table_name->remove($condition, $options);
return true;
}
catch (MongoCursorException $e)
{
$this->error = $e->getMessage();
return false;
} }
/**
* 查找记录
*
* 参数：
* $table_name:表名
* $query_condition:字段查找条件
* $result_condition:查询结果限制条件-limit/sort等
* $fields:获取字段
*
* 返回值：
* 成功：记录集
* 失败：false
*/
function find($table_name, $query_condition, $result_condition=array(), $fields=array())
{
$dbname = $this->curr_db_name;
$cursor = $this->mongo->$dbname->$table_name->find($query_condition, $fields);
if (!empty($result_condition['start']))
{
$cursor->skip($result_condition['start']);
}
if (!empty($result_condition['limit']))
{
$cursor->limit($result_condition['limit']);
}
if (!empty($result_condition['sort']))
{
$cursor->sort($result_condition['sort']);
}
$result = array();
try {
while ($cursor->hasNext())
{
$row= $cursor->getNext();
//print_r($cursor->getNext());exit;
if($this->_idType && is_object($row['_id'])){
$row['_id']=$row['_id']->__toString();//转换字符串
}
$result[] =$row;
}
}
catch (MongoConnectionException $e)
{
$this->error = $e->getMessage();
return false;
}
catch (MongoCursorTimeoutException $e)
{
$this->error = $e->getMessage();
return false;
}
return $result;
}
/**
* 查找一条记录
*
* 参数：
* $table_name:表名
* $condition:查找条件
* $fields:获取字段
*
* 返回值：
* 成功：一条记录
* 失败：false
*/
function findOne($table_name, $condition, $fields=array())
{
if($this->_idType && $condition['_id']){
$condition['_id']=new MongoId($condition['_id']);
}
$dbname = $this->curr_db_name;
$row=$this->mongo->$dbname->$table_name->findOne($condition, $fields);
if($this->_idType && is_object($row['_id'])){
$row['_id']=$row['_id']->__toString();//转换字符串
}
return $row;
}
/**
* 获取当前错误信息
*
* 参数：无
*
* 返回值：当前错误信息
*/
function getError()
{
return $this->error;
}
//返回原生mongodb collection集合对象
function getCollection($table_name){
$dbname = $this->curr_db_name;
return $this->mongo->$dbname->$table_name;
}
//分组统计
function group($table_name,$keys,$initial,$reduce,$condition){
$dbname = $this->curr_db_name;
$g = $this->mongo->$dbname->$table_name->group($keys, $initial, $reduce,$condition);
return $g;
}
}