<?php
/**
* Mongodb 基本操作API，支持基本类似关系统型数据库的操作接口
*
* @version 1.0
 * [说明]
 *
 * 1：该版本API实现了 Mongodb 中最基本的插入/修改/查询/删除操作的封装
 * 2：其它更高级的操作可通过 $this->getMongo() 得到原生的对象，更多API请自行查阅 Mongo PHP手册,后续版本将会对增加更多的原生API封装
 * 3：该类所有API接口中的 $query 查询参数的使用请以下有关 [查询条件说明文档]
 * 4: 如果要存储中文字符，则一定要使用 utf8 的编码．
 * 5：有了本类接口基本可以按关系型数据库的概念完成Mongodb的大部分开发操作。
 *
 * [查询条件说明文档]
 *
 * 参数：array('id'=>1)
 * 等同：where id=1
 *
 * 参数：array('id'=>1,'num'=>10)
 * 等同：where id=1 and num=10
 *
 * 参数：array('id'=>array($mongo->cmd('>')=>5))
 * 等同：where id>5
 *
 * 参数：array('id'=>array($mongo->cmd('!=')=>5))
 * 等同：where id!=5
 *
 * 参数：array('id'=>array($mongo->cmd('>')=>5, $mongo->cmd('<')=>10))
 * 等同：where id>5 and id<10
 *
 * 参数：array('id'=>array($mongo->cmd('in')=>array(2,5,6)))
 * 等同：where id in (2,5,6)
 *
 * 参数：array('id'=>array($mongo->cmd('%')=>array(2,1)))
 * 等同：where id % 2 = 1
 *
 * 参数：array($mongo->cmd('or') => array( array('id'=>array($mongo->cmd('>')=>5)), array('id'=>array($mongo->cmd('<')=>10)) ) )
 * 等同：where id>5 or id<10
 *
 * 参数：array('username' => new mongoRegex("/yhp.*/")) 
 * 等同 where  username  like "%yhp%"
 **/
class Library_Mongodb {

    /**
     * Mongodb 对象句柄
     *
     * @var object Mongo
     */
    private $_mongo = null;

    /**
     * 当前选择的数据库
     *
     * @var object MongoDB
     */
    private $_db = null;

    /**
     * 修改器命令前缀
     *
     * @var string
     */
    private $_cmd = '$';


    /**
     * 调试模式 TRUE 打开 FALSE 关闭
     * @var boolean
     */
    const DEBUG = TRUE;

    /**
     * 查询条件映射关系
     *
     * @var array
     */
    private $_condMap = array(
        '<'	=>    'lt', // id > 1
        '<='	=>    'lte', // id <= 10
        '>'	=>    'gt', // id > 5
        '>='    =>    'gte', // id >= 4
        '!='    =>    'ne', // id != 4
        '%'	=>    'mod', // id % 4 = 0
        'in'    =>    'in', // id in (1,2,3,4)
        'notin'	=>    'nin',// id not in (1,2,3,4)
        'or'    =>    'or', // id=1 or id=2
        'not'	=>    'not', // !(id=1)
    );


    /**
     * 构造函数
     *
     * @param array $config 服务器配置,默认为:
     * array(
     * 'host'=>'localhost', // 主机名或IP地址
     * 'port'=>27017, // 端口
     * 'cmd'=>'$', // 修改器命令前缀
     * )
     */
    public function __construct($config = array('host' => 'xxx', 'port' => 27017, 'username' => 'xxx', 'password' => 'xxx', 'db' => 'xxx',  'cmd' => '$')){
        $server = sprintf("mongodb://%s:%s@%s:%s/%s", $config['username'], $config['password'], $config['host'], $config['port'], $config['db']);

        try {
            $this->_mongo = new MongoClient($server, array('connect'=>true));// 立即连接
        }catch (MongoConnectionException $e){
            if(self::DEBUG) {
                echo $e->getMessage();
            }
            return false;
        }

        $this->selectDB($config['db']);
        // 命令前缀
        if(!isset($config['cmd'])){
            $this->_cmd = ini_get('mongo.cmd');
            if($this->_cmd == ''){
                $this->_cmd = '$';
            }
        }
    }

    /* ==================================== 基本操作接口API　================================= */

    /**
     * 向集合(表)中插入新文档
     *
     * 说明：
     * 1:类似mysql中的: insert into $colName set id=1,name='name1';
     *
     * @param string $colName 集合名
     * @param array $sets 数据,如: array('id'=>1,'name'=>'name1')
     * @param boolean $safe 是否安全操作 false:不等待服务器的响应直接返回 true:等待服务器的响应(数据非常重要时推荐)
     * @param boolean $fsync 操作后是否立即更新到碰盘,默认情况下由服务器决定
     *
     * @return boolean
     */
    public function insert($colName, $sets, $safe=false, $fsync=false){
        $col = $this->_getCol($colName);
        try {
            $col->insert($sets,array('w'=>$safe,'fsync'=>$fsync));
            return true;
        }catch (MongoCursorException $e){
            return false;
        }
    }

    /**
     * 保存文档
     *
     * 说明：
     * 1:如果 $sets 中有字段 "_id" 的话，则更新对应的文档；否则插入新文档
     *
     * @param string $colName 集合名
     * @param array $sets 数据,如: array('id'=>1,'name'=>'name1')
     * @param boolean $safe 是否安全操作 false:不等待服务器的响应直接返回 true:等待服务器的响应(数据非常重要时推荐)
     * @param boolean $fsync 操作后是否立即更新到碰盘,默认情况下由服务器决定
     *
     * @return boolean
     */
    public function save($colName, $sets, $safe=false, $fsync=false){
        // 处理 '_id' 字段
        $sets = $this->_parseId($sets);
        $ret = $this->_getCol($colName)->save($sets,array('w'=>$safe,'fsync'=>$fsync));
        return $ret;
    }

    /**
     * 删除集合中的文档记录
     *
     * 说明：
     * 1：类似mysql中的: delete from $colName where id=1;
     *
     * @param string $colName 集合名
     * @param array $query 查询条件,如果为空数组的话，则会删除所有记录．具体请看 [查询条件说明文档]
     * @param boolean $delAll 是否删除所以条例查询的记录,默认为 true,当为 false是，类似效果 delete from tab where id=1 limit 1;
     * @param boolean $safe 是否安全操作 false:不等待服务器的响应直接返回 true:等待服务器的响应(数据非常重要时推荐)
     * @param boolean $fsync 操作后是否立即更新到碰盘,默认情况下由服务器决定
     *
     * @return boolean
     */
    public function delete($colName,$query=array(),$delAll=true,$safe=false,$fsync=false){
        // 自动处理 '_id' 字段
        $query = $this->_parseId($query);
        // 删除选项
        $options = array(
            'justOne'    =>    !$delAll,
            'w'            =>    $safe,
            'fsync'        =>    $fsync,
        );
        $col = $this->_getCol($colName);
        return $col->remove($query,$options);
    }

    /**
     * 删除整个集合
     *
     * 说明：
     * 1：集合中的索引也会被删除
     *
     * @param string $colName 集合名
     *
     * @return array
     */
    public function dropCol($colName){
        return $this->_getCol($colName)->drop();
    }

    /**
     * 更新集合文档记录
     *
     * 说明:
     * 1：类似mysql中的: update $colName set name='mongo' where id=10;
     *
     * @param string $colName 集合名
     * @param array $newDoc 要更新的文档记录
     * @param array $query 查询条件,如果为空数组则更新所有记录．具体请看 [查询条件说明文档]
     * @param string $option 操作选项,可选择项如下；
     *
     * 'set'：只修改指定的字段（默认值,如果这个键不存在，则创建它。存在则更新）.
     * 示例: update('user', array('name'=>'mongo'), array('id'=>10));
     * 类似: update user set name='mongo' where id=10;
     *
     * 'inc'：将指定的字段累加/减(如果值为负数则是相减,不存在键则创建。字段类型一定要是数字)
     * 示例：update('user', array('num'=>1), array('id'=>10), 'inc');
     * 类似: update user set num=num+1 where id=10;
     *
     * 'push'：将文档添加到指定键中（数组），如果键不存在则会自动创建，存在则添加到该键的尾端。
     * 示例：update('user', array('comm'=>array('commid'=>1,'title'=>'title1')), array('id'=>1), 'push');
     * 解说：为 id=1 的记录添加一个 comm 的评论字段，该字段对应一个 array('commid'=>1,'title'=>'title1') 的新文档。
     *
     * 'pop':将指定键中的文档删除（数组）
     * 示例：update('user', array('comm'=>array('commid'=>1)), array('id'=>1), 'pop');
     * 解说：删除 id=1 的记录中 comm 对应的文档集合中 'commid'=>1 对应的文档.
     *
     * 'unset':在文档中删除指定的键
     * 示例：update('user', array('name'=>1), array('id'=>1), 'unset');
     * 解说: 将 user 集合中将 id=1 对应的文档中的 name 字段删除
     *
     * 'pull':删除文档中匹配其值的键
     * 示例：update('user', array('name'=>'youname'), array('id'=>1), 'pull');
     * 解说：将 user 集合中将 id=1 对应的文档中的 name='youname' 的字段删除
     *
     * 'addToSet':如果值不存在就添加（避免重复添加）
     * 示例：update('user', array('names'=>'youname'), array('id'=>1), 'addToSet');
     * 解说：向 user 集合中 id=1 对应的文档中的 names 字段添加 'youname' 这个值(不存在时才添加)
     *
     * 'replace'：用 $newDoc 新文档替换 $query 所找到的文档
     * 示例：update('user', array('newid'=>1,'newnames'=>'name1'), array('id'=>1), 'replace');
     * 解说：将 user 集合中 id=1 对应的文档用 array('newid'=>1,'newnames'=>'name1') 的新文档替换
     *
     * @param boolean $upAll 是否更新找到的所有记录
     * @param boolean $upsert 如果查询条件不存在时，是否以查询条件和要更新的字段一起新建一个集合
     * @param boolean $safe 是否安全删除 false:不等待服务器的响应直接返回 true:等待服务器的响应(数据非常重要时推荐)
     * @param boolean $fsync 操作后是否立即更新到碰盘,默认情况下由服务器决定
     *
     * @return boolean
     */
    public function update($colName,$newDoc,$query=array(),$option='set',$upAll=true,$upsert=false,$safe=false,$fsync=false){
        // 自动处理 '_id' 字段
        $query = $this->_parseId($query);
        // 得到集合
        $col = $this->_getCol($colName);
        // 重新组合新文档
        if($option != 'replace'){
            $newDoc = array($this->cmd($option) => $newDoc);
        }
        // 更新条件
        $options = array(
            'upsert'    =>    $upsert,
            'multiple'    =>    $upAll,
            'w'            =>    $safe,
            'fsync'        =>    $fsync,
        );
        return $col->update($query,$newDoc,$options);
    }

    /**
     * 查询文档集,返回二维数组
     *
     * 说明:
     * 1:类似mysql中的 select * from table
     *
     * 示例：select('user');
     * 类似：select * from user;
     *
     * 示例：select('user',array('id','name'));
     * 类似：select id,name from user;
     *
     * 示例：select('user',array('id','name'),array('id'=>1));
     * 类似：select id,name from user where id=1;
     *
     * 示例：select('user',array('id','name'),array('id'=>1),array('num'=>1));
     * 类似：select id,name from user where id=1 order by num asc;
     *
     * 示例：select('user',array('id','name'),array('id'=>1),array('num'=>1),10);
     * 类似：select id,name from user where id=1 order by num asc limit 10;
     *
     * 示例：select('user',array('id','name'),array('id'=>1),array('num'=>1),10,2);
     * 类似：select id,name from user where id=1 order by num asc limit 2,10;
     *
     *
     *
     * @param string $colName 集合名
     * @param array $query 查询条件,具体请看 [查询条件说明文档]
     * @param array $fields 结果集返回的字段, array():表示返回所有字段 array('id','name'):表示只返回字段 "id,name"
     * @param array $sort 排序字段, array('id'=>1):表示按id字段升序 array('id'=>-1):表示按id字段降序 array('id'=>1, 'age'=>-1):表示按id升序后再按age降序
     * @param int $limit 取多少条记录
     * @param int $skip 跳过多少条(从多少条开始)
     *
     * @return array
     */
    public function select($colName,$query=array(),$fields=array(),$sort=array(),$limit=0,$skip=0){
        // 得到集合
        $col = $this->_getCol($colName);
        // 自动处理 '_id' 字段
        $query = $this->_parseId($query);
        // 结果集偏历
        $cursor  = $col->find($query,$fields);
        // 排序
        if($sort){
            $cursor->sort($sort);
        }
        // 跳过记录数
        if($skip > 0){
            $cursor->skip($skip);
        }
        // 取多少行记录
        if($limit > 0){
            $cursor->limit($limit);
        }
        $result = array();
        foreach($cursor  as $row){
            $result[] = $this->_parseArr($row);
        }
        return $result;
    }

    /**
     * 统计文档记录数
     *
     * @param string $colName 集合名
     * @param array $query 查询条件,具体请看 [查询条件说明文档]
     * @param int $limit 取多少条记录
     * @param int $skip 跳过多少条
     * @return unknown
     */
    public function count($colName,$query=array(),$limit=0,$skip=0){
        return $this->_getCol($colName)->count($query,$limit,$skip);
    }

    /**
     * 返回集合中的一条记录(一维数组)
     *
     * @param string $colName 集合名
     * @param array $query 查询条件,具体请看 [查询条件说明文档]
     * @param array $fields 结果集返回的字段, array():表示返回所有字段 array('id','name'):表示只返回字段 "id,name"
     *
     * @return array
     */
    public function fetchRow($colName,$query=array(), $fields=array()){
        // 得到集合名
        $col = $this->_getCol($colName);
        // 自动处理 '_id' 字段
        $query = $this->_parseId($query);
        // 处理结果集
        return $this->_parseArr($col->findOne($query,$fields));
    }

    /**
     * 返回符合条件的文档中字段的值
     *
     * @param string $colName 集合名
     * @param array $query 查询条件,具体请看 [查询条件说明文档]
     * @param string $fields 要取其值的字段,默认为 "_id" 字段,类似mysql中的自增主键
     *
     * @return mixed
     */
    public function fetchOne($colName,$query=array(), $fields='_id'){
        $ret = $this->fetchRow($colName,$query,array($fields));
        return isset($ret[$fields]) ? $ret[$fields] : false;
    }

    /**
     * 返回查询文档集合集中指定字段的值(一维数组)
     *
     * @param string $colName 集合名
     * @param array $query 查询条件,具体请看 [查询条件说明文档]
     * @param string $fields 要取其值的字段,默认为 "_id" 字段,类似mysql中的自增主键
     *
     * @return array
     */
    public function fetchCol($colName,$query=array(), $fields='_id'){
        $result = array();
        $list = $this->select($colName,$query,array($fields));
        foreach ($list as $row){
            $result[] = $row[$fields];
        }
        return $result;
    }

    /**
     * 返回指定下标的查询文档集合(二维数组)
     *
     * @param string $colName 集合名
     * @param array $query 查询条件,具体请看 [查询条件说明文档]
     * @param string $fields 要取其值的字段,默认为 "_id" 字段,类似mysql中的自增主键
     *
     * @return array
     */
    public function fetchAssoc($colName,$query=array(), $fields='_id'){
        $result = array();
        $list = $this->select($colName,$query);
        foreach ($list as $row){
            $key = $row[$fields];
            $result[][$key] = $row;
        }
        return $result;
    }

    /* ==================================== 辅助操作接口API　================================= */

    /**
     * 返回命令或命令前缀
     *
     * @param string $option 命令，如果为空时则返回命令前缀
     *
     * @return string
     */
    public function cmd($option=''){
        // 只返回命令前缀
        if($option == ''){
            return $this->_cmd;
        }
        // 如果是操作符
        if(isset($this->_condMap[$option])){
            $option = $this->_condMap[$option];
        }
        return $this->_cmd.$option;
    }

    /**
     * 选择或创建数据库(注意：新创建的数据库如果在关闭连接前没有写入数据将会被自动删除)
     *
     * @param string $dbname 数据库名
     */
    public function selectDB($dbname){
        $this->_db = $this->_mongo->selectDB($dbname);
    }

    /**
     * 得到所有的数据库
     *
     * @param boolean $onlyName 是否只返回数据库名的数组
     * @return array
     */
    public function allDB($onlyName=false){
        $ary = $this->_mongo->listDBs();
        if($onlyName){
            $ret = array();
            foreach ($ary['databases'] as $row){
                $ret[] = $row['name'];
            }
            return $ret;
        }else{
            return $ary;
        }
    }

    /**
     * 删除数据库
     *
     * @return array
     */
    public function dropDB($dbname){
        return $this->_mongo->dropDB($dbname);
    }

    /**
     * 关闭连接
     *
     */
    public function close(){
        $this->_mongo->close();
    }

    /**
     * 得到 Mongo 原生对象，进行其它更高级的操作，详细请看PHP手册
     *
     */
    public function getMongo(){
        return $this->_mongo;
    }

    /**
     * 返回最后的错误信息
     *
     * @return array
     */
    public function getError(){
        return $this->_db->lastError();
    }

    /* ======================= 以下为私有方法 ====================== */

    // 解析数据组中的'_id'字段(如果有的话)
    private function _parseId($arr){
        if(isset($arr['_id'])){
            $arr['_id'] = new MongoId($arr['_id']);
        }
        return $arr;
    }

    // 得到集合对象
    private function _getCol($colName){
        return $this->_db->selectCollection($colName);
    }

    // 解析数组中的"_id"并且返回
    private function _parseArr($arr){
        if(!empty($arr)) {
            $ret = (array)$arr['_id'];
            $arr['_id'] = $ret['$id'];
        }
        return $arr;
    }
}//End Class
?>