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

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

模拟服务器

开发平台：

C/C++

IDBRoleServer.cpp：源码内容
  1. #include "stdafx.h"
  2. #include "IDBRoleServer.h"
  3. #include "DBTable.h"
  4. #include "DBBackup.h"
  5. #include "../../../../Headers/KGmProtocol.h"
  6. #include "CRC32.h"
  8. #include <iostream>
  9. #include <strstream>
  11. using namespace std;
  13. #include "Macro.h"
  15. static ZDBTable *db_table = NULL;
  16. static size_t nMaxRoleCount_InAccount = 3;
  17. static CDBBackup::TStatData GameStatData;//游戏统计数据（服务器初始化时填入）
  18. static CDBBackup* DBBackup = NULL;
  20. static HANDLE hDeadLockThread = NULL; //检查数据库有没有死锁的线程
  21. static HANDLE hRemoveLogThread = NULL; //删除没用日志文件的线程
  23. HWND hListOutput = NULL; //输出list
  25. int get_account(DB *db, const DBT *pkey, const DBT *pdata, DBT *ikey) 
  26. {
  27. //给定一个完整的buffer，得到account作为索引
  28. memset( ikey, 0, sizeof( DBT ) );
  29. TRoleData *pRoleData = (TRoleData *)pdata->data;
  31. ikey->data = pRoleData->BaseInfo.caccname;
  32. ikey->size = strlen( pRoleData->BaseInfo.caccname ) + 1;
  34. return 0;
  35. }
  37. //==========检查数据库有没有死锁的线程 Add By Fellow 2003.9.10==============
  38. DWORD WINAPI DeadlockProc(LPVOID lpParameter) {
  39. while(!db_table->bStop) {
  40. Sleep(5 * 1000); //5秒钟一次
  41. db_table->deadlock();
  42. }
  43. return 0;
  44. }
  45. //==========删除没用日志文件的线程 Add By Fellow 2003.9.10==============
  46. DWORD WINAPI RemoveLogProc(LPVOID lpParameter) {
  47. while(!db_table->bStop) {
  48. db_table->removeLog();
  49. Sleep(60 * 60 * 1000); //1小时一次
  50. }
  51. return 0;
  52. }
  54. BOOL InitDBInterface( size_t nMaxRoleCount )
  55. {
  56. nMaxRoleCount_InAccount = nMaxRoleCount;
  58. db_table = new ZDBTable( "database", "roledb" );
  60. db_table->addIndex( get_account );
  61. if ( db_table->open() )
  62. {
  63. DWORD dwThreadId, dwThrdParam = 1;
  64. /*
  66. hDeadLockThread = CreateThread(
  67. NULL, // no security attributes 
  68. 0, // use default stack size  
  69. DeadlockProc, // thread function 
  70. &dwThrdParam, // argument to thread function 
  71. 0, // use default creation flags 
  72. &dwThreadId); // returns the thread identifier 
  73. if(!hDeadLockThread)
  74. {
  75. //创建线程失败,暂时没有处理
  76. }
  77. */
  79. hRemoveLogThread = CreateThread(
  80. NULL, // no security attributes 
  81. 0, // use default stack size  
  82. RemoveLogProc, // thread function 
  83. &dwThrdParam, // argument to thread function 
  84. 0, // use default creation flags 
  85. &dwThreadId); // returns the thread identifier 
  86. if(!hRemoveLogThread)
  87. {
  88. //创建线程失败,暂时没有处理
  89. }
  91. return TRUE;
  92. }
  94. return FALSE;
  95. }
  97. void ReleaseDBInterface() //释放数据库引擎
  98. {
  99. if ( db_table )
  100. {
  101. db_table->commit();
  102. /*
  104. if(!hDeadLockThread)
  105. TerminateThread(hDeadLockThread, 0);
  106. */
  108. if(!hRemoveLogThread)
  109. TerminateThread(hRemoveLogThread, 0);
  111. db_table->removeLog();
  113. StopBackupTimer();//停止备份线程
  115. db_table->close();
  116. delete db_table;
  117. }
  118. }
  120. //[wxb 2003-7-23]
  121. void SetRoleInfoForGM(int nInfoID, char* pRoleBuffer, char* strUser, int nBufLen)
  122. {
  123. char* aBuffer = new char[64 * 1024];
  124. TRoleData* pRoleData = NULL;
  125. int size;
  127. GetRoleInfo(aBuffer, strUser, size);
  128. if(size)
  129. {
  130. pRoleData = (TRoleData*)aBuffer;
  131. switch (nInfoID)
  132. {
  133. case gm_role_entergame_position:
  134. ASSERT(nBufLen == sizeof(GM_ROLE_DATA_SUB_ENTER_POS));
  135. memcpy(&(pRoleData->BaseInfo.ientergameid), pRoleBuffer, sizeof(GM_ROLE_DATA_SUB_ENTER_POS));
  136. break;
  137. default:
  138. ASSERT(0);
  139. return;
  140. break;
  141. }
  142. SaveRoleInfo(aBuffer, strUser, false);
  143. }
  145.     delete aBuffer;
  146. }
  148. //[wxb 2003-7-22]
  149. void *GetRoleInfoForGM(int nInfoID, char * pRoleBuffer, char * strUser, int &nBufLen)
  150. {
  151. int size = 0;
  152. ZCursor *cursor = db_table->search( strUser, strlen( strUser ) + 1 );
  153. // char *buffer = db_table->search( strUser, strlen( strUser ) + 1, size );
  154. if ( cursor )
  155. {
  156. TRoleData* pRole = (TRoleData*)cursor->data;
  157. switch(nInfoID) {
  158. case gm_role_entergame_position:
  159. nBufLen = sizeof(pRole->BaseInfo.ientergameid) + sizeof(pRole->BaseInfo.ientergamex) + sizeof(pRole->BaseInfo.ientergamey);
  160. memcpy( pRoleBuffer, &(pRole->BaseInfo.ientergameid), nBufLen );
  161. break;
  162. default:
  163. ASSERT(0);
  164. nBufLen = 0;
  165. break;
  166. }
  168. db_table->closeCursor(cursor);
  169. }
  170. else
  171. {
  172. nBufLen = 0;
  173. }
  175. return pRoleBuffer;
  176. }
  178. void *GetRoleInfo( char * pRoleBuffer, char * strUser, int &nBufLen )
  179. {
  180. //输出数据======
  181. char aStr[1024];
  182. sprintf(aStr,"GetRoleInfo:%s",strUser);
  183. AddOutputString(hListOutput,aStr);
  184. //===============
  185. int size = 0;
  186. ZCursor *cursor = db_table->search( strUser, strlen( strUser ) + 1 );
  187. // char *buffer = db_table->search( strUser, strlen( strUser ) + 1, size );
  188. if ( cursor )
  189. {
  190. nBufLen = size;
  191. memcpy( pRoleBuffer, cursor->data, cursor->size );
  192. nBufLen = cursor->size;
  193. db_table->closeCursor(cursor);
  194. }
  195. else
  196. {
  197. nBufLen = 0;
  198. }
  200. return pRoleBuffer;
  201. }
  203. //保存角色的信息，如果数据库不存在该玩家，则增加该玩家
  204. //bAutoInsertWhenNoExistUser 设为TRUE时表示，如果需要保存的该玩家在数据库中并不存在则自动加入到数据库中，FALSE则不增加直接返回错误
  205. //注意INI文件只须存放将需要改动的数据，不需改动的数据将自动保存原状。
  206. int SaveRoleInfo( char * pRoleBuffer, const char *strUser, BOOL bAutoInsertWhenNoExistUser )
  207. {
  208. ASSERT( pRoleBuffer );
  210. //需要存放帐号首先找到数据
  211. TRoleData *pRoleData = ( TRoleData * )pRoleBuffer;
  213. //输出数据======
  214. char aStr[1024];
  215. sprintf(aStr,"SaveRoleInfo:%s dwDataLen=%d",pRoleData->BaseInfo.szName,pRoleData->dwDataLen);
  216. AddOutputString(hListOutput,aStr);
  217. //===============
  219. if(pRoleData->dwDataLen >= 64 * 1024) return 0;//如果数据大于64K就不添加到数据库
  221. if(bAutoInsertWhenNoExistUser)
  222. {//如果是新增角色就把账号名转成小写
  223. char *ptr = pRoleData->BaseInfo.caccname; //把账号名转成小写
  224. while(*ptr) {
  225. if(*ptr >= 'A' && *ptr <= 'Z') *ptr += 'a' - 'A';
  226. ptr++;
  227. }
  228. }
  230. if (!bAutoInsertWhenNoExistUser)
  231. {
  232. DWORD dwCRC = 0;
  233. dwCRC = CRC32(dwCRC, pRoleData, pRoleData->dwDataLen - 4);
  234. DWORD dwOrigCRC = *(DWORD *)(pRoleBuffer + pRoleData->dwDataLen - 4);
  235. if (dwCRC != dwOrigCRC)
  236. {
  237. // TODO:
  238. FILE *fLog = fopen("crc_error", "a+b");
  239. if(fLog) {
  240. char buffer[255];
  241. sprintf(buffer, "----rn%srb%srn", pRoleData->BaseInfo.szName, pRoleData->BaseInfo.caccname);
  242. fwrite(buffer, 1, strlen(buffer), fLog);
  243. fwrite(pRoleBuffer, 1, pRoleData->dwDataLen, fLog);
  244. fclose(fLog);
  245. }
  246. return 0;
  247. }
  248. }
  250. char szAccountName[32];
  252. int nLength = strlen( pRoleData->BaseInfo.caccname );
  254. ASSERT( nLength > 0 );
  256. nLength = nLength > 31 ? 31 : nLength;
  258. memcpy( szAccountName, pRoleData->BaseInfo.caccname, nLength );
  259. szAccountName[nLength] = '';
  261. char szName[32];
  262. const char *pName = szName;
  264. if ( NULL == strUser || strUser[0] == 0 )
  265. {
  266. int len = strlen( pRoleData->BaseInfo.szName );
  268. ASSERT( len > 0 );
  270. len = len > 31 ? 31 : len;
  272. memcpy( szName, pRoleData->BaseInfo.szName, len );
  273. szName[len] = '';
  274. }
  275. else
  276. {
  277. int len = strlen( strUser );
  279. ASSERT( len > 0 );
  281. memcpy( szName, strUser, len );
  282. szName[len] = '';
  283. }
  285. if ( bAutoInsertWhenNoExistUser )
  286. {
  287. int nCount = 0;
  288. int size = 0;
  290. /*
  291.  * Role of same name only
  292.  */
  293. ZCursor *user_cursor = db_table->search(pName, strlen( pName ) + 1 );
  294. // char *user_data = db_table->search( pName, strlen( pName ) + 1, size );
  295. if ( user_cursor )
  296. {
  297. db_table->closeCursor(user_cursor);
  299. return 0;
  300. }
  302. /*
  303.  * Get count of role by the key of account
  304.  */
  305. ZCursor *cursor = db_table->search( szAccountName, strlen( szAccountName ) + 1, 0 );
  306. // char *buffer = db_table->search( szAccountName, strlen( szAccountName ) + 1, size, 0 );
  308. while ( cursor )
  309. {
  310. nCount ++;
  311. if(!db_table->next( cursor ))break;
  312. }
  314. if ( nCount >= nMaxRoleCount_InAccount )
  315. {
  316. return 0;
  317. }
  318. }
  320. if ( db_table->add( pName, strlen( pName ) + 1, pRoleBuffer, pRoleData->dwDataLen ) )
  321. {
  322. return 1;
  323. }
  325. return 0;
  326. }
  328. int GetRoleListOfAccount( char * szAccountName, S3DBI_RoleBaseInfo * RoleBaseList, int nMaxCount )
  329. {
  330. int size = 0;
  331. int count = 0;
  333. char *ptr = szAccountName; //把账号名转成小写
  334. while(*ptr) {
  335. if(*ptr >= 'A' && *ptr <= 'Z') *ptr += 'a' - 'A';
  336. ptr++;
  337. }
  339. //输出数据======
  340. char aStr[1024];
  341. sprintf(aStr,"GetRoleListOfAccount:%s",szAccountName);
  342. AddOutputString(hListOutput,aStr);
  343. //===============
  345. S3DBI_RoleBaseInfo *base_ptr = RoleBaseList;
  347. ZCursor *cursor = db_table->search( szAccountName, strlen( szAccountName ) + 1, 0 );
  348. // char *buffer = db_table->search( szAccountName, strlen( szAccountName ) + 1, size, 0 );
  350. while ( count < nMaxCount && cursor )
  351. {
  352. TRoleData *pRoleData = (TRoleData *)cursor->data;
  354. strncpy( base_ptr->szName, pRoleData->BaseInfo.szName, 32 ); 
  355.         base_ptr->szName[31] = '';
  357. base_ptr->Sex = pRoleData->BaseInfo.bSex;
  358. base_ptr->Series = pRoleData->BaseInfo.ifiveprop;
  359. // base_ptr->HelmType = pRoleData->BaseInfo.ihelmres;
  360. // base_ptr->ArmorType = pRoleData->BaseInfo.iarmorres;
  361. // base_ptr->WeaponType = pRoleData->BaseInfo.iweaponres;
  362. base_ptr->Level = pRoleData->BaseInfo.ifightlevel;
  364. base_ptr++;
  366. /*
  367.  * Get next info from database
  368.  */
  369. count++;
  370. if(!db_table->next( cursor ))break;
  371. }
  372. return count;
  373. }
  375. bool DeleteRole( const char * strUser )
  376. {
  377. if ( db_table && strUser && strUser[0] )
  378. {
  379. return db_table->remove( strUser, strlen( strUser ) + 1 );
  380. }
  381. return false;
  382. }
  384. // 下面的函数可能存在内存泄漏问题
  385. //char* GetAccountByUser(char * strUser)
  386. //{//通过用户名查找帐户
  387. // char* aBuffer = new char[64 * 1024];
  388. // TRoleData* pRoleData;
  389. // int size;
  390. //
  391. // GetRoleInfo(aBuffer, strUser, size);
  392. // if(size)
  393. // {
  394. // pRoleData = (TRoleData*)aBuffer;
  395. // return pRoleData->BaseInfo.caccname;
  396. // }
  397. // else
  398. // {
  399. // return NULL;
  400. // }
  401. //}
  403. //------------------------------------------------------------------------
  404. //数据库备份与数据统计 Add By Fellow At 2003.08.14
  405. bool StartBackupTimer(int aTime)
  406. {//开始运行备份线程
  407. DBBackup = new CDBBackup( "database", "roledb", db_table );
  408. DBBackup->SaveStatInfo();
  409. bool aStartResult = DBBackup->Open(aTime);
  410. return aStartResult;
  411. }
  413. bool StopBackupTimer()
  414. {//结束运行备份线程
  415. if(!DBBackup)return false;
  416. while(IsBackupWorking()){}//等待备份线程结束
  418. bool aResult = DBBackup->Close();
  419. delete DBBackup;
  420. DBBackup = NULL;
  421. return aResult;
  422. }
  424. bool SuspendBackupTimer()
  425. {//挂起线程
  426. if(!DBBackup)return false;
  427. return DBBackup->Suspend();
  428. }
  430. bool ResumeBackupTimer()
  431. {//继续运行线程
  432. if(!DBBackup)return false;
  433. return DBBackup->Resume();
  434. }
  436. bool IsBackupThreadWorking()
  437. {//线程是否正在运行
  438. if(!DBBackup)return false;
  439. return DBBackup->IsWorking();
  440. }
  442. bool IsBackupWorking()
  443. {//是否在备份
  444. if(!DBBackup)return false;
  445. return DBBackup->IsBackuping();
  446. }
  448. bool DoManualBackup()
  449. {//手工备份
  450. if(!DBBackup)return false;
  451. return DBBackup->ManualBackup();
  452. }
  454. bool GetGameStat(TGAME_STAT_DATA* aStatData)
  455. {
  456. if(!DBBackup)return false;
  457. TGAME_STAT_DATA tmpStatData = DBBackup->GetSendStatData();
  458. memcpy(aStatData, &tmpStatData, sizeof(TGAME_STAT_DATA));
  459. return true;
  460. }
  462. void AddOutputString(HWND hListCtrl, char* aStr)
  463. {//添加操作输出文字
  464. if ( hListCtrl && ::IsWindow( hListCtrl ) )
  465. {
  466. int nCount = ::SendMessage( hListCtrl, LB_GETCOUNT, 0, 0 );
  467. if(nCount >= 100)
  468. {
  469. ::SendMessage( hListCtrl, LB_DELETESTRING, 100, 0 );
  470. }
  471. int nIndex = ::SendMessage( hListCtrl, LB_INSERTSTRING, 0, ( LPARAM )aStr );//把新的信息添到第一个
  472. }
  473. }