added:auto now bucket size Calculate

Author: xengine-qyt

XEngine_Source/StorageModule_APIHelp/APIHelp_Define.h

@@ -180,6 +180,30 @@ extern "C" bool APIHelp_Distributed_GetPathKey(list<XENGINE_STORAGEBUCKET>* pStl
 备注：
 *********************************************************************/
 extern "C" __int64u APIHelp_Distributed_GetSize(LPCXSTR lpszMsgBuffer);
+/********************************************************************
+函数名称：APIHelp_Distributed_SetSize
+函数功能：设置指定BUCKET当前大小
+ 参数.一：pStl_ListBucket
+  In/Out：In
+  类型：STL容器
+  可空：N
+  意思：输入BUCKET列表
+ 参数.二：lpszBuckKey
+  In/Out：In
+  类型：常量字符指针
+  可空：N
+  意思：输入要查询的BUCKET名称
+ 参数.三：nSize
+  In/Out：In
+  类型：整数型
+  可空：N
+  意思：支持+ - 操作
+返回值
+  类型：逻辑型
+  意思：是否成功
+备注：
+*********************************************************************/
+extern "C" bool APIHelp_Distributed_SetSize(list<XENGINE_STORAGEBUCKET>* pStl_ListBucket, LPCXSTR lpszBuckKey, __int64x nSize);
 /************************************************************************/
 /*                       帮助函数                                       */
 /************************************************************************/

XEngine_Source/StorageModule_APIHelp/APIHelp_Distributed/APIHelp_Distributed.cpp

@@ -523,6 +523,57 @@ __int64u CAPIHelp_Distributed::APIHelp_Distributed_GetSize(LPCXSTR lpszMsgBuffer
 
 	return nllSize;
 }
+/********************************************************************
+函数名称：APIHelp_Distributed_SetSize
+函数功能：设置指定BUCKET当前大小
+ 参数.一：pStl_ListBucket
+  In/Out：In
+  类型：STL容器
+  可空：N
+  意思：输入BUCKET列表
+ 参数.二：lpszBuckKey
+  In/Out：In
+  类型：常量字符指针
+  可空：N
+  意思：输入要查询的BUCKET名称
+ 参数.三：nSize
+  In/Out：In
+  类型：整数型
+  可空：N
+  意思：支持+ - 操作
+返回值
+  类型：逻辑型
+  意思：是否成功
+备注：
+*********************************************************************/
+bool CAPIHelp_Distributed::APIHelp_Distributed_SetSize(list<XENGINE_STORAGEBUCKET>* pStl_ListBucket, LPCXSTR lpszBuckKey, __int64x nSize)
+{
+	APIHelp_IsErrorOccur = false;
+
+	if ((NULL == pStl_ListBucket) || (NULL == lpszBuckKey))
+	{
+		APIHelp_IsErrorOccur = true;
+		APIHelp_dwErrorCode = ERROR_STORAGE_MODULE_APIHELP_PARAMENT;
+		return false;
+	}
+	bool bFound = false;
+	for (auto stl_ListIterator = pStl_ListBucket->begin(); stl_ListIterator != pStl_ListBucket->end(); stl_ListIterator++)
+	{
+		if (0 == _tcsxncmp(lpszBuckKey, stl_ListIterator->tszBuckKey, _tcsxlen(stl_ListIterator->tszBuckKey)))
+		{
+			stl_ListIterator->nBuckSize += nSize;
+			bFound = true;
+			break;
+		}
+	}
+	if (!bFound)
+	{
+		APIHelp_IsErrorOccur = true;
+		APIHelp_dwErrorCode = ERROR_STORAGE_MODULE_APIHELP_NOTFOUND;
+		return false;
+	}
+	return true;
+}
 //////////////////////////////////////////////////////////////////////////
 //                               保护函数
 //////////////////////////////////////////////////////////////////////////

XEngine_Source/StorageModule_APIHelp/APIHelp_Distributed/APIHelp_Distributed.h

@@ -25,6 +25,7 @@ class CAPIHelp_Distributed
 	bool APIHelp_Distributed_UPStorage(list<XENGINE_STORAGEBUCKET>* pStl_ListBucket, XENGINE_STORAGEBUCKET* pSt_StorageBucket, int nMode);
 	bool APIHelp_Distributed_GetPathKey(list<XENGINE_STORAGEBUCKET>* pStl_ListBucket, LPCXSTR lpszBuckKey, XCHAR* ptszFilePath);
 	__int64u APIHelp_Distributed_GetSize(LPCXSTR lpszMsgBuffer);
+	bool APIHelp_Distributed_SetSize(list<XENGINE_STORAGEBUCKET>* pStl_ListBucket, LPCXSTR lpszBuckKey, __int64x nSize);
 protected:
 	bool APIHelp_Distributed_FileListParse(LPCXSTR lpszMsgBuffer, int nMsgLen, XSTORAGECORE_DBFILE* pSt_DBFile);
 private:

XEngine_Source/StorageModule_APIHelp/StorageModule_APIHelp.def

@@ -10,6 +10,7 @@ EXPORTS
 	APIHelp_Distributed_UPStorage
 	APIHelp_Distributed_GetPathKey
 	APIHelp_Distributed_GetSize
+	APIHelp_Distributed_SetSize
 
 	APIHelp_Api_RangeFile
 	APIHelp_Api_VerHash

XEngine_Source/StorageModule_APIHelp/pch.cpp

@@ -59,6 +59,10 @@ extern "C" __int64u APIHelp_Distributed_GetSize(LPCXSTR lpszMsgBuffer)
 {
 	return m_APIDistributed.APIHelp_Distributed_GetSize(lpszMsgBuffer);
 }
+extern "C" bool APIHelp_Distributed_SetSize(list<XENGINE_STORAGEBUCKET>* pStl_ListBucket, LPCXSTR lpszBuckKey, __int64x nSize)
+{
+	return m_APIDistributed.APIHelp_Distributed_SetSize(pStl_ListBucket, lpszBuckKey, nSize);
+}
 /************************************************************************/
 /*                       帮助函数                                       */
 /************************************************************************/

XEngine_Source/XEngine_StorageApp/StorageApp_UPLoader.cpp

@@ -541,6 +541,7 @@ bool XEngine_Task_HttpUPLoader(LPCXSTR lpszClientAddr, LPCXSTR lpszMsgBuffer, in
 			XEngine_Net_SendMsg(lpszClientAddr, tszSDBuffer, nSDLen, nNetType);
 			XLOG_PRINT(xhLog, XENGINE_HELPCOMPONENTS_XLOG_IN_LOGLEVEL_WARN, _X("上传客户端:%s,请求上传文件成功,文件名:%s,大小:%d,数据库没有启用,不插入数据库"), lpszClientAddr, tszFileDir, nRVCount);
 		}
+		APIHelp_Distributed_SetSize(st_LoadbalanceCfg.st_LoadBalance.pStl_ListBucket, st_StorageBucket.tszBuckKey, st_ProtocolFile.st_ProtocolFile.nFileSize);
 	}
 	else
 	{