123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891 |
- <?php
- // +----------------------------------------------------------------------
- // | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
- // +----------------------------------------------------------------------
- // | Copyright (c) 2006-2014 http://thinkphp.cn All rights reserved.
- // +----------------------------------------------------------------------
- // | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
- // +----------------------------------------------------------------------
- // | Author: liu21st <liu21st@gmail.com>
- // +----------------------------------------------------------------------
- namespace Think\Db\Driver;
- use Think\Db\Driver;
- /**
- * Mongo数据库驱动
- */
- class Mongo extends Driver
- {
- protected $_mongo = null; // MongoDb Object
- protected $_collection = null; // MongoCollection Object
- protected $_dbName = ''; // dbName
- protected $_collectionName = ''; // collectionName
- protected $_cursor = null; // MongoCursor Object
- protected $comparison = array('neq' => 'ne', 'ne' => 'ne', 'gt' => 'gt', 'egt' => 'gte', 'gte' => 'gte', 'lt' => 'lt', 'elt' => 'lte', 'lte' => 'lte', 'in' => 'in', 'not in' => 'nin', 'nin' => 'nin');
- /**
- * 架构函数 读取数据库配置信息
- * @access public
- * @param array $config 数据库配置数组
- */
- public function __construct($config = '')
- {
- if (!class_exists('mongoClient')) {
- E(L('_NOT_SUPPORT_') . ':Mongo');
- }
- if (!empty($config)) {
- $this->config = array_merge($this->config, $config);
- if (empty($this->config['params'])) {
- $this->config['params'] = array();
- }
- }
- }
- /**
- * 连接数据库方法
- * @access public
- */
- public function connect($config = '', $linkNum = 0)
- {
- if (!isset($this->linkID[$linkNum])) {
- if (empty($config)) {
- $config = $this->config;
- }
- $host = 'mongodb://' . ($config['username'] ? "{$config['username']}" : '') . ($config['password'] ? ":{$config['password']}@" : '') . $config['hostname'] . ($config['hostport'] ? ":{$config['hostport']}" : '') . '/' . ($config['database'] ? "{$config['database']}" : '');
- try {
- $this->linkID[$linkNum] = new \mongoClient($host, $this->config['params']);
- } catch (\MongoConnectionException $e) {
- E($e->getmessage());
- }
- }
- return $this->linkID[$linkNum];
- }
- /**
- * 切换当前操作的Db和Collection
- * @access public
- * @param string $collection collection
- * @param string $db db
- * @param boolean $master 是否主服务器
- * @return void
- */
- public function switchCollection($collection, $db = '', $master = true)
- {
- // 当前没有连接 则首先进行数据库连接
- if (!$this->_linkID) {
- $this->initConnect($master);
- }
- try {
- if (!empty($db)) {
- // 传人Db则切换数据库
- // 当前MongoDb对象
- $this->_dbName = $db;
- $this->_mongo = $this->_linkID->selectDb($db);
- }
- // 当前MongoCollection对象
- if ($this->config['debug']) {
- $this->queryStr = $this->_dbName . '.getCollection(' . $collection . ')';
- }
- if ($this->_collectionName != $collection) {
- $this->queryTimes++;
- N('db_query', 1); // 兼容代码
- $this->debug(true);
- $this->_collection = $this->_mongo->selectCollection($collection);
- $this->debug(false);
- $this->_collectionName = $collection; // 记录当前Collection名称
- }
- } catch (MongoException $e) {
- E($e->getMessage());
- }
- }
- /**
- * 释放查询结果
- * @access public
- */
- public function free()
- {
- $this->_cursor = null;
- }
- /**
- * 执行命令
- * @access public
- * @param array $command 指令
- * @return array
- */
- public function command($command = array(), $options = array())
- {
- $cache = isset($options['cache']) ? $options['cache'] : false;
- if ($cache) {
- // 查询缓存检测
- $key = is_string($cache['key']) ? $cache['key'] : md5(serialize($command));
- $value = S($key, '', '', $cache['type']);
- if (false !== $value) {
- return $value;
- }
- }
- N('db_write', 1); // 兼容代码
- $this->executeTimes++;
- try {
- if ($this->config['debug']) {
- $this->queryStr = $this->_dbName . '.' . $this->_collectionName . '.runCommand(';
- $this->queryStr .= json_encode($command);
- $this->queryStr .= ')';
- }
- $this->debug(true);
- $result = $this->_mongo->command($command);
- $this->debug(false);
- if ($cache && $result['ok']) {
- // 查询缓存写入
- S($key, $result, $cache['expire'], $cache['type']);
- }
- return $result;
- } catch (\MongoCursorException $e) {
- E($e->getMessage());
- }
- }
- /**
- * 执行语句
- * @access public
- * @param string $code sql指令
- * @param array $args 参数
- * @return mixed
- */
- public function execute($code, $args = array())
- {
- $this->executeTimes++;
- N('db_write', 1); // 兼容代码
- $this->debug(true);
- $this->queryStr = 'execute:' . $code;
- $result = $this->_mongo->execute($code, $args);
- $this->debug(false);
- if ($result['ok']) {
- return $result['retval'];
- } else {
- E($result['errmsg']);
- }
- }
- /**
- * 关闭数据库
- * @access public
- */
- public function close()
- {
- if ($this->_linkID) {
- $this->_linkID->close();
- $this->_linkID = null;
- $this->_mongo = null;
- $this->_collection = null;
- $this->_cursor = null;
- }
- }
- /**
- * 数据库错误信息
- * @access public
- * @return string
- */
- public function error()
- {
- $this->error = $this->_mongo->lastError();
- trace($this->error, '', 'ERR');
- return $this->error;
- }
- /**
- * 插入记录
- * @access public
- * @param mixed $data 数据
- * @param array $options 参数表达式
- * @param boolean $replace 是否replace
- * @return false | integer
- */
- public function insert($data, $options = array(), $replace = false)
- {
- if (isset($options['table'])) {
- $this->switchCollection($options['table']);
- }
- $this->model = $options['model'];
- $this->executeTimes++;
- N('db_write', 1); // 兼容代码
- if ($this->config['debug']) {
- $this->queryStr = $this->_dbName . '.' . $this->_collectionName . '.insert(';
- $this->queryStr .= $data ? json_encode($data) : '{}';
- $this->queryStr .= ')';
- }
- try {
- $this->debug(true);
- $result = $replace ? $this->_collection->save($data) : $this->_collection->insert($data);
- $this->debug(false);
- if ($result) {
- $_id = $data['_id'];
- if (is_object($_id)) {
- $_id = $_id->__toString();
- }
- $this->lastInsID = $_id;
- }
- return $result;
- } catch (\MongoCursorException $e) {
- E($e->getMessage());
- }
- }
- /**
- * 插入多条记录
- * @access public
- * @param array $dataList 数据
- * @param array $options 参数表达式
- * @return bool
- */
- public function insertAll($dataList, $options = array())
- {
- if (isset($options['table'])) {
- $this->switchCollection($options['table']);
- }
- $this->model = $options['model'];
- $this->executeTimes++;
- N('db_write', 1); // 兼容代码
- try {
- $this->debug(true);
- $result = $this->_collection->batchInsert($dataList);
- $this->debug(false);
- return $result;
- } catch (\MongoCursorException $e) {
- E($e->getMessage());
- }
- }
- /**
- * 生成下一条记录ID 用于自增非MongoId主键
- * @access public
- * @param string $pk 主键名
- * @return integer
- */
- public function getMongoNextId($pk,$options=array())
- {
- if (isset($options['table'])) {
- $this->switchCollection($options['table']);
- }
- if ($this->config['debug']) {
- $this->queryStr = $this->_dbName . '.' . $this->_collectionName . '.find({},{' . $pk . ':1}).sort({' . $pk . ':-1}).limit(1)';
- }
- try {
- $this->debug(true);
- $result = $this->_collection->find(array(), array($pk => 1))->sort(array($pk => -1))->limit(1);
- $this->debug(false);
- } catch (\MongoCursorException $e) {
- E($e->getMessage());
- }
- $data = $result->getNext();
- return isset($data[$pk]) ? $data[$pk] + 1 : 1;
- }
- /**
- * 更新记录
- * @access public
- * @param mixed $data 数据
- * @param array $options 表达式
- * @return bool
- */
- public function update($data, $options)
- {
- if (isset($options['table'])) {
- $this->switchCollection($options['table']);
- }
- $this->executeTimes++;
- N('db_write', 1); // 兼容代码
- $this->model = $options['model'];
- $query = $this->parseWhere(isset($options['where']) ? $options['where'] : array());
- $set = $this->parseSet($data);
- if ($this->config['debug']) {
- $this->queryStr = $this->_dbName . '.' . $this->_collectionName . '.update(';
- $this->queryStr .= $query ? json_encode($query) : '{}';
- $this->queryStr .= ',' . json_encode($set) . ')';
- }
- try {
- $this->debug(true);
- if (isset($options['limit']) && 1 == $options['limit']) {
- $multiple = array("multiple" => false);
- } else {
- $multiple = array("multiple" => true);
- }
- $result = $this->_collection->update($query, $set, $multiple);
- $this->debug(false);
- return $result;
- } catch (\MongoCursorException $e) {
- E($e->getMessage());
- }
- }
- /**
- * 删除记录
- * @access public
- * @param array $options 表达式
- * @return false | integer
- */
- public function delete($options = array())
- {
- if (isset($options['table'])) {
- $this->switchCollection($options['table']);
- }
- $query = $this->parseWhere(isset($options['where']) ? $options['where'] : array());
- $this->model = $options['model'];
- $this->executeTimes++;
- N('db_write', 1); // 兼容代码
- if ($this->config['debug']) {
- $this->queryStr = $this->_dbName . '.' . $this->_collectionName . '.remove(' . json_encode($query) . ')';
- }
- try {
- $this->debug(true);
- $result = $this->_collection->remove($query);
- $this->debug(false);
- return $result;
- } catch (\MongoCursorException $e) {
- E($e->getMessage());
- }
- }
- /**
- * 清空记录
- * @access public
- * @param array $options 表达式
- * @return false | integer
- */
- public function clear($options = array())
- {
- if (isset($options['table'])) {
- $this->switchCollection($options['table']);
- }
- $this->model = $options['model'];
- $this->executeTimes++;
- N('db_write', 1); // 兼容代码
- if ($this->config['debug']) {
- $this->queryStr = $this->_dbName . '.' . $this->_collectionName . '.remove({})';
- }
- try {
- $this->debug(true);
- $result = $this->_collection->drop();
- $this->debug(false);
- return $result;
- } catch (\MongoCursorException $e) {
- E($e->getMessage());
- }
- }
- /**
- * 查找记录
- * @access public
- * @param array $options 表达式
- * @return iterator
- */
- public function select($options = array())
- {
- if (isset($options['table'])) {
- $this->switchCollection($options['table'], '', false);
- }
- $this->model = $options['model'];
- $this->queryTimes++;
- N('db_query', 1); // 兼容代码
- $query = $this->parseWhere(isset($options['where']) ? $options['where'] : array());
- $field = $this->parseField(isset($options['field']) ? $options['field'] : array());
- try {
- if ($this->config['debug']) {
- $this->queryStr = $this->_dbName . '.' . $this->_collectionName . '.find(';
- $this->queryStr .= $query ? json_encode($query) : '{}';
- if (is_array($field) && count($field)) {
- foreach ($field as $f => $v) {
- $_field_array[$f] = $v ? 1 : 0;
- }
- $this->queryStr .= $field ? ', ' . json_encode($_field_array) : ', {}';
- }
- $this->queryStr .= ')';
- }
- $this->debug(true);
- $_cursor = $this->_collection->find($query, $field);
- if (!empty($options['order'])) {
- $order = $this->parseOrder($options['order']);
- if ($this->config['debug']) {
- $this->queryStr .= '.sort(' . json_encode($order) . ')';
- }
- $_cursor = $_cursor->sort($order);
- }
- if (isset($options['page'])) {
- // 根据页数计算limit
- list($page, $length) = $options['page'];
- $page = $page > 0 ? $page : 1;
- $length = $length > 0 ? $length : (is_numeric($options['limit']) ? $options['limit'] : 20);
- $offset = $length * ((int) $page - 1);
- $options['limit'] = $offset . ',' . $length;
- }
- if (isset($options['limit'])) {
- list($offset, $length) = $this->parseLimit($options['limit']);
- if (!empty($offset)) {
- if ($this->config['debug']) {
- $this->queryStr .= '.skip(' . intval($offset) . ')';
- }
- $_cursor = $_cursor->skip(intval($offset));
- }
- if ($this->config['debug']) {
- $this->queryStr .= '.limit(' . intval($length) . ')';
- }
- $_cursor = $_cursor->limit(intval($length));
- }
- $this->debug(false);
- $this->_cursor = $_cursor;
- $resultSet = iterator_to_array($_cursor);
- return $resultSet;
- } catch (\MongoCursorException $e) {
- E($e->getMessage());
- }
- }
- /**
- * 查找某个记录
- * @access public
- * @param array $options 表达式
- * @return array
- */
- public function find($options = array())
- {
- $options['limit'] = 1;
- $find = $this->select($options);
- return array_shift($find);
- }
- /**
- * 统计记录数
- * @access public
- * @param array $options 表达式
- * @return iterator
- */
- public function count($options = array())
- {
- if (isset($options['table'])) {
- $this->switchCollection($options['table'], '', false);
- }
- $this->model = $options['model'];
- $this->queryTimes++;
- N('db_query', 1); // 兼容代码
- $query = $this->parseWhere(isset($options['where']) ? $options['where'] : array());
- if ($this->config['debug']) {
- $this->queryStr = $this->_dbName . '.' . $this->_collectionName;
- $this->queryStr .= $query ? '.find(' . json_encode($query) . ')' : '';
- $this->queryStr .= '.count()';
- }
- try {
- $this->debug(true);
- $count = $this->_collection->count($query);
- $this->debug(false);
- return $count;
- } catch (\MongoCursorException $e) {
- E($e->getMessage());
- }
- }
- public function group($keys, $initial, $reduce, $options = array())
- {
- if (isset($options['table']) && $this->_collectionName != $options['table']) {
- $this->switchCollection($options['table'], '', false);
- }
- $cache = isset($options['cache']) ? $options['cache'] : false;
- if ($cache) {
- $key = is_string($cache['key']) ? $cache['key'] : md5(serialize($options));
- $value = S($key, '', '', $cache['type']);
- if (false !== $value) {
- return $value;
- }
- }
- $this->model = $options['model'];
- $this->queryTimes++;
- N('db_query', 1); // 兼容代码
- $query = $this->parseWhere(isset($options['where']) ? $options['where'] : array());
- if ($this->config['debug']) {
- $this->queryStr = $this->_dbName . '.' . $this->_collectionName . '.group({key:' . json_encode($keys) . ',cond:' .
- json_encode($options['condition']) . ',reduce:' .
- json_encode($reduce) . ',initial:' .
- json_encode($initial) . '})';
- }
- try {
- $this->debug(true);
-
- $option = array();
- isset($options['condition'])&&$option['condition']=$options['condition'];
- isset($options['finalize'])&&$option['finalize']=$options['condition'];
- isset($options['maxTimeMS'])&&$option['maxTimeMS']=$options['condition'];
- $group = $this->_collection->group($keys,$initial,$reduce,$option);
- $this->debug(false);
- if ($cache && $group['ok']) {
- S($key, $group, $cache['expire'], $cache['type']);
- }
- return $group;
- } catch (\MongoCursorException $e) {
- E($e->getMessage());
- }
- }
- /**
- * 取得数据表的字段信息
- * @access public
- * @return array
- */
- public function getFields($collection = '')
- {
- if (!empty($collection) && $collection != $this->_collectionName) {
- $this->switchCollection($collection, '', false);
- }
- $this->queryTimes++;
- N('db_query', 1); // 兼容代码
- if ($this->config['debug']) {
- $this->queryStr = $this->_dbName . '.' . $this->_collectionName . '.findOne()';
- }
- try {
- $this->debug(true);
- $result = $this->_collection->findOne();
- $this->debug(false);
- } catch (\MongoCursorException $e) {
- E($e->getMessage());
- }
- if ($result) {
- // 存在数据则分析字段
- $info = array();
- foreach ($result as $key => $val) {
- $info[$key] = array(
- 'name' => $key,
- 'type' => getType($val),
- );
- }
- return $info;
- }
- // 暂时没有数据 返回false
- return false;
- }
- /**
- * 取得当前数据库的collection信息
- * @access public
- */
- public function getTables()
- {
- if ($this->config['debug']) {
- $this->queryStr = $this->_dbName . '.getCollenctionNames()';
- }
- $this->queryTimes++;
- N('db_query', 1); // 兼容代码
- $this->debug(true);
- $list = $this->_mongo->listCollections();
- $this->debug(false);
- $info = array();
- foreach ($list as $collection) {
- $info[] = $collection->getName();
- }
- return $info;
- }
- /**
- * 取得当前数据库的对象
- * @access public
- * @return object mongoClient
- */
- public function getDB()
- {
- return $this->_mongo;
- }
- /**
- * 取得当前集合的对象
- * @access public
- * @return object MongoCollection
- */
- public function getCollection()
- {
- return $this->_collection;
- }
- /**
- * set分析
- * @access protected
- * @param array $data
- * @return string
- */
- protected function parseSet($data)
- {
- $result = array();
- foreach ($data as $key => $val) {
- if (is_array($val)) {
- switch ($val[0]) {
- case 'inc':
- $result['$inc'][$key] = (float) $val[1];
- break;
- case 'set':
- case 'unset':
- case 'push':
- case 'pushall':
- case 'addtoset':
- case 'pop':
- case 'pull':
- case 'pullall':
- $result['$' . $val[0]][$key] = $val[1];
- break;
- default:
- $result['$set'][$key] = $val;
- }
- } else {
- $result['$set'][$key] = $val;
- }
- }
- return $result;
- }
- /**
- * order分析
- * @access protected
- * @param mixed $order
- * @return array
- */
- protected function parseOrder($order)
- {
- if (is_string($order)) {
- $array = explode(',', $order);
- $order = array();
- foreach ($array as $key => $val) {
- $arr = explode(' ', trim($val));
- if (isset($arr[1])) {
- $arr[1] = 'asc' == $arr[1] ? 1 : -1;
- } else {
- $arr[1] = 1;
- }
- $order[$arr[0]] = $arr[1];
- }
- }
- return $order;
- }
- /**
- * limit分析
- * @access protected
- * @param mixed $limit
- * @return array
- */
- protected function parseLimit($limit)
- {
- if (strpos($limit, ',')) {
- $array = explode(',', $limit);
- } else {
- $array = array(0, $limit);
- }
- return $array;
- }
- /**
- * field分析
- * @access protected
- * @param mixed $fields
- * @return array
- */
- public function parseField($fields)
- {
- if (empty($fields)) {
- $fields = array();
- }
- if (is_string($fields)) {
- $_fields = explode(',', $fields);
- $fields = array();
- foreach ($_fields as $f) {
- $fields[$f] = true;
- }
- } elseif (is_array($fields)) {
- $_fields = $fields;
- $fields = array();
- foreach ($_fields as $f => $v) {
- if (is_numeric($f)) {
- $fields[$v] = true;
- } else {
- $fields[$f] = $v ? true : false;
- }
- }
- }
- return $fields;
- }
- /**
- * where分析
- * @access protected
- * @param mixed $where
- * @return array
- */
- public function parseWhere($where)
- {
- $query = array();
- $return = array();
- $_logic = '$and';
- if (isset($where['_logic'])) {
- $where['_logic'] = strtolower($where['_logic']);
- $_logic = in_array($where['_logic'], array('or', 'xor', 'nor', 'and')) ? '$' . $where['_logic'] : $_logic;
- unset($where['_logic']);
- }
- foreach ($where as $key => $val) {
- if ('_id' != $key && 0 === strpos($key, '_')) {
- // 解析特殊条件表达式
- $parse = $this->parseThinkWhere($key, $val);
- $query = array_merge($query, $parse);
- } else {
- // 查询字段的安全过滤
- if (!preg_match('/^[A-Z_\|\&\-.a-z0-9]+$/', trim($key))) {
- E(L('_ERROR_QUERY_') . ':' . $key);
- }
- $key = trim($key);
- if (strpos($key, '|')) {
- $array = explode('|', $key);
- $str = array();
- foreach ($array as $k) {
- $str[] = $this->parseWhereItem($k, $val);
- }
- $query['$or'] = $str;
- } elseif (strpos($key, '&')) {
- $array = explode('&', $key);
- $str = array();
- foreach ($array as $k) {
- $str[] = $this->parseWhereItem($k, $val);
- }
- $query = array_merge($query, $str);
- } else {
- $str = $this->parseWhereItem($key, $val);
- $query = array_merge($query, $str);
- }
- }
- }
- if ('$and' == $_logic) {
- return $query;
- }
- foreach ($query as $key => $val) {
- $return[$_logic][] = array($key => $val);
- }
- return $return;
- }
- /**
- * 特殊条件分析
- * @access protected
- * @param string $key
- * @param mixed $val
- * @return string
- */
- protected function parseThinkWhere($key, $val)
- {
- $query = array();
- $_logic = array('or', 'xor', 'nor', 'and');
- switch ($key) {
- case '_query': // 字符串模式查询条件
- parse_str($val, $query);
- if (isset($query['_logic']) && strtolower($query['_logic']) == 'or') {
- unset($query['_logic']);
- $query['$or'] = $query;
- }
- break;
- case '_complex': // 子查询模式查询条件
- $__logic = strtolower($val['_logic']);
- if (isset($val['_logic']) && in_array($__logic, $_logic)) {
- unset($val['_logic']);
- $query['$' . $__logic] = $val;
- }
- break;
- case '_string': // MongoCode查询
- $query['$where'] = new \MongoCode($val);
- break;
- }
- //兼容 MongoClient OR条件查询方法
- if (isset($query['$or']) && !is_array(current($query['$or']))) {
- $val = array();
- foreach ($query['$or'] as $k => $v) {
- $val[] = array($k => $v);
- }
- $query['$or'] = $val;
- }
- return $query;
- }
- /**
- * where子单元分析
- * @access protected
- * @param string $key
- * @param mixed $val
- * @return array
- */
- protected function parseWhereItem($key, $val)
- {
- $query = array();
- if (is_array($val)) {
- if (is_string($val[0])) {
- $con = strtolower($val[0]);
- if (in_array($con, array('neq', 'ne', 'gt', 'egt', 'gte', 'lt', 'lte', 'elt'))) {
- // 比较运算
- $k = '$' . $this->comparison[$con];
- $query[$key] = array($k => $val[1]);
- } elseif ('like' == $con) {
- // 模糊查询 采用正则方式
- $query[$key] = new \MongoRegex("/" . $val[1] . "/");
- } elseif ('mod' == $con) {
- // mod 查询
- $query[$key] = array('$mod' => $val[1]);
- } elseif ('regex' == $con) {
- // 正则查询
- $query[$key] = new \MongoRegex($val[1]);
- } elseif (in_array($con, array('in', 'nin', 'not in'))) {
- // IN NIN 运算
- $data = is_string($val[1]) ? explode(',', $val[1]) : $val[1];
- $k = '$' . $this->comparison[$con];
- $query[$key] = array($k => $data);
- } elseif ('all' == $con) {
- // 满足所有指定条件
- $data = is_string($val[1]) ? explode(',', $val[1]) : $val[1];
- $query[$key] = array('$all' => $data);
- } elseif ('between' == $con) {
- // BETWEEN运算
- $data = is_string($val[1]) ? explode(',', $val[1]) : $val[1];
- $query[$key] = array('$gte' => $data[0], '$lte' => $data[1]);
- } elseif ('not between' == $con) {
- $data = is_string($val[1]) ? explode(',', $val[1]) : $val[1];
- $query[$key] = array('$lt' => $data[0], '$gt' => $data[1]);
- } elseif ('exp' == $con) {
- // 表达式查询
- $query['$where'] = new \MongoCode($val[1]);
- } elseif ('exists' == $con) {
- // 字段是否存在
- $query[$key] = array('$exists' => (bool) $val[1]);
- } elseif ('size' == $con) {
- // 限制属性大小
- $query[$key] = array('$size' => intval($val[1]));
- } elseif ('type' == $con) {
- // 限制字段类型 1 浮点型 2 字符型 3 对象或者MongoDBRef 5 MongoBinData 7 MongoId 8 布尔型 9 MongoDate 10 NULL 15 MongoCode 16 32位整型 17 MongoTimestamp 18 MongoInt64 如果是数组的话判断元素的类型
- $query[$key] = array('$type' => intval($val[1]));
- } else {
- $query[$key] = $val;
- }
- return $query;
- }
- }
- $query[$key] = $val;
- return $query;
- }
- }
|