1. 程式人生 > >php mongodb操作類,適合mongodb2.x和mongodb3.x

php mongodb操作類,適合mongodb2.x和mongodb3.x

在別人基礎上修改的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;
	}	
	
}