<?php
/*** Mongodb類**
 examples:
* $mongo = new HMongodb("127.0.0.1:11223");
* $mongo->selectDb("test_db");
* 創(chuàng)建索引
* $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"));
* 更新記錄-存在時(shí)更新，不存在時(shí)添加-相當(dāng)于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操作的錯(cuò)誤信息
* $mongo->getError();
*/
class HMongodb {
  public $_idType=1;//主鍵類型 1 string 0 obj
  //Mongodb連接
  var $mongo;
  var $curr_db_name;
  var $curr_table_name;
  var $error;
  /**
  * 構(gòu)造函數(shù)
  * 支持傳入多個(gè)mongo_server(1.一個(gè)出問(wèn)題時(shí)連接其它的server 2.自動(dòng)將查詢均勻分發(fā)到不同server)
  *
  * 參數(shù)：
  * $mongo_server:數(shù)組或字符串-array("127.0.0.1:1111", "127.0.0.1:2222")-"127.0.0.1:1111"
  * $connect:初始化mongo對(duì)象時(shí)是否連接，默認(rèn)連接
  * $auto_balance:是否自動(dòng)做負(fù)載均衡，默認(rèn)是
  *
  * 返回值：
  * 成功：mongo object
  * 失?。篺alse
  */
  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
  *
  * 參數(shù)：無(wú)
  *
  * 返回值：
  * 成功：true
  * 失?。篺alse
  */
  function connect()
  {
    try {
      $this->mongo->connect();
      return true;
    }
    catch (MongoConnectionException $e)
    {
      $this->error = $e->getMessage();
      return false;
    }
  }
  /**
  * select db
  *
  * 參數(shù)：$dbname
  *
  * 返回值：無(wú)
  */
  function selectDb($dbname)
  {
    $this->curr_db_name = $dbname;
  }
  /**
  * 創(chuàng)建索引：如索引已存在，則返回。
  *
  * 參數(shù)：
  * $table_name:表名
  * $index:索引-array("id"=>1)-在id字段建立升序索引
  * $index_param:其它條件-是否唯一索引等
  *
  * 返回值：
  * 成功：true
  * 失?。篺alse
  */
  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;
    }
  }
  /**
  * 插入記錄
  *
  * 參數(shù)：
  * $table_name:表名
  * $record:記錄
  *
  * 返回值：
  * 成功：true
  * 失?。篺alse
  */
  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;
    }
  }
  /**
  * 查詢表的記錄數(shù)
  *
  * 參數(shù)：
  * $table_name:表名
  *
  * 返回值：表的記錄數(shù)
  */
  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);
  }
  /**
  * 更新記錄
  *
  * 參數(shù)：
  * $table_name:表名
  * $condition:更新條件
  * $newdata:新的數(shù)據(jù)記錄
  * $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;
    }     }
  /**
  * 刪除記錄
  *
  * 參數(shù)：
  * $table_name:表名
  * $condition:刪除條件
  * $options:刪除選擇-justOne
  *
  * 返回值：
  * 成功：true
  * 失?。篺alse
  */
  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;
    }     }
  /**
  * 查找記錄
  *
  * 參數(shù)：
  * $table_name:表名
  * $query_condition:字段查找條件
  * $result_condition:查詢結(jié)果限制條件-limit/sort等
  * $fields:獲取字段
  *
  * 返回值：
  * 成功：記錄集
  * 失?。篺alse
  */
  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();//轉(zhuǎn)換字符串
        }
        $result[] =$row;
      }
    }
    catch (MongoConnectionException $e)
    {
      $this->error = $e->getMessage();
      return false;
    }
    catch (MongoCursorTimeoutException $e)
    {
      $this->error = $e->getMessage();
      return false;
    }
    return $result;
  }
  /**
  * 查找一條記錄
  *
  * 參數(shù)：
  * $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();//轉(zhuǎn)換字符串
    }
    return $row;
  }
  /**
  * 獲取當(dāng)前錯(cuò)誤信息
  *
  * 參數(shù)：無(wú)
  *
  * 返回值：當(dāng)前錯(cuò)誤信息
  */
  function getError()
  {
    return $this->error;
  }
  //返回原生mongodb collection集合對(duì)象
  function getCollection($table_name){
    $dbname = $this->curr_db_name;
    return $this->mongo->$dbname->$table_name;
  }
  //分組統(tǒng)計(jì)
  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;
  }
}