mongo.class.php,thinkphp/Mongo.class.php at master · top-think/thinkphp · GitHub

Elena.w

于 2021-03-26 18:07:59 发布

阅读量93

点赞数

文章标签： mongo.class.php

// +----------------------------------------------------------------------

// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]

// +----------------------------------------------------------------------

// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )

// +----------------------------------------------------------------------

// | Author: liu21st

// +----------------------------------------------------------------------

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;

}

Elena.w

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
mongo.class.php,thinkphp/Mongo.class.php at master · top-think/thinkphp · GitHub

// +----------------------------------------------------------------------// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]// +----------------------------------------------------------------------// | Cop...
复制链接

扫一扫