mongo php类
<?php /** * mongo 类 * User: wujp <wujp@nalashop.com> * 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']); } } } }
附件:http://down.51cto.com/data/2364629