<?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;
}