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

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

模拟服务器

开发平台：

C/C++

KDBProcessThread.cpp：源码内容
  1. #include "stdafx.h"
  2. #include <objbase.h>
  3. #include <initguid.h>
  4. #include "KRoleInfomation.h"
  5. #include "KRoleDBServer.h"
  6. #include "KException.h"
  7. #include "S3DBINTERFACE.H"
  8. #include "KDBProcessThread.h"
  9. #include "KROleDBHeader.h"
  10. #include "resource.h"
  11. #include "MainFrm.h"
  12. #include "RoleDBServerDoc.h"
  13. #include "RoleDBServerView.h"
  18. #pragma warning(disable:4786)
  19. #include <crtdbg.h>
  20. #include <list>
  21. #include "stdio.h"
  22. using namespace std;
  24. static const int g_snMaxPlayerCount = 5;
  25. static const int g_snPrecision = 1;
  26. static const int g_snMaxBuffer = 15;
  27. static const int g_snBufferSize = 1024;
  29. IServer *g_pNetServer = NULL;
  30. static HMODULE hDllModule;
  32. DWORD g_dwPerSaveTime;//每次存档的时间
  33. DWORD g_dwMaxUnModifyTime;
  34. list<TCmd*> g_MainThreadCmdList;
  35. list<TGetRoleInfoFromDB*> g_DBLoadThreadCmdList;
  36. list<TCmd*> g_NetServiceThreadCmdList;
  37. extern BOOL InitNetEngine();
  38. extern RoleDBServer::KNetServer g_NetServer;
  40. CRITICAL_SECTION g_RoleInfoMutex;
  41. CRITICAL_SECTION g_RoleInfoListMutex;
  42. CRITICAL_SECTION g_RoleInfoMapMutex;
  43. CRITICAL_SECTION g_MainMsgListMutex;
  44. CRITICAL_SECTION g_NetMsgListMutex;
  45. CRITICAL_SECTION g_GetRoleInfoMutex;
  46. HANDLE  g_NetEventMutex = 0;
  49. TThreadData g_ThreadData[4];
  51. char * GetRoleNameFromRoleBuffer(char * pRoleBuffer)
  52. {
  53. if (pRoleBuffer == NULL)
  54. return NULL;
  55. try
  56. {
  57. TRoleData *pRoleData = (TRoleData*) pRoleBuffer;
  58. return pRoleData->BaseInfo.szName;
  59. }
  60. catch(...)
  61. {
  62. throw KException("GetRoleNameFromRoleBuffer", "内存有问题");
  63. }
  64. }
  66. int InitServer()
  67. {
  68. int nReturn;
  69. S3DBI_InitDBInterface();
  70. nReturn = InitNetEngine();
  71. InitializeCriticalSection(&g_RoleInfoMutex);
  72. InitializeCriticalSection(&g_MainMsgListMutex);
  73. InitializeCriticalSection(&g_NetMsgListMutex);
  74. InitializeCriticalSection(&g_GetRoleInfoMutex);
  75. InitializeCriticalSection(&g_RoleInfoListMutex);
  76. InitializeCriticalSection(&g_RoleInfoMapMutex);
  77. g_NetEventMutex = CreateMutex(NULL, FALSE, NULL);
  78. if (!g_NetEventMutex) return 0;
  79. return nReturn;
  80. }
  82. DWORD ReleaseServer()
  83. {
  85. for(int i = 0; i < 4; i ++)
  86. TerminateThread(g_ThreadData[i].hThread,0);
  88. DeleteCriticalSection(&g_RoleInfoMutex);
  89. DeleteCriticalSection(&g_MainMsgListMutex);
  90. DeleteCriticalSection(&g_NetMsgListMutex);
  91. DeleteCriticalSection(&g_GetRoleInfoMutex);
  92. DeleteCriticalSection(&g_RoleInfoListMutex);
  93. DeleteCriticalSection(&g_RoleInfoMapMutex);
  94. CloseHandle(g_NetEventMutex);
  96. g_pNetServer->CloseService();
  97. g_pNetServer->Cleanup();
  98. g_pNetServer->Release();
  99. ::FreeLibrary( hDllModule );
  101. S3DBI_ReleaseDBInterface();
  102. return 1;
  103. }
  105. //
  106. DWORD WINAPI DatabaseSaveThreadFun(void * pParam)
  107. {
  108. /*
  109. while(1)
  110. {
  111. list<KRoleInfomation*>::iterator I = g_RoleInfoList.begin();
  112. while(!I->end())
  113. {
  114. if (I->m_bModify)
  115. {
  116. if (I->CheckRoleInfoValid())
  117. {
  118. S3DBI_SaveRoleInfo()
  119. (*I)->m_nSaveDBCount ++;
  120. (*I)->m_nUnModifyTime = 0;
  121. }
  122. else
  123. {
  124. ASSERT(0);
  125. }
  127.   }
  128.   else
  129.   {
  130.   //长时间未存则从存档列表中删除
  131.   if (I->m_nUnModifyTime >= g_dwMaxUnModifyTime)
  132.   {
  133.   //Question
  134.   I->erase();
  135.   }
  136.   else
  137.   I->m_nLastSaveTime ++;
  138.   
  139. }
  140. I ++;
  141. }
  143.   Sleep(g_dwPerSaveTime);
  144.   }
  145. */
  146. return 1;
  147. }
  149. //循环处理向数据库请求获得玩家信息的操作 
  150. DWORD WINAPI DatabaseLoadThreadFun(void * pParam)
  151. {
  152. CMainFrame * pFrame = (CMainFrame *) AfxGetMainWnd(); CRoleDBServerView *pView = (CRoleDBServerView*)pFrame->GetActiveView();
  153. while(1)
  154. {
  155. list<TGetRoleInfoFromDB *>::iterator I = g_DBLoadThreadCmdList.begin();
  156. pView->UpdateDBServiceList();
  157. while(I != g_DBLoadThreadCmdList.end())
  158. {
  159. static unsigned char szRoleInfo[MAX_ROLEINFORMATION_SIZE];
  160. int nBufferSize = 0;
  161. char * szName1 =  (*I)->szRoleName;
  162. if (!S3DBI_GetRoleInfo(szRoleInfo, (*I)->szRoleName, nBufferSize)) continue;
  164. static ii = 0;
  165. char FileName[100] ;
  166. sprintf(FileName, "d:\LoadRoleout_%d.txt", ii++);
  167. TextOutRoleInfo(szRoleInfo ,FileName );
  170. //增加角色数据和向两个列表增加数据
  172. EnterCriticalSection(&g_RoleInfoMutex);//同时只能有一个线程修改角色数据
  174. char * szName = (char *)((*I)->szRoleName);
  175. if (szName)
  176. {
  177. KRoleInfomation * pRoleInfomation =   g_RoleInfoMap[szName] ;
  179. if (!pRoleInfomation)//新加角色！需要立即存档
  180. {
  181. KRoleInfomation * pNewRoleInfomation = new KRoleInfomation;
  183. pNewRoleInfomation->SetRoleInfoBuffer(szRoleInfo, nBufferSize);
  184. LeaveCriticalSection(&g_RoleInfoMutex);
  186. EnterCriticalSection(&g_RoleInfoMapMutex);
  188. g_RoleInfoMap[szName] = pNewRoleInfomation;
  190. EnterCriticalSection(&g_RoleInfoListMutex);
  191. g_RoleInfoList.push_back(pNewRoleInfomation);
  193. LeaveCriticalSection(&g_RoleInfoMapMutex);
  194. LeaveCriticalSection(&g_RoleInfoListMutex);
  196. }
  199. TCmd * pNewCmd = (TCmd *)new char [sizeof(TCmd)  + strlen(szName)];
  200. pNewCmd->ulNetId = (*I)->nNetId;
  201. pNewCmd->nProtoId = PROTOL_LOADROLE;
  202. pNewCmd->nDataLen = strlen(szName) + 1;
  203. strcpy((char *)&pNewCmd->pDataBuffer[0], szName);
  205. EnterCriticalSection(&g_MainMsgListMutex);
  206. g_MainThreadCmdList.push_back(pNewCmd);
  207. LeaveCriticalSection(&g_MainMsgListMutex);
  209. /*
  210. //要求服务列表增加一个读档服务
  211. if (I->nNetId != 0)
  212. {
  213. unsigned char * pBuffer = new unsigned char [nBufferSize + sizeof(TCmd) - 1];
  214. TCmd * pCmd = (TCmd *)pBuffer;
  215. pCmd->nNetId = I->nNetId ;
  216. pCmd->Size = nBufferSize;
  217. pCmd->Cmd = PROTOL_LOADROLE;
  218. memcpy(&pCmd->pDataBuffer[0], szRoleInfo, )
  219. }
  220. WaitForSingleObject();//同时只能有一个线程修改网络服务列表的数据
  221. */
  223. }
  224. else
  225. LeaveCriticalSection(&g_RoleInfoMutex);
  227. // delete *I;
  228. g_DBLoadThreadCmdList.erase(I);
  230. I = g_DBLoadThreadCmdList.begin();
  231. pView->UpdateDBServiceList();
  232. }
  233. Sleep(10);
  234. }
  235. return 1;
  236. }
  238. //主循环，负责处理网络端发来的完整的请求数据
  239. DWORD  WINAPI RoleServerMainThreadFun(void * pParam)
  240. {
  241. CMainFrame * pFrame = (CMainFrame *) AfxGetMainWnd(); CRoleDBServerView *pView = (CRoleDBServerView*)pFrame->GetActiveView();
  243. while(1)
  244. {
  245. pView->UpdateMainLoopList();
  246. list<TCmd *>::iterator I = g_MainThreadCmdList.begin();
  248. while (I != g_MainThreadCmdList.end())
  249. {
  250. switch((*I)->nProtoId)
  251. {
  252. //Save
  253. case PROTOL_ONLYSAVEROLE://游戏服务器向数据库服务器要求保存数据
  254. case PROTOL_SAVEORCREATEROLE:
  255. {
  256. EnterCriticalSection(&g_RoleInfoMutex);
  257. const char * szName = GetRoleNameFromRoleBuffer((char *)&(*I)->pDataBuffer[0]);
  259. static ii = 0;
  260. char FileName[100] ;
  261. sprintf(FileName, "d:\SaveRoleout_%d.txt", ii++);
  262. TextOutRoleInfo((unsigned char *)&(*I)->pDataBuffer[0],FileName );
  264. if (szName)
  265. {
  266. map<std::string , KRoleInfomation * >::iterator II;
  267. II =  g_RoleInfoMap.find((const char *)szName) ;
  268. KRoleInfomation * pRoleInfomation = II->second;
  270. if (!pRoleInfomation)//新加角色！需要立即存档
  271. {
  272. KRoleInfomation * pNewRoleInfomation = new KRoleInfomation;
  274. pRoleInfomation->SetRoleInfoBuffer((unsigned char *)&((*I)->pDataBuffer[0]), (*I)->nDataLen);
  276. EnterCriticalSection(&g_RoleInfoMapMutex);
  278. g_RoleInfoMap[szName] = pNewRoleInfomation;
  280. EnterCriticalSection(&g_RoleInfoListMutex);
  281. g_RoleInfoList.push_back(pNewRoleInfomation);
  283. LeaveCriticalSection(&g_RoleInfoMapMutex);
  284. LeaveCriticalSection(&g_RoleInfoListMutex);
  286. }
  287. else
  288. {
  289. pRoleInfomation->SetRoleInfoBuffer((unsigned char *)&((*I)->pDataBuffer), (*I)->nDataLen);
  290. }
  291. }
  292. LeaveCriticalSection(&g_RoleInfoMutex);
  293. }
  294. break;
  296. //Load
  297. case PROTOL_LOADROLE://获得角色数据
  298. {
  299. char szName[100] ;
  300. memcpy(szName, (*I)->pDataBuffer, (*I)->nDataLen);// GetRoleNameFromRoleBuffer((char *)&(*I)->pDataBuffer);
  301. szName[(*I)->nDataLen] = 0;
  302. if (szName)
  303. {
  304. KRoleInfomation * pRoleInfomation = g_RoleInfoMap[szName];
  306. /*map<const char * , KRoleInfomation * >::iterator II;
  307. II =  g_RoleInfoMap.find((const char *)szName) ;
  308. KRoleInfomation * pRoleInfomation = II->second;
  309. */
  310. if (pRoleInfomation )
  311. {
  312. size_t BufferSize = 0;
  313. BYTE * pBuffer = pRoleInfomation->GetRoleInfoBuffer(BufferSize);
  314. _ASSERT(BufferSize && pBuffer);
  315. BYTE * pNewBuffer = new BYTE[sizeof(TCmd) + BufferSize - 1];
  316. TCmd * pNewCmd = (TCmd*)pNewBuffer;
  317. pNewCmd->nProtoId = PROTOL_ROLEINFO;
  318. pNewCmd->nDataLen = BufferSize;
  319. pNewCmd->ulNetId = (*I)->ulNetId;
  320. memcpy(pNewCmd->pDataBuffer, pBuffer, BufferSize);
  322. EnterCriticalSection(&g_NetMsgListMutex);
  323. g_NetServiceThreadCmdList.push_back(pNewCmd);
  324. LeaveCriticalSection(&g_NetMsgListMutex);
  325. }
  326. else//没有则需要实际的取数据，需要向LoadThread发送消息
  327. {
  328. TGetRoleInfoFromDB * pNewCmd = new TGetRoleInfoFromDB;
  329. pNewCmd->nNetId = (*I)->ulNetId;
  330. strcpy(pNewCmd->szRoleName, szName);
  331. EnterCriticalSection(&g_GetRoleInfoMutex);
  332. g_DBLoadThreadCmdList.push_back(pNewCmd);
  333. LeaveCriticalSection(&g_GetRoleInfoMutex);
  334. }
  335. }
  337. }break;
  339. //LoadRoleList
  340. case PROTOL_GETROLELIST://获得某个帐号的角色列表
  341. {
  342. unsigned char  MaxRoleCount = (int)(*I)->pDataBuffer[0];
  343. unsigned char * pNewBuffer = new  unsigned char [sizeof(S3DBI_RoleBaseInfo) * MaxRoleCount + 1];//列表和1个
  344. TCmd * pCmd = (TCmd*)pNewBuffer;
  345. pCmd->ulNetId = (*I)->ulNetId;
  346. pCmd->nProtoId =  (*I)->nProtoId;
  347. pCmd->nDataLen  =  sizeof(S3DBI_RoleBaseInfo) * MaxRoleCount + 1;
  349. S3DBI_RoleBaseInfo *pRoleBaseInfoList  =(S3DBI_RoleBaseInfo*) ((char*)&pCmd->pDataBuffer[0] + 1);
  350. char  szAccountName[32];
  351. memcpy(szAccountName, &(*I)->pDataBuffer[0] + 1, (*I)->nDataLen - 1 );
  352. szAccountName[(*I)->nDataLen - 1] = 0;
  354. pCmd->pDataBuffer[0]  = (unsigned char )S3DBI_GetRoleListOfAccount(szAccountName, pRoleBaseInfoList, MaxRoleCount);
  356. for (int i  = 0; i < pCmd->pDataBuffer[0]; i ++)
  357. {
  358. KRoleInfomation * pRoleInfo = g_RoleInfoMap[pRoleBaseInfoList->szRoleName];
  359. if (pRoleInfo)
  360. {
  361. size_t nSize = 0;
  362. TRoleData * pRoleData = (TRoleData*)pRoleInfo->GetRoleInfoBuffer(nSize);
  363. if (pRoleData && nSize)
  364. {
  365. pRoleBaseInfoList->nArmorType = pRoleData->BaseInfo.iarmorres;
  366. pRoleBaseInfoList->nHelmType  = pRoleData->BaseInfo.ihelmres;
  367. pRoleBaseInfoList->nLevel     = pRoleData->BaseInfo.ifightlevel;
  368. pRoleBaseInfoList->nSex   = pRoleData->BaseInfo.bSex;
  369. pRoleBaseInfoList->nWeaponType = pRoleData->BaseInfo.iweaponres;
  370. }
  371. }
  372. }
  374. EnterCriticalSection(&g_NetMsgListMutex);
  375. g_NetServiceThreadCmdList.push_back(pCmd);
  376. LeaveCriticalSection(&g_NetMsgListMutex);
  377. }
  378. break;
  379. case PROTOL_DELETEROLE:
  380. {
  382. }break;
  383. //Exit
  386. }
  388. g_MainThreadCmdList.erase(I);
  389. I = g_MainThreadCmdList.begin();
  390. pView->UpdateMainLoopList();
  391. }
  392. Sleep(20);
  394. }
  395. return 1;
  396. }
  399. //
  400. DWORD WINAPI  RoleNetWorkServiceThreadFun(void * pParam)
  401. {
  402. g_NetServer.Init();
  403. //g_NetServer.RegisterClient(1);
  404. while(1)
  405. {
  406. g_NetServer.Receive();
  407. g_NetServer.Send();
  408. Sleep(10);
  409. }
  410. }
  411. TRoleNetMsg  * pTestCmd = NULL;
  412. void * EnumSendMsg(unsigned long nId, size_t& nDataLen)
  413. {
  414. if (pTestCmd)
  415. {
  416. nDataLen = pTestCmd->nDataLen;
  417. return  pTestCmd;
  418. }
  419. else 
  420. {
  421. nDataLen = 0;
  422. return NULL;
  423. }
  425. }
  427. FILE *streamout;
  428. FILE *streamin;
  430. void SendToClient(unsigned long nId, TRoleNetMsg * pMsg, size_t nMsgLen)
  431. {
  432. static i = 0;
  433. char FileName[100] ;
  434. sprintf(FileName, "d:\testout_%d.txt", i++);
  435. if( (streamout = fopen( FileName, "wb" )) != NULL )
  436. {
  437. fwrite(pMsg, 1, nMsgLen, streamout);
  438. }
  439. fclose(streamout);
  440. }
  442. void TextOutRoleInfo(BYTE * pDataBuffer, char * FileName)
  443. {
  444. if (!pDataBuffer) return ;
  445. char showString[50000];
  446. S3DBI_ShowDebugInfo(pDataBuffer, showString);
  447. FILE * streamout;
  448. if( (streamout = fopen( FileName, "wb" )) != NULL )
  449. {
  450. fwrite(showString, 1, strlen(showString), streamout);
  451. }
  452. fclose(streamout);
  453. }
  457. typedef HRESULT ( __stdcall * pfnCreateServerInterface )(
  458. REFIID riid,
  459. void **ppv
  460. );
  462. void __stdcall ServerEventNotify(
  463. LPVOID lpParam,
  464. const unsigned long &ulnID,
  465. const unsigned long &ulnEventType )
  466. {
  467. CMainFrame * pFrame = (CMainFrame *) AfxGetMainWnd(); CRoleDBServerView *pView = (CRoleDBServerView*)pFrame->GetActiveView();
  468. switch( ulnEventType )
  469. {
  470. case enumClientConnectCreate:
  471. WaitForSingleObject(g_NetEventMutex, INFINITE);
  472. g_NetServer.RegisterClient(ulnID);
  473. ReleaseMutex(g_NetEventMutex);
  474. break;
  475. case enumClientConnectClose:
  476. WaitForSingleObject(g_NetEventMutex, INFINITE);
  477. g_NetServer.DestoryClient(ulnID);
  478. ReleaseMutex(g_NetEventMutex);
  479. break;
  480. }
  481. pView->UpdateRoleServerList();
  483. }
  485. unsigned _stdcall ThreadFunction( void *pParam )
  486. {
  487. IServer *pServer = reinterpret_cast< IServer * >( pParam );
  489. while ( pServer )
  490. {
  491. Sleep( 1 );
  493. size_t datalength = 0;
  495. const void *pData = pServer->GetPackFromClient( 4, datalength );
  497. if ( !pData || 0 == datalength )
  498. {
  499. continue;
  500. }
  502. }
  504. if ( pServer )
  505. {
  506. pServer->Release();
  507. pServer = NULL;
  508. }
  510. return 0L;
  511. }
  513. BOOL InitNetEngine()
  514. {
  516. hDllModule = ::LoadLibrary( "heaven.dll" );
  518. if ( hDllModule )
  519. {
  520. pfnCreateServerInterface pFactroyFun = ( pfnCreateServerInterface )GetProcAddress( hDllModule, "CreateInterface" );
  522. IServerFactory *pServerFactory = NULL;
  524. if ( SUCCEEDED( pFactroyFun( IID_IServerFactory, reinterpret_cast< void ** >( &pServerFactory ) ) ) )
  525. {
  526. pServerFactory->SetEnvironment( g_snMaxPlayerCount, g_snPrecision, g_snMaxBuffer, g_snBufferSize  );
  528. pServerFactory->CreateServerInterface( IID_IIOCPServer, reinterpret_cast< void ** >( &g_pNetServer ) );
  530. pServerFactory->Release();
  531. }
  532. }
  533. else
  534. return FALSE;
  536. g_pNetServer->Startup();
  538. g_pNetServer->RegisterMsgFilter( reinterpret_cast< void * >( g_pNetServer ), ServerEventNotify );
  540. g_pNetServer->OpenService( 0, 5001 );
  543. //unsigned int threadid = 0;
  544. //HANDLE handle = (HANDLE)::_beginthreadex( 0, 0, ThreadFunction, reinterpret_cast< void * >( pClonServer ), 0, &threadid );
  546. //CloseHandle( handle );
  548. /* const size_t len = strlen( "to client" );
  550. while ( pServer )
  551. {
  552. pServer->PreparePackSink();
  554. pServer->PackDataToClient( 4, "to client", len );
  556. pServer->SendPackToClient();
  558. Sleep( 100 );
  559. }
  560. */
  561. return 1;
  562. }