<?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
|
*
|
*/
|
class System_Db_Mongo {
|
|
/**
|
* 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 = Yaf_Application::app()->getConfig();
|
$options = array();
|
$options['connect'] = $config->mongo->connect;
|
!empty($config->mongo->username) ? $options['username'] = $config->mongo->username : '';
|
!empty($config->mongo->password) ? $options['password'] = $config->mongo->password : '';
|
$options['db'] = $config->mongo->db;
|
//$server = sprintf('mongodb://%s:%s@%s:%s/%s', $config['username'], $config['password'], $config['host'], $config['port'], $config['db']);
|
|
// echo 'connect\n';
|
try {
|
$this->_mongo = new MongoClient($config->mongo->server, $options);// 立即连接
|
}catch (MongoConnectionException $e){
|
if(self::DEBUG) {
|
echo $e->getMessage();
|
}
|
return false;
|
}
|
|
$this->selectDB($config->mongo->db);
|
// 命令前缀
|
if(!isset($config->mongo->cmd)){
|
$this->_cmd = ini_get('mongo.cmd');
|
if($this->_cmd == ''){
|
$this->_cmd = '$';
|
}
|
}
|
}
|
|
/**
|
* 类结束调用时关闭连接
|
* [__destruct description]
|
*/
|
public function __destruct(){
|
$this->close();
|
}
|
|
/* ==================================== 基本操作接口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;
|
}
|
|
/**
|
* 聚合
|
* @param [type] $colName [description]
|
* @param [type] $params [description]
|
* @return [type] [description]
|
*/
|
public function aggregate($colName, $params){
|
$col = $this->_getCol($colName);
|
// 自动处理 '_id' 字段
|
//$query = $this->_parseId($query);
|
$result = $col->aggregate($params);
|
|
return $result['result'];
|
}
|
|
/**
|
* 返回最后的错误信息
|
*
|
* @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
|
?>
|
