* Date: 13-10-12
* Time: 下午5:48
*/
class Mongo_DB
{
#链接
public $conn = null;
#mongodb
public $db = null;
#链接参数
private $url = '';
#判断db是否链接
private $collected = null;
#查询、更新 映射 有|
private $four_map = array( #值是字符串,值不用处理
'>' => '$gt',
'>=' => '$gte',
'<' => '$le',
'<=' => '$lte',
'!=' => '$ne',
'@' => '$exists', #键是否存在
'?' => 'perl', #正则
);
private $ele_map = array( #数组,值需要处理
'~' => '$elemMatch' #内嵌文档
);
private $log_map = array( #值必须是数组
'^' => '$in',
'!^' => '$nin',
'*' => '$all', #指定数组元素
);
#没有|
private $lgi_map = array( #必须是二维数组
'$' => '$or',
'&' => '$and',
);
private $where_map = array(
'%' => '$where', #javascript 函数
);
private $sort_map = array(
'DESC' => -1,
'ASC' => 1
);
#当前db下所有集合=>键数组
public $collections = null;
#基本配置参数
private $config = array(
'host' => 'localhost',
'port' => '27017',
'dbname' => '',
'user' => '',
'pass' => '',
'logpath' => '', #日志目录
'replicaset' => array( #集群
'host' => array()
// 'host' => array('host:port'), #主机群
// 'options' => array('readPreference', 'readPreferenceTags', 'replicaSet','slaveok')
),
);
/**
* 构造函数，基本参数设置
* @param $config
*
*/
public function __construct($config)
{
if (is_array($config) && $config) {
foreach ($this->config as $k => $v) {
if (isset($config[$k]) && $config[$k])
$this->config[$k] = !is_array($config[$k]) ? trim($config[$k]) : $config[$k];
}
$this->connect();
}
}
/**
* 初始化mongo
*
*/
private function connect()
{
if (!extension_loaded('mongo'))
exit('ERROR:MONGO EXTENSION NOT LOAD');
if ($this->conn === NULL) {
try {
if ($this->config['replicaset']['host']) {
#todo 主从配置，或者副本集，读扩展,
} else {
$this->url .= "mongodb://{$this->config['user']}:{$this->config['pass']}@{$this->config['host']}:
{$this->config['port']}/{$this->config['dbname']}";
$this->conn = new MongoClient($this->url);
}
$this->db = $this->conn->selectDB($this->config['dbname']);
if ($this->db) {
$this->collected = true;
$this->GetAllCollection(); #获取所有集合列表
}
} catch (MongoConnectionException $e) {
$this->ErrLog($e->getMessage());
exit('MONGO CONNECT ERROR');
}
}
}
/**
* 析构函数
*/
public function __destruct()
{
$this->collected = $this->collections = $this->db = $this->conn = null;
}
/**
* 获取db下所有集合名称
*
* @internal param $dbname
* @return bool
*/
private function GetAllCollection()
{
if (!$this->collected)
return false;
$colls = $this->db->listCollections();
if ($colls) {
foreach ($colls as $v) {
$pos = strpos($v, $this->config['dbname']);
if ($pos !== false)
$this->collections[] = substr($v, ($pos + strlen($this->config['dbname']) + 1));
}
return true;
} else {
return false;
}
}
/**
* 初始化集合对象
* @param $collname 集合名
* @return mixed
*/
public function GetMonCollection($collname)
{
if (!$this->collected)
return false;
return $this->db->selectCollection($collname);
}
/**
* 转换字符编码
* @param $array
* @return bool
*/
private function ConvertEncode($array)
{
if (is_array($array)) {
foreach ($array as &$v) {
if (is_array($v)) {
$v = $this->ConvertEncode($v);
if (!$v)
return false;
} else {
$code = mb_detect_encoding($v, array('UTF-8', 'ASCII', 'GB2312', 'GBK', 'CP936'), true);
if ($code !== 'UTF-8')
$v = mb_convert_encoding($v, 'UTF-8', $code);
}
}
return $array;
}
return false;
}
/**
* 获取where参数
* @param $where
* @param bool $type
* @return string
*
*/
private function GetWhere($where, $type = false)
{
$wheres = array();
$maps = array('four_map', 'log_map', 'ele_map');
$maps_np = array_merge($this->lgi_map, $this->where_map);
if (is_array($where) && $where) { #过滤查询条件
foreach ($where as $field => $val) {
$pos = strpos($field, '|');
if ($pos !== false) { #四则、正则、函数、数组、in、多重条件内嵌文档
$tep = substr($field, 0, $pos);
$key = substr($field, ($pos + 1));
if ($key !== false) {
foreach ($maps as $v) {
$arr = $this->$v;
if (in_array($tep, array_keys($arr))) {
if ($v == 'ele_map' && is_array($val))
$val = $this->GetWhere($val, true);
if ($tep == '?') { #正则
$val = new MongoRegex($val);
$wheres[$key] = $val;
} else {
$wheres[$key][$arr[$tep]] = $val;
}
}
}
}
} elseif (in_array($field, array_keys($maps_np))) {
if (in_array($field, array_keys($this->lgi_map)) && is_array($val)) { #逻辑
foreach ($val as $v) {
$val = $this->GetWhere($v, true);
$wheres[$maps_np[$field]][] = $val;
}
} else {
$wheres[$maps_np[$field]] = $val;
}
} else { #普通查询、单一条件内嵌文档
if (strpos($field, "[") !== false)
$field = str_replace("[", ".", $field);
if (is_array($val) && $type) {
$arr = $this->GetWhere($val);
foreach ($arr as $k => $v) {
$wheres[$k][] = $v;
}
} elseif (is_null($val)) {
$wheres[$field] = array('$in' => array(null), '$exists' => 1);
} else {
$wheres[$field] = $val; #支持 array('age'=>array('>'=>'18','<='=>'20'))
}
}
}
}
return $wheres;
}
/**
* 插入一行
* @param $collect
* @param $value
* @return bool
*/
public function InsOne($collect, $value)
{
if (!$this->collected || !is_array($value))
return false;
$id = array_search('id', array_keys($value)); #处理有id字段的情况
if ($id !== false && array_search('_id', array_keys($value)) === false) {
$value['_id'] = $value['id'];
unset($value['id']);
}
$value = $this->ConvertEncode($value);
if (!$value)
return false;
try {
$result = $this->GetMonCollection($collect)->insert($value);
$result = $result['err'] ? false : true;
} catch (MongoException $e) {
$this->ErrLog($e->getMessage());
return false;
}
return $result;
}
/**
* 插入多行
* @param $collect 集合
* @param $fields
* @param $values 键值对:array(0=>array(''=>''))
* @param bool $continueOnError 是否忽略插入错误,默认忽略
* @return bool
*/
public function InsMulit($collect, $fields, $values, $continueOnError = false)
{
if (!is_array($fields) || !is_array($values) || !$this->collected)
return false;
$id = array_search('id', $fields); #处理有id字段的情况
if ($id !== false && array_search('_id', $fields) === false)
$fields[$id] = '_id';
$data = array();
$values = $this->ConvertEncode($values);
if ($values) {
foreach ($values as $v) {
if (is_array($v)) {
$v = array_combine($fields, $v);
if ($v)
$data[] = $v;
}
}
}
if (!$data)
return false;
$option['continueOnError'] = $continueOnError ? true : false;
try {
$result = $this->GetMonCollection($collect)->batchinsert($data, $option);
$result = $result['err'] ? false : true;
} catch (MongoException $e) {
$this->ErrLog($e->getMessage());
return false;
}
return $result;
}
/**
* 查询一个
* @param $collect
* @param array $where
* @param $field string
* @return bool
*/
function FindOne($collect, $where = array(), $field)
{
if (!$this->collected)
return false;
$wheres = $this->GetWhere($where);
if ($where && !$wheres)
return false;
if (is_string($field) && trim($field))
$field = trim($field);
else
return false;
$result = $this->GetMonCollection($collect)->findOne($wheres, array($field));
if ($result) {
$arr = get_object_vars($result['_id']);
$result['_id'] = $arr['$id'];
return $result[$field];
}
return false;
}
/**
* 查询一条
* @param $collect
* @param array|string $where
* @param array $fields
* @param string $type
* @return array|bool
*/
function FindRow($collect, $where = array(), $fields = array(), $type = 'assoc')
{
if (!$this->collected)
return false;
$wheres = $this->GetWhere($where);
if ($where && !$wheres)
return false;
if ($fields && is_array($fields)) { #过滤fields
$val = '';
for ($i = 0; $i < count($fields); $i++) {
$val[] = 1;
}
$fields = array_combine($fields, $val);
} else {
$fields = array();
}
$result = $this->GetMonCollection($collect)->findOne($wheres, $fields);
if ($result) {
if (in_array('_id', $fields)) {
$arr = get_object_vars($result['_id']);
$result['_id'] = $arr['$id'];
} else {
unset($result['_id']);
}
if (strtolower($type) == 'array')
$result = array_values($result);
return $result;
}
return false;
}
/**
* 复合查询 todo 查询高级选项扩展 snapshot maxscan min max hint explain
* <-----------------------
* $gt:>|$gte:>=|$le:<|$lte:<=|$ne:!=
* array('x'=>array('$gt'=>'20'))
*
* $or:$|$end:&
* array('x'=>array('$in'=>array('1','2')))
*
* $where:%
* array('x'=>array('$where'=>function))
*
* 正则表达式：
* ?
*
* 数组：
* $all:*
*
* 内嵌文档：
* 单个条件：username.user username[user
* 多个条件：$elemMatch
*
* 逻辑操作符：
* not:!
* or:$
* end:&
* sort
* array('x'=>0)
*
* limit|skip
* $where = array('&|'=>array('she_hash' => '48b6c531ef2469469ede4ac21eebcf51', 'type' => 'doing'));
* $field = array('time', '_id' => 0);
* $res = $mongo->FindMix('sapilog', $where, $field,'1,2',array('time'=>'desc'));
* ------------------------->
* @param $collect |string
* @param string $where |array array('>|x'=>'10','<=|x'=>'20','^|x'=>array(),'%|x'=>'fun')
* @param array|string $fields |array|string 返回字段 默认为全部 字符串返回一个字段，键为数字则为返回该字段，键为字段，值为-1为不返回该字段
* @param string $limit |string skip,limit 0,10 如果没有 默认 skip为0 limit为当前变量
* @param string $sort |array array('x'=>'desc|asc') desc=>-1 asc=>
* @param string $type | string 返回类型 默认为关联数组 assoc|array
* @return bool
*/
public function FindMix($collect, $where = '', $fields = array(), $limit = '', $sort = '', $type = 'assoc')
{
if (!$this->collected)
return false;
$wheres = $this->GetWhere($where);
if ($where && !$wheres)
return false;
if ($fields && is_array($fields)) { #过滤fields
$val = '';
for ($i = 0; $i < count($fields); $i++) {
$val[] = 1;
}
$fields = array_combine($fields, $val);
} else {
$fields = array();
}
$limits = '';
$skip = '';
if (is_string($limit)) { #过滤limit
$limit = explode(",", trim($limit));
if ($limit && is_numeric(implode('', $limit))) {
if (count($limit) == 1) {
$limits = $limit[0];
} else {
$skip = $limit[0];
$limits = $limit[1];
}
}
} elseif (is_numeric($limit)) {
$limits = trim($limit);
}
$sorts = '';
if (is_array($sort) && $sort) { #过滤sort
foreach ($sort as $k => $v) {
$k = trim($k);
$v = strtoupper(trim($v));
if (in_array($v, array_keys($this->sort_map))) {
$sorts[$k] = $this->sort_map[$v];
}
}
}
$result = $this->GetMonCollection($collect)->find($wheres, $fields);
if ($skip)
$result = $result->skip($skip);
if ($limits)
$result = $result->limit($limits);
if ($sorts)
$result = $result->sort($sorts);
if ($result) {
$return = array();
foreach ($result as $v) {
$return[] = $v;
}
foreach ($return as &$v) {
if (in_array('_id', $fields)) {
$arr = get_object_vars($v['_id']);
$v['_id'] = $arr['$id'];
} else {
unset($v['_id']);
}
if (strtolower($type) == 'array')
$v = array_values($v);
}
return $return;
}
return false;
}
/**
* 修改记录
* $inc、$set、$unset 修改普通文档，如果有就修改，没有就条件加值创建，前者只支持数字类型，使用后者,$unset删除元素
* $push、$addToSet 添加数组元素，前者不去重，使用后者
* $pop、$pull 删除数组元素，前者删除前后，后者删除指定元素，使用后者
*
* @param $collect
* @param $where
* @param $newdata |array
* @param string $type 操作类型 upd修改 unset删除指定元素 arr修改数组元素 pull删除数组元素
* @param bool $upsert 是否创建
* @param int $return |1：返回bool,2:返回影响行数
* @return bool
*/
public function UpdMix($collect, $where, $newdata, $type = 'upd', $upsert = false, $return = 1)
{
if (!$this->collected || !is_array($newdata))
return false;
$wheres = $this->GetWhere($where);
if (!$wheres) {
$this->ErrLog('where is error');
return false;
}
$newdata = $this->ConvertEncode($newdata);
$type = strtolower(trim($type));
$types = array('upd' => '$set', 'unset' => '$unset', 'arr' => '$addToSet', 'pull' => '$pull');
if (isset($types[$type])) {
$option = array();
if ($type == 'upd')
$option = array('multiple' => true);
if (in_array($type, array('upd', 'arr')) && $upsert)
$option['upsert'] = true;
$newdata = array($types[$type] => $newdata);
} else {
return false;
}
try {
$result = $this->GetMonCollection($collect)->update($wheres, $newdata, $option);
$result = $return == 1 ? ($result['err'] ? false : true) : $result['n'];
} catch (MongoConnectionException $e) {
$this->ErrLog($e->getMessage());
return false;
}
return $result;
}
#todo EXPLAIN 函数
/**
* 对集合执行命令
* @param $data
* @return bool
*/
public function RunCommand($data)
{
if (is_array($data) && $data && $this->collected) {
$result = $this->db->command($data);
if (isset($result['values']))
return $result['values'];
else
$this->ErrLog("command error,info:" . $result['errmsg'] . ",bad_cmd:" . json_encode($result['bad cmd']));
}
return false;
}
/**
* todo 聚合
*
*/
public function Group()
{
}
public function Count($collect, $where = array())
{
if (!$this->collected)
return false;
$wheres = $this->GetWhere($where);
if ($where && !$wheres)
return false;
return $this->GetMonCollection($collect)->count($wheres);
}
/**
* 返回指定键的唯一值列表
* @param $collect
* @param $key
* @param array $where 额外查询条件
* @return bool
*/
public function Distinct($collect, $key, $where = array())
{
if (!$this->collected)
return false;
$wheres = $this->GetWhere($where);
if ($where && !$wheres)
return false;
try {
$result = $this->GetMonCollection($collect)->distinct($key, $wheres);
} catch (MongoException $e) {
$this->ErrLog($e->getMessage());
return false;
}
return $result;
}
/**
* 删除集合中的记录
* @param $collect
* @param $where
* @param int $return 1：返回bool,2:返回影响行数
* @return bool
*/
public function RemoveColl($collect, $where, $return = 1)
{
if (!$this->collected)
return false;
$wheres = $this->GetWhere($where);
if (!$wheres) {
$this->ErrLog('where is error');
return false;
}
try {
$result = $this->GetMonCollection($collect)->remove($wheres);
$result = $return == 1 ? ($result['err'] ? false : true) : $result['n'];
} catch (MongoConnectionException $e) {
$this->ErrLog($e->getMessage());
return false;
}
return $result;
}
/**
* mongo 错误日志函数
*
* @param $msg
* @param int $lever
*
*/
private function ErrLog($msg, $lever = 2)
{
global $__CFG;
$trace = debug_backtrace();
$error_log = '';
$date = date("Ymd", time());
$line = isset($trace[$lever]['line']) ? trim($trace[$lever]['line']) : '';
$file = isset($trace[$lever]['file']) ? trim($trace[$lever]['file']) : '';
$object = isset($trace[$lever]['object']) ? get_class($trace[$lever]['object']) : '';
$args = isset($trace[$lever]['args']) ? json_encode($trace[$lever]['args']) : '';
$error_log .= "line {$line} " . ($object ? 'of ' . $object : '') . "(in {$file})\n";
$error_log .= $args ? "args:{$args}\n" : '';
$error_log .= "msg:{$msg}";
if (isset($__CFG['com']['id']) && $__CFG['com']['id']) {
$com_id = $__CFG['com']['id'];
} else {
$com_id = 'common';
}
$log_dir = $this->config['logpath'] ? $this->config['logpath'] : __ROOT__ . "data/nginx_error_log/{$com_id}/{$date}/mongo.nginx_error_log";
error_log("Date:" . date("Y-m-d H:i:s", $date) . "\nstamp:{$date}\ntrace:{$error_log}\n\n", 3, $log_dir);
if (isset($__CFG['currUser']) && $__CFG['currUser'] == 'root' && isset($__CFG['daemon']['user'])) {
$paths = explode("nginx_error_log/", $log_dir);
$pathc = explode("/", $paths[1]);
$pathd = $paths[0] . "nginx_error_log";
foreach ($pathc as $v) {
$pathd .= "/" . $v;
chgrp($pathd, $__CFG['daemon']['user']);
chown($pathd, $__CFG['daemon']['user']);
}
}
}
}
