mirror of
https://gitee.com/fastadminnet/framework.git
synced 2026-07-07 07:12:47 +08:00
调整类的文件命名规范为psr-4
This commit is contained in:
thinkphp
823
library/think/db/driver/Mongo.php
Normal file
823
library/think/db/driver/Mongo.php
Normal file
@@ -0,0 +1,823 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2006~2016 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;
|
||||
use think\Exception;
|
||||
use think\Lang;
|
||||
use think\Log;
|
||||
|
||||
/**
|
||||
* 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 = ['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|string $config 数据库配置数组
|
||||
*
|
||||
* @throws Exception
|
||||
*/
|
||||
public function __construct($config = '')
|
||||
{
|
||||
if (!class_exists('mongoClient')) {
|
||||
throw new Exception(Lang::get('_NOT_SUPPERT_') . ':Mongo');
|
||||
}
|
||||
if (!empty($config)) {
|
||||
$this->config = array_merge($this->config, $config);
|
||||
if (empty($this->config['params'])) {
|
||||
$this->config['params'] = [];
|
||||
}
|
||||
}
|
||||
$this->config = $config;
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
*
|
||||
* @param string $config
|
||||
* @param int $linkNum
|
||||
*
|
||||
* @return
|
||||
* @throws Exception
|
||||
*/
|
||||
public function connect($config = '', $linkNum = 0, $autoConnection = false)
|
||||
{
|
||||
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, !empty($this->config['params'])?$this->config['params']:array());
|
||||
} catch (\MongoConnectionException $e) {
|
||||
throw new Exception($e->getmessage());
|
||||
}
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 切换当前操作的Db和Collection
|
||||
* @access public
|
||||
*
|
||||
* @param string $collection collection
|
||||
* @param string $db db
|
||||
* @param boolean $master 是否主服务器
|
||||
*
|
||||
* @throws Exception
|
||||
*/
|
||||
public function switchCollection($collection, $db = '', $master = true)
|
||||
{
|
||||
// 当前没有连接 则首先进行数据库连接
|
||||
if (!$this->_linkID) {
|
||||
$this->initConnect($master);
|
||||
}
|
||||
|
||||
$db = $db?$db:$this->config['database'];
|
||||
|
||||
try {
|
||||
if (!empty($db)) {
|
||||
// 传人Db则切换数据库
|
||||
// 当前MongoDb对象
|
||||
$this->_dbName = $db;
|
||||
$this->_mongo = $this->_linkID->selectDb($db);
|
||||
}
|
||||
// 当前MongoCollection对象
|
||||
if (!empty($this->config['debug'])) {
|
||||
$this->queryStr = $this->_dbName . '.getCollection(' . $collection . ')';
|
||||
}
|
||||
if ($this->_collectionName != $collection) {
|
||||
$this->queryTimes++;
|
||||
$this->debug(true);
|
||||
$this->_collection = $this->_mongo->selectCollection($collection);
|
||||
$this->debug(false);
|
||||
$this->_collectionName = $collection; // 记录当前Collection名称
|
||||
}
|
||||
} catch (\MongoException $e) {
|
||||
throw new Exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free()
|
||||
{
|
||||
$this->_cursor = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行命令
|
||||
* @access public
|
||||
* @param array $command 指令
|
||||
* @return array
|
||||
*/
|
||||
public function command($command = [])
|
||||
{
|
||||
$this->executeTimes++;
|
||||
$this->debug(true);
|
||||
$this->queryStr = 'command:' . json_encode($command);
|
||||
$result = $this->_mongo->command($command);
|
||||
$this->debug(false);
|
||||
if (!$result['ok']) {
|
||||
throw new Exception($result['errmsg']);
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
*
|
||||
* @param string $code sql指令
|
||||
* @param array $args 参数
|
||||
*
|
||||
* @return mixed
|
||||
* @throws Exception
|
||||
*/
|
||||
public function execute($code, $args = [])
|
||||
{
|
||||
$this->executeTimes++;
|
||||
$this->debug(true);
|
||||
$this->queryStr = 'execute:' . $code;
|
||||
$result = $this->_mongo->execute($code, $args);
|
||||
$this->debug(false);
|
||||
if ($result['ok']) {
|
||||
return $result['retval'];
|
||||
} else {
|
||||
throw new Exception($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();
|
||||
Log::record($this->error, 'error');
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* 插入记录
|
||||
* @access public
|
||||
*
|
||||
* @param mixed $data 数据
|
||||
* @param array $options 参数表达式
|
||||
* @param boolean $replace 是否replace
|
||||
*
|
||||
* @return false|int
|
||||
* @throws Exception
|
||||
*/
|
||||
public function insert($data, $options = [], $replace = false)
|
||||
{
|
||||
if (isset($options['table'])) {
|
||||
$this->switchCollection($options['table']);
|
||||
}
|
||||
$this->model = $options['model'];
|
||||
$this->executeTimes++;
|
||||
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) {
|
||||
throw new Exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 插入多条记录
|
||||
* @access public
|
||||
*
|
||||
* @param array $dataList 数据
|
||||
* @param array $options 参数表达式
|
||||
*
|
||||
* @return bool
|
||||
* @throws Exception
|
||||
*/
|
||||
public function insertAll($dataList, $options = [], $replace = false)
|
||||
{
|
||||
if (isset($options['table'])) {
|
||||
$this->switchCollection($options['table']);
|
||||
}
|
||||
$this->model = $options['model'];
|
||||
$this->executeTimes++;
|
||||
try {
|
||||
$this->debug(true);
|
||||
$result = $this->_collection->batchInsert($dataList);
|
||||
$this->debug(false);
|
||||
return $result;
|
||||
} catch (\MongoCursorException $e) {
|
||||
throw new Exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成下一条记录ID 用于自增非MongoId主键
|
||||
* @access public
|
||||
*
|
||||
* @param string $pk 主键名
|
||||
*
|
||||
* @return int
|
||||
* @throws Exception
|
||||
*/
|
||||
public function getMongoNextId($pk)
|
||||
{
|
||||
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([], [$pk => 1])->sort([$pk => -1])->limit(1);
|
||||
$this->debug(false);
|
||||
} catch (\MongoCursorException $e) {
|
||||
throw new Exception($e->getMessage());
|
||||
}
|
||||
$data = $result->getNext();
|
||||
return isset($data[$pk]) ? $data[$pk] + 1 : 1;
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新记录
|
||||
* @access public
|
||||
*
|
||||
* @param mixed $data 数据
|
||||
* @param array $options 表达式
|
||||
*
|
||||
* @return bool
|
||||
* @throws Exception
|
||||
*/
|
||||
public function update($data, $options)
|
||||
{
|
||||
if (isset($options['table'])) {
|
||||
$this->switchCollection($options['table']);
|
||||
}
|
||||
$this->executeTimes++;
|
||||
$this->model = $options['model'];
|
||||
$query = $this->parseWhere($options['where']);
|
||||
$set = $this->parseSet($data);
|
||||
if (!empty($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 = ["multiple" => false];
|
||||
} else {
|
||||
$multiple = ["multiple" => true];
|
||||
}
|
||||
$result = $this->_collection->update($query, $set, $multiple);
|
||||
$this->debug(false);
|
||||
return $result;
|
||||
} catch (\MongoCursorException $e) {
|
||||
throw new Exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return false | integer
|
||||
*/
|
||||
public function delete($options = [])
|
||||
{
|
||||
if (isset($options['table'])) {
|
||||
$this->switchCollection($options['table']);
|
||||
}
|
||||
$query = $this->parseWhere($options['where']);
|
||||
$this->model = $options['model'];
|
||||
$this->executeTimes++;
|
||||
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) {
|
||||
throw new Exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 清空记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return false | integer
|
||||
*/
|
||||
public function clear($options = [])
|
||||
{
|
||||
if (isset($options['table'])) {
|
||||
$this->switchCollection($options['table']);
|
||||
}
|
||||
$this->model = $options['model'];
|
||||
$this->executeTimes++;
|
||||
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) {
|
||||
throw new Exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 查找记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return iterator
|
||||
*/
|
||||
public function select($options = [])
|
||||
{
|
||||
if (isset($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++;
|
||||
$query = $this->parseWhere($options['where']);
|
||||
$field = $this->parseField($options['field']);
|
||||
try {
|
||||
if ($this->config['debug']) {
|
||||
$this->queryStr = $this->_dbName . '.' . $this->_collectionName . '.find(';
|
||||
$this->queryStr .= $query ? json_encode($query) : '{}';
|
||||
$this->queryStr .= $field ? ',' . json_encode($field) : '';
|
||||
$this->queryStr .= ')';
|
||||
}
|
||||
$this->debug(true);
|
||||
$_cursor = $this->_collection->find($query, $field);
|
||||
if ($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
|
||||
if (strpos($options['page'], ',')) {
|
||||
list($page, $length) = explode(',', $options['page']);
|
||||
} else {
|
||||
$page = $options['page'];
|
||||
}
|
||||
$page = $page ? $page : 1;
|
||||
$length = isset($length) ? $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);
|
||||
if ($cache && $resultSet) {
|
||||
// 查询缓存写入
|
||||
S($key, $resultSet, $cache['expire'], $cache['type']);
|
||||
}
|
||||
return $resultSet;
|
||||
} catch (\MongoCursorException $e) {
|
||||
throw new Exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 查找某个记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return array
|
||||
*/
|
||||
public function find($options = [])
|
||||
{
|
||||
if (isset($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++;
|
||||
$query = $this->parseWhere(!empty($options['where'])?$options['where']:'');
|
||||
$fields = $this->parseField(!empty($options['field'])?$options['field']:'');
|
||||
if (!empty($this->config['debug'])) {
|
||||
$this->queryStr = $this->_dbName . '.' . $this->_collectionName . '.findOne(';
|
||||
$this->queryStr .= $query ? json_encode($query) : '{}';
|
||||
$this->queryStr .= $fields ? ',' . json_encode($fields) : '';
|
||||
$this->queryStr .= ')';
|
||||
}
|
||||
try {
|
||||
$this->debug(true);
|
||||
$result = $this->_collection->findOne($query, $fields);
|
||||
$this->debug(false);
|
||||
if ($cache && $result) {
|
||||
// 查询缓存写入
|
||||
S($key, $result, $cache['expire'], $cache['type']);
|
||||
}
|
||||
return $result;
|
||||
} catch (\MongoCursorException $e) {
|
||||
throw new Exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 统计记录数
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return iterator
|
||||
*/
|
||||
public function count($options = [])
|
||||
{
|
||||
if (isset($options['table'])) {
|
||||
$this->switchCollection($options['table'], '', false);
|
||||
}
|
||||
$this->model = $options['model'];
|
||||
$this->queryTimes++;
|
||||
$query = $this->parseWhere($options['where']);
|
||||
if (!empty($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) {
|
||||
throw new Exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
public function group($keys, $initial, $reduce, $options = [])
|
||||
{
|
||||
$this->_collection->group($keys, $initial, $reduce, $options);
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
* @access public
|
||||
* @return array
|
||||
*/
|
||||
public function getFields($collection = '')
|
||||
{
|
||||
if (!empty($collection) && $collection != $this->_collectionName) {
|
||||
$this->switchCollection($collection, '', false);
|
||||
}
|
||||
$this->queryTimes++;
|
||||
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) {
|
||||
throw new Exception($e->getMessage());
|
||||
}
|
||||
if ($result) {
|
||||
// 存在数据则分析字段
|
||||
$info = [];
|
||||
foreach ($result as $key => $val) {
|
||||
$info[$key] = [
|
||||
'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++;
|
||||
$this->debug(true);
|
||||
$list = $this->_mongo->listCollections();
|
||||
$this->debug(false);
|
||||
$info = [];
|
||||
foreach ($list as $collection) {
|
||||
$info[] = $collection->getName();
|
||||
}
|
||||
return $info;
|
||||
}
|
||||
|
||||
/**
|
||||
* set分析
|
||||
* @access protected
|
||||
* @param array $data
|
||||
* @return string
|
||||
*/
|
||||
protected function parseSet($data)
|
||||
{
|
||||
$result = [];
|
||||
foreach ($data as $key => $val) {
|
||||
if (is_array($val)) {
|
||||
switch ($val[0]) {
|
||||
case 'inc':
|
||||
$result['$inc'][$key] = (int) $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_array($order)) {
|
||||
$order = join(',', $order);
|
||||
}
|
||||
if (is_string($order)) {
|
||||
$array = explode(',', $order);
|
||||
$order = [];
|
||||
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 = [0, $limit];
|
||||
}
|
||||
return $array;
|
||||
}
|
||||
|
||||
/**
|
||||
* field分析
|
||||
* @access protected
|
||||
* @param mixed $fields
|
||||
* @return array
|
||||
*/
|
||||
public function parseField($fields)
|
||||
{
|
||||
if (empty($fields)) {
|
||||
$fields = [];
|
||||
}
|
||||
if (is_string($fields)) {
|
||||
$fields = explode(',', $fields);
|
||||
}
|
||||
return $fields;
|
||||
}
|
||||
|
||||
/**
|
||||
* where分析
|
||||
* @access protected
|
||||
* @param mixed $where
|
||||
* @return array
|
||||
*/
|
||||
public function parseWhere($where)
|
||||
{
|
||||
$query = [];
|
||||
foreach ($where as $key => $val) {
|
||||
if ('_id' != $key && 0 === strpos($key, '_')) {
|
||||
// 解析特殊条件表达式
|
||||
$query = $this->parseThinkWhere($key, $val);
|
||||
} else {
|
||||
// 查询字段的安全过滤
|
||||
if (!preg_match('/^[A-Z_\|\&\-.a-z0-9]+$/', trim($key))) {
|
||||
throw new Exception(Lang::get('_ERROR_QUERY_') . ':' . $key);
|
||||
}
|
||||
$key = trim($key);
|
||||
if (strpos($key, '|')) {
|
||||
$array = explode('|', $key);
|
||||
$str = [];
|
||||
foreach ($array as $k) {
|
||||
$str[] = $this->parseWhereItem($k, $val);
|
||||
}
|
||||
$query['$or'] = $str;
|
||||
} elseif (strpos($key, '&')) {
|
||||
$array = explode('&', $key);
|
||||
$str = [];
|
||||
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);
|
||||
}
|
||||
}
|
||||
}
|
||||
return $query;
|
||||
}
|
||||
|
||||
/**
|
||||
* 特殊条件分析
|
||||
* @access protected
|
||||
* @param string $key
|
||||
* @param mixed $val
|
||||
* @return string
|
||||
*/
|
||||
protected function parseThinkWhere($key, $val)
|
||||
{
|
||||
$query = [];
|
||||
switch ($key) {
|
||||
case '_query': // 字符串模式查询条件
|
||||
parse_str($val, $query);
|
||||
if (isset($query['_logic']) && strtolower($query['_logic']) == 'or') {
|
||||
unset($query['_logic']);
|
||||
$query['$or'] = $query;
|
||||
}
|
||||
break;
|
||||
case '_string': // MongoCode查询
|
||||
$query['$where'] = new \MongoCode($val);
|
||||
break;
|
||||
}
|
||||
return $query;
|
||||
}
|
||||
|
||||
/**
|
||||
* where子单元分析
|
||||
* @access protected
|
||||
* @param string $key
|
||||
* @param mixed $val
|
||||
* @return array
|
||||
*/
|
||||
protected function parseWhereItem($key, $val)
|
||||
{
|
||||
$query = [];
|
||||
if (is_array($val)) {
|
||||
if (is_string($val[0])) {
|
||||
$con = strtolower($val[0]);
|
||||
if (in_array($con, ['neq', 'ne', 'gt', 'egt', 'gte', 'lt', 'lte', 'elt'])) {
|
||||
// 比较运算
|
||||
$k = '$' . $this->comparison[$con];
|
||||
$query[$key] = [$k => $val[1]];
|
||||
} elseif ('like' == $con) {
|
||||
// 模糊查询 采用正则方式
|
||||
$query[$key] = new \MongoRegex("/" . $val[1] . "/");
|
||||
} elseif ('mod' == $con) {
|
||||
// mod 查询
|
||||
$query[$key] = ['$mod' => $val[1]];
|
||||
} elseif ('regex' == $con) {
|
||||
// 正则查询
|
||||
$query[$key] = new \MongoRegex($val[1]);
|
||||
} elseif (in_array($con, ['in', 'nin', 'not in'])) {
|
||||
// IN NIN 运算
|
||||
$data = is_string($val[1]) ? explode(',', $val[1]) : $val[1];
|
||||
$k = '$' . $this->comparison[$con];
|
||||
$query[$key] = [$k => $data];
|
||||
} elseif ('all' == $con) {
|
||||
// 满足所有指定条件
|
||||
$data = is_string($val[1]) ? explode(',', $val[1]) : $val[1];
|
||||
$query[$key] = ['$all' => $data];
|
||||
} elseif ('between' == $con) {
|
||||
// BETWEEN运算
|
||||
$data = is_string($val[1]) ? explode(',', $val[1]) : $val[1];
|
||||
$query[$key] = ['$gte' => $data[0], '$lte' => $data[1]];
|
||||
} elseif ('not between' == $con) {
|
||||
$data = is_string($val[1]) ? explode(',', $val[1]) : $val[1];
|
||||
$query[$key] = ['$lt' => $data[0], '$gt' => $data[1]];
|
||||
} elseif ('exp' == $con) {
|
||||
// 表达式查询
|
||||
$query['$where'] = new \MongoCode($val[1]);
|
||||
} elseif ('exists' == $con) {
|
||||
// 字段是否存在
|
||||
$query[$key] = ['$exists' => (bool) $val[1]];
|
||||
} elseif ('size' == $con) {
|
||||
// 限制属性大小
|
||||
$query[$key] = ['$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] = ['$type' => intval($val[1])];
|
||||
} else {
|
||||
$query[$key] = $val;
|
||||
}
|
||||
return $query;
|
||||
}
|
||||
}
|
||||
$query[$key] = $val;
|
||||
return $query;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user