<?php
|
|
namespace DBOper;
|
|
include_once "/Common/CommFunc.php";
|
include_once "/Common/Logging.php";
|
include_once 'RedisOper.php';
|
|
class_exists("\MongoLog") && \MongoLog::setLevel(\MongoLog::NONE);
|
class_exists("\MongoLog") && \MongoLog::setModule(\MongoLog::NONE);
|
|
class MongoDBPool
|
{
|
// 注：由于php特殊原因，单次请求结束后会释放所有内存及链接，window系统下暂时无法处理链接池，故暂时只处理单次请求仅创建一个链接
|
private static $dbConn = null;
|
public static $dbConfig = null;
|
|
public static function GetDBConfig()
|
{
|
if (!self::$dbConfig) {
|
$cfgArray = parse_ini_file("/InterfaceConfig.php", true);
|
$session = "db";
|
if (!array_key_exists($session, $cfgArray)) {
|
\Logging\LogError("Config session key not exists! session=" . $session);
|
return;
|
}
|
self::$dbConfig = $cfgArray[$session];
|
\Logging\LogInfo("加载dbconfig.");
|
} else {
|
// \Logging\LogInfo("dbConfig已存在: ");
|
}
|
return self::$dbConfig;
|
}
|
|
public static function Get()
|
{
|
if (self::$dbConn) {
|
// \Logging\LogInfo("get static dbConn ok! ");
|
return self::$dbConn;
|
}
|
|
$dbConfig = self::GetDBConfig();
|
if (!$dbConfig) {
|
return;
|
}
|
|
$dbIP = $dbConfig["DBIP"];
|
$dbPort = $dbConfig["DBPort"];
|
$dbUser = $dbConfig["DBUser"];
|
$dbPws = $dbConfig["DBPsw"];
|
$dbUser = \Commfunc\GetDecodePsw($dbUser);
|
$dbPws = \Commfunc\GetDecodePsw($dbPws);
|
try {
|
$conn = new \MongoClient("mongodb://" . $dbUser . ":" . $dbPws . "@" . $dbIP . ":" . $dbPort);
|
\Logging\LogInfo("Mongo connect ok: " . $conn);
|
} catch (\Exception $e) {
|
\Logging\LogError('Mongo connect error: ' . $e->getMessage() . " db->" . print_r($dbConfig, true));
|
}
|
self::$dbConn = $conn;
|
return self::$dbConn;
|
}
|
|
public static function Close($conn)
|
{
|
//归还，暂不处理
|
}
|
}
|
|
class MongoDBOper
|
{
|
|
/**
|
* 建集合
|
* @param string $collectionName 集合名称
|
* @param array $indexArray 索引 array("key1"=>1, "key2"=>-1, ...) 1-升序，-1-降序
|
* @param array $indexParam 索引参数 array()， 如唯一索引 array(‘unique'=>true)
|
*/
|
public static function CreateCollection($collectionName, $indexArray = array(), $indexParam = array())
|
{
|
$conn = MongoDBPool::Get();
|
if (!$conn) {
|
return;
|
}
|
$dbName = GetDBName();
|
$retor = $conn->$dbName->$collectionName->findOne(array(), array("_id" => 1));
|
if ($retor) {
|
return;
|
}
|
$ret = $conn->$dbName->createCollection($collectionName);
|
if (!$ret) {
|
\Logging\LogError("CreateCollection error! " . $collectionName);
|
return;
|
}
|
\Logging\LogInfo("CreateCollection :" . $ret);
|
if ($indexArray) {
|
$ok = $conn->$dbName->$collectionName->ensureIndex($indexArray, $indexParam);
|
}
|
MongoDBPool::Close($conn);
|
return true;
|
}
|
|
public static function EnsureIndex($collectionName, $indexArray = array(), $indexParam = array())
|
{
|
// 1. 将等值索引放在最前面
|
// 2. 尽量将排序字段放在范围字段的前面
|
// 3. $nin和$ne跟索引没有关系
|
$conn = MongoDBPool::Get();
|
if (!$conn) {
|
return;
|
}
|
$dbName = GetDBName();
|
$ret = $conn->$dbName->$collectionName->ensureIndex($indexArray, $indexParam);
|
if (!isset($ret) || $ret["ok"] != 1) {
|
\Logging\LogError('Mongo EnsureIndex error: collectionName:' . $collectionName
|
. " indexArray:" . print_r($indexArray, true) . " indexParam:" . print_r($indexParam, true)
|
. " ret:" . print_r($ret, true));
|
return false;
|
}
|
MongoDBPool::Close($conn);
|
return true;
|
}
|
|
#查询单条数据, 以array形式返回内容
|
public static function FindOne(
|
$collectionName,
|
$findArray,
|
&$retArray,
|
$fields = null
|
) {
|
$conn = MongoDBPool::Get();
|
if (!$conn) {
|
return;
|
}
|
$dbName = GetDBName();
|
#查找数据
|
if (!isset($fields)) {
|
$retArray = $conn->$dbName->$collectionName->findOne($findArray);
|
} else {
|
$retArray = $conn->$dbName->$collectionName->findOne($findArray, $fields);
|
}
|
MongoDBPool::Close($conn);
|
return true;
|
}
|
|
#查询多天数据,以array形式返回内容
|
public static function Find(
|
$collectionName,
|
$findArray,
|
&$retArray,
|
$fields = null,
|
$sortArray = null,
|
$limit = 0,
|
$skip = 0
|
) {
|
$conn = MongoDBPool::Get();
|
if (!$conn) {
|
return;
|
}
|
$dbName = GetDBName();
|
$db = $conn->$dbName;
|
#判断集合是否存在
|
$collection = $db->$collectionName;
|
if (!isset($collection)) {
|
\Logging\LogInfo("Find 表名不存在! " . $collectionName);
|
// 当做没有数据
|
} else {
|
#查找数据
|
if ($fields) {
|
$retor = $collection->find($findArray, $fields);
|
} else {
|
$retor = $collection->find($findArray);
|
}
|
if (!$sortArray) {
|
$retor = $retor->skip($skip)->limit($limit);
|
} else {
|
$retor = $retor->skip($skip)->limit($limit)->sort($sortArray);
|
}
|
|
if ($retor) {
|
$retArray = iterator_to_array($retor);
|
} else {
|
$retArray = array();
|
}
|
}
|
MongoDBPool::Close($conn);
|
return true;
|
}
|
|
public static function Count($collectionName, $find)
|
{
|
$conn = MongoDBPool::Get();
|
if (!$conn) {
|
return 0;
|
}
|
$dbName = GetDBName();
|
$db = $conn->$dbName;
|
$count = $db->$collectionName->count($find);
|
MongoDBPool::Close($conn);
|
return $count;
|
}
|
|
public static function Insert($collectionName, $arrayInsert)
|
{
|
$conn = MongoDBPool::Get();
|
if (!$conn) {
|
return;
|
}
|
try {
|
$dbName = GetDBName();
|
$ret = $conn->$dbName->$collectionName->insert($arrayInsert);
|
MongoDBPool::Close($conn);
|
// (
|
// [n] => 0
|
// [connectionId] => 40
|
// [err] =>
|
// [ok] => 1
|
// )
|
if (!isset($ret) || $ret["ok"] != 1) {
|
\Logging\LogError('Mongo insert error: ' . print_r($arrayInsert, true) . " ret:" . print_r($ret, true));
|
return false;
|
}
|
} catch (\Exception $e) {
|
MongoDBPool::Close($conn);
|
\Logging\LogError('Mongo insert error: Exception: ' . $e->getMessage() . print_r($arrayInsert, true));
|
return false;
|
}
|
return true;
|
}
|
|
public static function BatchInsert($collectionName, $arrayInsert)
|
{
|
$conn = MongoDBPool::Get();
|
if (!$conn) {
|
return;
|
}
|
try {
|
$dbName = GetDBName();
|
$ret = $conn->$dbName->$collectionName->batchInsert($arrayInsert);
|
MongoDBPool::Close($conn);
|
if (!isset($ret) || $ret["ok"] != 1) {
|
\Logging\LogError('Mongo batchInsert error: ' . print_r($arrayInsert, true) . " ret:" . print_r($ret, true));
|
return false;
|
}
|
// \Logging\LogInfo('Mongo batchInsert ok: ret=' . print_r($ret, true));
|
} catch (\Exception $e) {
|
MongoDBPool::Close($conn);
|
\Logging\LogError('Mongo batchInsert error: Exception: ' . $e->getMessage() . print_r($arrayInsert, true));
|
return false;
|
}
|
return true;
|
}
|
|
public static function Update($collectionName, $find, $setArray, $upsert = false, $multi = false)
|
{
|
$conn = MongoDBPool::Get();
|
if (!$conn) {
|
return;
|
}
|
$dbName = GetDBName();
|
try {
|
// $options { upsert: <boolean>, multi: <boolean>, ...}
|
// * 更新 multiple VS multi 更新时，需要指定是否更新匹配的所有记录，如果设置为false,表示只更新匹配到的一行。
|
// * 这个坑很大，mongo扩展(php5.x用)里指定字段名为 multiple ，而 mongodb扩展(php7用)里使用的字段是 multi
|
$set = array('$set' => $setArray);
|
$options = array("upsert" => $upsert, "multiple" => $multi);
|
$ret = $conn->$dbName->$collectionName->update($find, $set, $options);
|
MongoDBPool::Close($conn);
|
// \Logging\LogInfo('Mongo update: $ret=' . $ret . ', find=' . json_encode($find) . " set=" . json_encode($set) . " options=" . json_encode($options));
|
if (!$ret) {
|
\Logging\LogError('Mongo update error: find=' . json_encode($find) . " set=" . json_encode($set) . " options=" . json_encode($options));
|
return false;
|
}
|
} catch (\Exception $e) {
|
MongoDBPool::Close($conn);
|
\Logging\LogError('Mongo update error: Exception: ' . $e->getMessage() . ' find=' . print_r($find, true) . " set=" . print_r($setArray, true));
|
return false;
|
}
|
return true;
|
}
|
|
public static function Remove($collectionName, $find)
|
{
|
$conn = MongoDBPool::Get();
|
if (!$conn) {
|
return;
|
}
|
$dbName = GetDBName();
|
$ret = $conn->$dbName->$collectionName->remove($find);
|
\Logging\LogInfo('Mongo Remove:' . $collectionName . ' $ret=' . $ret . ', find=' . print_r($find, true));
|
MongoDBPool::Close($conn);
|
return true;
|
}
|
|
public static function Aggregate($collectionName, $pipeline, &$retArray)
|
{
|
$conn = MongoDBPool::Get();
|
if (!$conn) {
|
return;
|
}
|
$dbName = GetDBName();
|
$ret = $conn->$dbName->$collectionName->aggregate($pipeline);
|
if (isset($ret) && $ret["ok"] == 1) {
|
$retArray = $ret["result"];
|
}
|
// \Logging\LogInfo('Mongo Aggregate: $ret=' . $ret);
|
MongoDBPool::Close($conn);
|
return true;
|
}
|
}
|
|
function GetDBName()
|
{
|
$dbConfig = MongoDBPool::GetDBConfig();
|
return $dbConfig ? $dbConfig["DataBaseName"] : "";
|
}
|
|
/**
|
* 建集合
|
* @param string $collectionName 集合名称
|
* @param array $indexArray 索引 array("key1"=>1, "key2"=>-1, ...) 1-升序，-1-降序
|
* @param array $indexParam 索引参数 array()， 如唯一索引 array(‘unique'=>true)
|
*/
|
function CreateCollection($collectionName, $indexArray = array(), $indexParam = array())
|
{
|
return MongoDBOper::CreateCollection($collectionName, $indexArray, $indexParam);
|
}
|
|
function EnsureIndex($collectionName, $indexArray = array(), $indexParam = array())
|
{
|
return MongoDBOper::EnsureIndex($collectionName, $indexArray, $indexParam);
|
}
|
|
/**
|
* 获取数据，优先从redis中获取，找不到的话再从db中查询
|
*/
|
function FindOne($collectionName, $find, &$retArray, $fields = null, $useRedis = true)
|
{
|
$dbName = GetDBName();
|
if (!$dbName) {
|
return false;
|
}
|
|
$redisKey = "";
|
if ($useRedis) {
|
$redisKey = \RedisOper\GetDBRedisKey($dbName, $collectionName, $find);
|
// \Logging\LogInfo("Start to get redis data! redisKey=" . $redisKey);
|
if (\RedisOper\GetHashObj($redisKey, $retArray)) {
|
return true;
|
}
|
}
|
|
// 取不到数据从db取
|
if (!MongoDBOper::FindOne($collectionName, $find, $retArray, $fields)) {
|
return false;
|
}
|
|
if ($useRedis) {
|
if ($retArray) {
|
\RedisOper\SetHashObj($redisKey, $retArray);
|
}
|
}
|
return true;
|
}
|
|
function FindOneSort($collectionName, $find, &$retArray, $fields = null, $sortArray = null)
|
{
|
$dbName = GetDBName();
|
if (!$dbName) {
|
return false;
|
}
|
if (!MongoDBOper::Find($collectionName, $find, $retFindArray, $fields, $sortArray, 1)) {
|
return false;
|
}
|
if (isset($retFindArray)) {
|
foreach ($retFindArray as $value) {
|
$retArray = $value;
|
break;
|
}
|
}
|
return true;
|
}
|
|
function Find(
|
$collectionName,
|
$find,
|
&$retArray,
|
$fields = null,
|
$sortArray = null,
|
$limit = 0,
|
$skip = 0
|
) {
|
$dbName = GetDBName();
|
if (!$dbName) {
|
return false;
|
}
|
|
// find多条数据的暂不从redis取
|
// $redisKey = \RedisOper\GetDBRedisKey($dbName, $collectionName, $find);
|
// \Logging\LogInfo("Start to get redis data! redisKey=" . $redisKey);
|
// if (\RedisOper\GetHashObj($redisKey, $retArray)) {
|
// return true;
|
// }
|
|
// 取不到数据从db取
|
if (!MongoDBOper::Find($collectionName, $find, $retArray, $fields, $sortArray, $limit, $skip)) {
|
return false;
|
}
|
|
// if (!$retArray || count($retArray) <= 0) {
|
// $retArray = $find;
|
// }
|
// \RedisOper\SetHashObj($redisKey, $retArray);
|
return true;
|
}
|
|
|
function Count($collectionName, $find)
|
{
|
return MongoDBOper::Count($collectionName, $find);
|
}
|
|
function Insert($collectionName, $insArray, $redisKeyFind = null)
|
{
|
$dbName = GetDBName();
|
if (!$dbName) {
|
return false;
|
}
|
|
if (!MongoDBOper::Insert($collectionName, $insArray)) {
|
return false;
|
}
|
|
if ($redisKeyFind) {
|
$redisKey = \RedisOper\GetDBRedisKey($dbName, $collectionName, $redisKeyFind);
|
\RedisOper\SetHashObj($redisKey, $insArray);
|
}
|
return true;
|
}
|
|
function BatchInsert($collectionName, $batchInsArray)
|
{
|
$dbName = GetDBName();
|
if (!$dbName) {
|
return false;
|
}
|
|
if (!MongoDBOper::BatchInsert($collectionName, $batchInsArray)) {
|
return false;
|
}
|
|
return true;
|
}
|
|
/**
|
* 更新数据
|
* @param array $upsert 如果查不到对应数据，那么则以更新条件新增一条数据，默认false
|
* @param array $multi 是否批量更新，默认false
|
*/
|
function Update($collectionName, $find, $setArray, $updRedis = false, $upsert = false, $multi = false)
|
{
|
$dbName = GetDBName();
|
if (!$dbName) {
|
return false;
|
}
|
|
if (!MongoDBOper::Update($collectionName, $find, $setArray, $upsert, $multi)) {
|
return false;
|
}
|
|
if ($updRedis) {
|
$redisKey = \RedisOper\GetDBRedisKey($dbName, $collectionName, $find);
|
\RedisOper\SetHashValues($redisKey, $setArray);
|
}
|
return true;
|
}
|
|
function Remove($collectionName, $find, $delReids = true)
|
{
|
$dbName = GetDBName();
|
if (!$dbName) {
|
return false;
|
}
|
|
if (!MongoDBOper::Remove($collectionName, $find)) {
|
return false;
|
}
|
|
if ($delReids) {
|
$redisKey = \RedisOper\GetDBRedisKey($dbName, $collectionName, $find);
|
\RedisOper\DelKey($redisKey);
|
}
|
return true;
|
}
|
|
function Aggregate($collectionName, $pipeline, &$retArray)
|
{
|
$dbName = GetDBName();
|
if (!$dbName) {
|
return false;
|
}
|
|
if (!MongoDBOper::Aggregate($collectionName, $pipeline, $retArray)) {
|
return false;
|
}
|
|
return true;
|
}
|
