开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 游戏 > 模拟服务器 > 查看源码
KRoleDBServer.cpp
资源名称：SwordOnline.rar　[点击查看]
上传用户：dzyhzl
上传日期：2019-04-29
资源大小：56270k
文件大小：8k
源码类别：

模拟服务器

开发平台：

C/C++

KRoleDBServer.cpp：源码内容
  1. #include "KRoleDBServer.h"
  2. #include "KRoleInfomation.h"
  3. #include "Windows.h"
  4. #include "KDBProcessThread.h"
  5. #include <crtdbg.h>
  6. #include "KException.h"
  7. #pragma warning(disable:4786)
  8. #include <objbase.h>
  9. #include <initguid.h>
  10. extern void * EnumSendMsg(unsigned long nId, size_t& nDataLen);
  11. //using namespace RoleDBServer
  12. RoleDBServer::KNetServer g_NetServer;
  13. size_t g_nMaxSendBuffer = 350;
  14. extern TRoleNetMsg  * pTestCmd;
  15. extern void SendToClient(unsigned long nId, TRoleNetMsg * pMsg, size_t nMsgLen);
  16. extern IServer * g_pNetServer;
  17. namespace RoleDBServer
  18. {
  20. KClientUser::KClientUser()
  21. {
  22. m_pCurrentRecvProcessData = NULL;
  23. m_pCurrentSendProcessData = NULL;
  24. m_nProcessLoadCount = m_nProcessWriteCount = 0;
  25. this->m_nCurSendLen = 0;
  26. this->m_nCurGetRecvLen = 0;
  27. this->m_szClientName[0] = 0;
  28. }
  30. int KNetServer::Init()
  31. {
  32. m_pNetServer = NULL;
  33. if (g_pNetServer)
  34. {
  35. g_pNetServer->QueryInterface( IID_IIOCPServer, reinterpret_cast< void ** >( &m_pNetServer ) );
  36. }
  38. if (!m_pNetServer)
  39. {
  40. throw KException("NetServer", "Cloning Net Server Error");
  41. }
  42. return 1;
  43. }
  45. void KNetServer::RegisterClient(unsigned long nId)
  46. {
  47. KClientUser * pNewUser = new KClientUser;
  48. pNewUser->m_ID = nId;
  49. m_ClientUserSet[nId] = pNewUser;
  50. }
  52. void KNetServer::DestoryClient(unsigned long nId)
  53. {
  54. KClientUser * pDelUse = m_ClientUserSet[nId];
  55. if (pDelUse)
  56. {
  57. m_ClientUserSet.erase(nId);
  58. delete pDelUse;
  59. }
  60. }
  62. //public
  63. int KNetServer::Receive()
  64. {
  65. KClientUserSet::iterator I;
  66. try{
  67. WaitForSingleObject(g_NetEventMutex, INFINITE);
  68. for ( I = m_ClientUserSet.begin(); I != m_ClientUserSet.end(); I ++ )
  69. {
  70. Receive(I->first);
  71. }
  72. ReleaseMutex(g_NetEventMutex);
  73. }
  74. catch(...)
  75. {
  77. }
  78. return 1;
  79. }
  81. int KNetServer::Send()
  82. {
  84. KClientUserSet::iterator I;
  85. m_pNetServer->PreparePackSink();
  86. for ( I = m_ClientUserSet.begin(); I != m_ClientUserSet.end(); I ++)
  87. {
  88. Send(I->first);
  89. }
  90. m_pNetServer->SendPackToClient(-1);
  91. return 1;
  92. }
  95. //private
  96. int KNetServer::Receive(unsigned long nId)
  97. {
  98. //获得数据
  99. size_t nDataLen = 0;
  100. TRoleNetMsg *pMsg;
  101. pMsg =(TRoleNetMsg*) m_pNetServer->GetPackFromClient(nId, nDataLen);
  103. if (nDataLen == 0 || pMsg == NULL) return 0;
  104. return AppendData(nId, pMsg, nDataLen);
  105. }
  107. int KNetServer::AppendData(unsigned long nId, TRoleNetMsg * pMsgBuffer, size_t nMsgLen)
  108. {
  110. switch(pMsgBuffer->bDataState)
  111. {
  112. case 1://游戏服务器向数据库服务器要求保存数据
  113. {
  114. KClientUser * pClientUser = m_ClientUserSet[nId];
  115. _ASSERT(!pClientUser->m_pCurrentRecvProcessData);
  116. if (!nMsgLen) return 0;
  117. pClientUser->m_pCurrentRecvProcessData =(TCmd*) new char[pMsgBuffer->nOffset + sizeof(TCmd) - 1];
  118. pClientUser->m_pCurrentRecvProcessData->nDataLen = pMsgBuffer->nOffset;
  119. pClientUser->m_pCurrentRecvProcessData->nProtoId = pMsgBuffer->nProtoId;
  120. pClientUser->m_pCurrentRecvProcessData->ulNetId = nId;
  122. memcpy(pClientUser->m_pCurrentRecvProcessData->pDataBuffer, &pMsgBuffer->pDataBuffer, pMsgBuffer->nDataLen);
  123. pClientUser->m_nCurGetRecvLen = pMsgBuffer->nDataLen;
  125. if (pClientUser->m_pCurrentRecvProcessData->nDataLen == pClientUser->m_nCurGetRecvLen)
  126. {
  127. InsertProcessData2Set(nId);
  128. ClearCurProcessData(nId);
  129. }
  131. }
  132. break;
  133. case 0:
  134. {
  135. KClientUser * pClientUser = m_ClientUserSet[nId];
  136. _ASSERT(pClientUser->m_pCurrentRecvProcessData->pDataBuffer);
  137. //_ASSERT(pClientUser->m_nCurGetRecvLen == pMsgBuffer->nOffset);
  138. _ASSERT(pClientUser->m_pCurrentRecvProcessData->nProtoId == pMsgBuffer->nProtoId);
  140. if (pClientUser->m_nCurGetRecvLen + pMsgBuffer->nDataLen > pClientUser->m_pCurrentRecvProcessData->nDataLen)
  141. _ASSERT(0);
  142. else
  143. {
  144. memcpy(pClientUser->m_pCurrentRecvProcessData->pDataBuffer + pClientUser->m_nCurGetRecvLen , &pMsgBuffer->pDataBuffer, pMsgBuffer->nDataLen);
  145. pClientUser->m_nCurGetRecvLen += pMsgBuffer->nDataLen;
  146. if (pClientUser->m_pCurrentRecvProcessData->nDataLen == pClientUser->m_nCurGetRecvLen)
  147. {
  148. InsertProcessData2Set(nId);
  149. ClearCurProcessData(nId);
  150. }
  151. }
  153. }break;
  155. default:
  156. {
  157. _ASSERT(0);
  159. }
  160. }
  161. return 1;
  162. }
  164. void KNetServer::InsertProcessData2Set(int nId)
  165. {
  166. KClientUser * pClientUser = m_ClientUserSet[nId];
  167. TCmd * pNewCmd = (TCmd *) (new char [sizeof(TCmd) + pClientUser->m_pCurrentRecvProcessData->nDataLen - 1]);
  168. pNewCmd->ulNetId = nId;
  169. pNewCmd->nDataLen = pClientUser->m_pCurrentRecvProcessData->nDataLen;
  170. pNewCmd->nProtoId  = pClientUser->m_pCurrentRecvProcessData->nProtoId;
  171. memcpy(&pNewCmd->pDataBuffer[0], pClientUser->m_pCurrentRecvProcessData->pDataBuffer, pClientUser->m_pCurrentRecvProcessData->nDataLen );
  173. EnterCriticalSection(&g_MainMsgListMutex);
  174. g_MainThreadCmdList.push_back(pNewCmd);
  175. pClientUser->m_nProcessLoadCount ++;
  176. LeaveCriticalSection(&g_MainMsgListMutex);
  177. }
  180. int KNetServer::Send(unsigned long nId)
  181. {
  182. KClientUser * pClient = m_ClientUserSet[nId];
  183. size_t ntmpSendLen = 0; //发送实际数据中的长度
  184. TRoleNetMsg * pMsg = NULL;
  185. bool bCompleteSend = false;
  186. char * pSendBuffer = NULL;
  187. if (pClient->m_pCurrentSendProcessData)
  188. {
  189. if (pClient->m_nCurSendLen == 0)
  190. {
  191. //如果一次性能够发完的
  192. if (pClient->m_pCurrentSendProcessData->nDataLen + sizeof(TRoleNetMsg) - 1 < g_nMaxSendBuffer)
  193. {
  194. pSendBuffer = new char [pClient->m_pCurrentSendProcessData->nDataLen + sizeof(TRoleNetMsg) - 1];
  195. ntmpSendLen = pClient->m_pCurrentSendProcessData->nDataLen ;
  196. bCompleteSend = true;
  198. }
  199. else
  200. {
  201. pSendBuffer = new char[g_nMaxSendBuffer];
  202. ntmpSendLen = g_nMaxSendBuffer - sizeof(TRoleNetMsg) + 1;
  203. bCompleteSend = false;
  204. pClient->m_nCurSendLen = ntmpSendLen;
  205. }
  207. pMsg = (TRoleNetMsg*)pSendBuffer;
  208. pMsg->nProtoId = pClient->m_pCurrentSendProcessData->nProtoId;
  209. pMsg->bDataState = 1;//first package send;
  210. pMsg->nOffset = pClient->m_pCurrentSendProcessData->nDataLen;
  211. pMsg->nDataLen = ntmpSendLen;
  213. memcpy(pMsg->pDataBuffer, pClient->m_pCurrentSendProcessData->pDataBuffer, ntmpSendLen);
  215. }
  216. else
  217. {
  218. if (pClient->m_pCurrentSendProcessData->nDataLen > pClient->m_nCurSendLen + g_nMaxSendBuffer - sizeof(TRoleNetMsg) + 1 )
  219. {
  220. ntmpSendLen = g_nMaxSendBuffer - sizeof(TRoleNetMsg) + 1;
  221. pSendBuffer = new char[g_nMaxSendBuffer] ;
  222. bCompleteSend = false;
  223. }
  224. else
  225. {
  226. ntmpSendLen = pClient->m_pCurrentSendProcessData->nDataLen - pClient->m_nCurSendLen ;
  227. pSendBuffer = new char[ntmpSendLen + sizeof(TRoleNetMsg) - 1];
  228. bCompleteSend = true;
  229. }
  230. pMsg = (TRoleNetMsg*)pSendBuffer;
  231. pMsg->nProtoId = pClient->m_pCurrentSendProcessData->nProtoId;
  232. pMsg->nOffset =  pClient->m_nCurSendLen;
  233. pMsg->nDataLen = ntmpSendLen;
  234. pMsg->bDataState = 0;
  235. memcpy(pMsg->pDataBuffer, pClient->m_pCurrentSendProcessData->pDataBuffer + pClient->m_nCurSendLen, ntmpSendLen);
  236. pClient->m_nCurSendLen += ntmpSendLen;
  238. }
  239.  m_pNetServer->PackDataToClient(nId, pMsg, ntmpSendLen + sizeof(TRoleNetMsg) - 1);
  240. }
  241. else
  242. {
  243. bCompleteSend = true;
  244. }
  248. if (bCompleteSend)
  249. {
  250. if (pClient->m_pCurrentSendProcessData)
  251. {
  252. delete pClient->m_pCurrentSendProcessData;
  253. pClient->m_pCurrentSendProcessData = NULL;
  254. }
  256. EnterCriticalSection(&g_NetMsgListMutex);
  257. list<TCmd *>::iterator I = g_NetServiceThreadCmdList.begin();
  259. if (I != g_NetServiceThreadCmdList.end())
  260. {
  261. TProcessData * pNewProcess = (TCmd *)new char[(*I)->nDataLen + sizeof(TProcessData) - 1];
  262. pNewProcess->nDataLen = (*I)->nDataLen;
  263. pNewProcess->nProtoId = (*I)->nProtoId;
  264. pNewProcess->ulNetId = (*I)->ulNetId;
  267. memcpy(&pNewProcess->pDataBuffer[0], (*I)->pDataBuffer, (*I)->nDataLen);
  269. pClient->m_pCurrentSendProcessData = pNewProcess;
  270. //how to delete 
  271. g_NetServiceThreadCmdList.pop_front();
  272. pClient->m_nCurSendLen = 0;
  273. pClient->m_nProcessWriteCount ++;
  274. }
  276. LeaveCriticalSection(&g_NetMsgListMutex);
  278. }
  279. return 1;
  280. }
  282. void KNetServer::ClearCurProcessData(unsigned long nId)
  283. {
  284. KClientUser * pClientUser = m_ClientUserSet[nId];
  285. delete pClientUser->m_pCurrentRecvProcessData;
  286. pClientUser->m_pCurrentRecvProcessData = NULL;
  288. }
  290. }