php mongodb操作類,適合mongodb2.x和mongodb3.x
阿新 • • 發佈:2018-12-24
在別人基礎上修改的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; } }