401 lines
9.4 KiB
C++
401 lines
9.4 KiB
C++
#include "StdAfx.h"
|
||
#ifdef WIN32
|
||
#include <dbghelp.h>
|
||
#include <mbctype.h>
|
||
#include <conio.h>
|
||
#else
|
||
// #include "client/linux/handler/exception_handler.h"
|
||
#endif
|
||
#include <string>
|
||
#include <iostream>
|
||
#include "PathThreadLocale.h"
|
||
|
||
bool DBEngineRunning = true ;
|
||
//定义转储文件名称
|
||
extern const TCHAR szExceptionDumpFile[] = _T("./DBEngine.dmp");
|
||
BOOL SetupDBEngineConfig(CDBServer *lpDBEngine, const char *pszConfigFileName = NULL);
|
||
VOID ServerMain(int argc, char **argv);
|
||
VOID ServiceRun(int argc, char** argv);
|
||
|
||
//数据引擎版本号
|
||
#define DBEGN_KN_VERION MAKEFOURCC(17, 3, 1, 1)
|
||
//数据引擎的数据结构版本
|
||
|
||
//#define DBSDATATYPE_VERSION 0x010A1B0A
|
||
#ifndef WIN32
|
||
/*
|
||
static bool dumpCallback(const google_breakpad::MinidumpDescriptor& descriptor,
|
||
void* context,
|
||
bool succeeded)
|
||
{
|
||
printf("Dump path: %s\n", descriptor.path());
|
||
return succeeded;
|
||
}
|
||
|
||
*/
|
||
static void signal_handler(int sig_num)
|
||
{
|
||
if( sig_num == SIGHUP ||
|
||
sig_num == SIGINT ||
|
||
sig_num == SIGTERM )
|
||
{
|
||
OutputMsg( rmTip, _T("正在停止网关管理器...") );
|
||
DBEngineRunning = false;
|
||
}
|
||
}
|
||
#endif
|
||
|
||
#ifndef WIN32
|
||
int _kbhit(void)
|
||
{
|
||
struct termios oldt, newt;
|
||
int ch;
|
||
int oldf;
|
||
tcgetattr(STDIN_FILENO, &oldt);
|
||
newt = oldt;
|
||
newt.c_lflag &= ~(ICANON | ECHO);
|
||
tcsetattr(STDIN_FILENO, TCSANOW, &newt);
|
||
oldf = fcntl(STDIN_FILENO, F_GETFL, 0);
|
||
fcntl(STDIN_FILENO, F_SETFL, oldf | O_NONBLOCK);
|
||
ch = getchar();
|
||
tcsetattr(STDIN_FILENO, TCSANOW, &oldt);
|
||
fcntl(STDIN_FILENO, F_SETFL, oldf);
|
||
if(ch != EOF)
|
||
{
|
||
ungetc(ch, stdin);
|
||
return 1;
|
||
}
|
||
return 0;
|
||
}
|
||
#endif
|
||
|
||
#ifdef WIN32
|
||
int initLocalePath()
|
||
{
|
||
//_MB_CP_SBCS 视所有字符均作为单字节字符，以便同时支持UTF8以及MBCS
|
||
//Use single-byte code page.
|
||
//When the code page is set to _MB_CP_SBCS, a routine such as _ismbblead always returns false.
|
||
_setmbcp(_MB_CP_SBCS);
|
||
//★关键★ 设置"C"locale，视所有字符均作为单字节字符，以便同时支持UTF8以及MBCS
|
||
return InstallThreadLocalePath("C");
|
||
}
|
||
#endif
|
||
|
||
void TestMemory()
|
||
{
|
||
#ifdef _DEBUG
|
||
|
||
#else
|
||
/*
|
||
DWORD dwStart = GetTickCount();
|
||
for(int i = 0; i < 1000000; i++) { char *p = (char*)malloc(i%1000000); free(p); }
|
||
|
||
DWORD dwEnd = GetTickCount();
|
||
printf("***********************",dwEnd - dwStart);
|
||
printf("100wMemoryTest:%d ms\n",dwEnd - dwStart);
|
||
*/
|
||
|
||
#endif
|
||
}
|
||
|
||
#ifdef WIN32
|
||
class CDbService : public CWinService
|
||
{
|
||
public:
|
||
CDbService(LPCSTR pServiceName, DWORD dwServiceType = SERVICE_WIN32_OWN_PROCESS):CWinService(pServiceName, dwServiceType){};
|
||
int AppMain()
|
||
{
|
||
SetCurrentDirectory("../"); //CWinService里会把当前目录改成exe允许的目录，所以这里再转一下
|
||
CFileLogger flog(_T("log/DBServer_%s.log"), getCurrentTimeDesc());
|
||
CDBServer *pDBEngine = new CDBServer();
|
||
|
||
if ( !SetupDBEngineConfig(pDBEngine) )
|
||
{
|
||
OutputMsg( rmError, _T("读入配置文件失败，服务停止！"));
|
||
}
|
||
else if ( pDBEngine->Startup() )
|
||
{
|
||
while (!m_boServiceExit)
|
||
{
|
||
Sleep(1000);
|
||
}
|
||
pDBEngine->Shutdown();
|
||
}
|
||
|
||
delete pDBEngine;
|
||
|
||
return 0;
|
||
}
|
||
};
|
||
#endif
|
||
|
||
int main(int argc, TCHAR* argv[])
|
||
{
|
||
#ifdef WIN32
|
||
SetUnhandledExceptionFilter( DefaultUnHandleExceptionFilter );
|
||
#else
|
||
//seh_init();
|
||
/*
|
||
google_breakpad::MinidumpDescriptor descriptor(".");
|
||
google_breakpad::ExceptionHandler eh(descriptor,
|
||
NULL,
|
||
dumpCallback,
|
||
NULL,
|
||
true,
|
||
-1);
|
||
*/
|
||
(void)signal(SIGHUP, SIG_IGN);
|
||
(void)signal(SIGINT, signal_handler);
|
||
(void)signal(SIGTERM, signal_handler);
|
||
(void)signal(SIGPIPE, SIG_IGN);
|
||
#endif
|
||
InitDefMsgOut();
|
||
#ifdef WIN32
|
||
CoInitialize(NULL);
|
||
#endif
|
||
CTimeProfDummy::SetOpenFlag(false);
|
||
|
||
|
||
|
||
#ifdef WIN32
|
||
//flog.SetNeedOutput(true);
|
||
#else
|
||
if (argc >=2)
|
||
{
|
||
// flog.SetNeedOutput(true);
|
||
}
|
||
else
|
||
{
|
||
// flog.SetNeedOutput(false);
|
||
}
|
||
#endif
|
||
|
||
#ifdef WIN32
|
||
if (initLocalePath())
|
||
{
|
||
OutputMsg( rmError, _T("can not set locale path") );
|
||
getc(stdin);
|
||
return -2;
|
||
}
|
||
#endif
|
||
CTimeProfDummy::SetOpenFlag(false);
|
||
#ifndef _SERVICE
|
||
//if (argc == 2 && _tcsncicmp("/cmd",argv[1],4)==0)//平时调试用
|
||
{
|
||
#ifdef WIN32
|
||
SetCurrentDirectory("../");
|
||
#else
|
||
std::string filename(argv[1]);
|
||
size_t found = filename.find_last_of("/\\");
|
||
filename = filename.substr(0, found);
|
||
if( filename[0] == '.' && filename.length()==1 )
|
||
filename = "./" ;
|
||
SetCurrentDirectory(filename.c_str());
|
||
#endif
|
||
ServerMain(argc, argv);
|
||
}
|
||
#else
|
||
//else
|
||
{
|
||
ServiceRun(argc, argv);
|
||
}
|
||
#endif
|
||
|
||
#ifdef WIN32
|
||
CoUninitialize();
|
||
#endif
|
||
UninitDefMsgOut();
|
||
#ifdef _MLIB_DUMP_MEMORY_LEAKS_
|
||
_CrtDumpMemoryLeaks();
|
||
#endif
|
||
return 0;
|
||
}
|
||
|
||
#ifdef WIN32
|
||
VOID ServiceRun(int argc, char** argv)
|
||
{
|
||
LPTSTR sCmd = NULL;
|
||
if (argc >= 2)
|
||
{
|
||
sCmd = argv[1];
|
||
}
|
||
TestMemory();
|
||
CDbService service("DbService");
|
||
service.Run(sCmd);
|
||
}
|
||
|
||
BOOL WINAPI CustomConsoleCtrlHandler(DWORD dwCtrlType)
|
||
{
|
||
if (dwCtrlType == CTRL_CLOSE_EVENT)
|
||
{
|
||
Beep( 600, 200 );
|
||
printf( "Ctrl-Close event\n\n" );
|
||
return TRUE;
|
||
}
|
||
else
|
||
return FALSE;
|
||
}
|
||
#endif
|
||
|
||
|
||
VOID ServerMain(int argc, char **argv)
|
||
{
|
||
const char *pszConfigFileName = NULL;
|
||
if (argc >= 2)
|
||
{
|
||
pszConfigFileName = argv[1];
|
||
}else
|
||
{
|
||
pszConfigFileName = "DBServerLinux.txt";
|
||
}
|
||
|
||
int nError;
|
||
TCHAR sCmdBuf[512];
|
||
CFileLogger flog(_T("./log/DBServer_%s.log"), getCurrentTimeDesc());
|
||
CDBServer *pDBEngine = new CDBServer();
|
||
CDBServer::s_pDBEngine = pDBEngine;
|
||
//SetConsoleCtrlHandler(CustomConsoleCtrlHandler, TRUE);
|
||
//DeleteMenu(GetSystemMenu(GetConsoleWindow(), FALSE), SC_CLOSE, MF_BYCOMMAND);
|
||
#ifdef WIN32
|
||
if (initLocalePath())
|
||
{
|
||
OutputMsg( rmError, _T("can not set locale path") );
|
||
getc(stdin);
|
||
return ;
|
||
}
|
||
#endif
|
||
|
||
|
||
if ( !SetupDBEngineConfig(pDBEngine,pszConfigFileName) )
|
||
{
|
||
OutputMsg( rmError, _T("读入配置文件失败，服务停止！"));
|
||
}
|
||
else if ( pDBEngine->Startup() )
|
||
{
|
||
//设置窗口标题
|
||
CTimeProfMgr::getSingleton().InitMgr();
|
||
|
||
char *pBuff = (char*)malloc(256);
|
||
in_addr ia;
|
||
ia.s_addr = DBEGN_KN_VERION;
|
||
strcpy(pBuff, pDBEngine->getServerName());
|
||
strcat(pBuff,"-V");
|
||
strcat(pBuff,inet_ntoa(ia));
|
||
SetConsoleTitle(pBuff);
|
||
free(pBuff);
|
||
pBuff =NULL;
|
||
|
||
TestMemory();
|
||
|
||
OutputMsg( rmTip, _T("-------------------------------------------") );
|
||
OutputMsg( rmTip, _T("数据服务器启动成功，核心版本号是%s"), inet_ntoa(ia) );
|
||
|
||
//OutputMsg( rmTip, _T("数据结构版本号是：%X"), DBSDATATYPE_VERSION );
|
||
// OutputMsg( rmTip, _T("角色基础数据结构大小是：%d"), sizeof(DBCHARBASICDATA) );
|
||
OutputMsg( rmTip, _T("quit命令可以停止服务并退出程序") );
|
||
OutputMsg( rmTip, _T("lgr 命令可以重新加载游戏网关路由表") );
|
||
OutputMsg( rmTip, _T("-------------------------------------------") );
|
||
unsigned long long startTick = _getTickCount();
|
||
while ( pDBEngine->GetDbStartFlag() )
|
||
{
|
||
#ifdef _MSC_VER
|
||
sCmdBuf[0] = 0;
|
||
|
||
if(_kbhit())
|
||
{
|
||
//_getts(sCmdBuf);
|
||
std::cin >> (sCmdBuf);
|
||
}
|
||
else
|
||
{
|
||
Sleep(100);
|
||
continue;
|
||
}
|
||
|
||
//重新加载路由命令
|
||
if ( _tcsncicmp(sCmdBuf, _T("lgr"), 3) == 0 )
|
||
{
|
||
nError = pDBEngine->LoadGameServerRoute();
|
||
if ( nError >= 0 )
|
||
OutputMsg( rmTip, _T("已加载%d个游戏网关路由数据"), nError);
|
||
else OutputMsg( rmError, _T("加载游戏网关路由数据失败"));
|
||
continue;
|
||
}
|
||
//退出命令
|
||
if ( _tcsncicmp(sCmdBuf, _T("\\q"), 2) == 0
|
||
|| _tcsncicmp(sCmdBuf, _T("exit"), 4) == 0
|
||
|| _tcsncicmp(sCmdBuf, _T("quit"), 4) == 0 )
|
||
{
|
||
OutputMsg( rmTip, _T("正在停止网关管理器...") );
|
||
break;
|
||
}
|
||
else if ( _tcsncicmp(sCmdBuf, _T("spf"), 3) ==0)
|
||
{
|
||
CTimeProfMgr::getSingleton().dump();
|
||
pDBEngine->TraceGameServerRoute();
|
||
}
|
||
// asi interval (interval:save time interval, s)
|
||
else if (_tcsncicmp(sCmdBuf, _T("asi"), 3) == 0)
|
||
{
|
||
char *pParam = sCmdBuf + 3;
|
||
while (*pParam == ' ')
|
||
pParam++;
|
||
if (pParam)
|
||
{
|
||
UINT_PTR nInterval = atoi(pParam);
|
||
if (nInterval > 0)
|
||
CDBDataCache::s_nActorCacheSaveInterval = nInterval*1000;
|
||
}
|
||
}
|
||
else if(_tcsncicmp(sCmdBuf,"dmp",3) ==0 )
|
||
{
|
||
DebugBreak();
|
||
}
|
||
else if(_tcsncicmp(sCmdBuf,"opentrace",9) ==0 )
|
||
{
|
||
CTimeProfDummy::SetOpenFlag(true);
|
||
}
|
||
else if(_tcsncicmp(sCmdBuf,"closetrace",10) ==0 )
|
||
{
|
||
CTimeProfDummy::SetOpenFlag(false);
|
||
}
|
||
else if(_tcsncicmp(sCmdBuf,"memory",6) ==0 )
|
||
{
|
||
pDBEngine->Trace();
|
||
}
|
||
#else
|
||
if (!DBEngineRunning) break;
|
||
#endif
|
||
Sleep(10);
|
||
}
|
||
|
||
//pDBEngine->Shutdown();
|
||
}
|
||
|
||
delete pDBEngine;
|
||
}
|
||
|
||
|
||
|
||
BOOL SetupDBEngineConfig(CDBServer *lpDBEngine, const char *pszConfigFileName)
|
||
{
|
||
CDBConfig config;
|
||
if ( !config.ReadConfig(pszConfigFileName) )
|
||
return FALSE;
|
||
|
||
lpDBEngine->RunEsqlFile(config.m_EsqlToolPath,config.DbConf.szDbName);
|
||
lpDBEngine->SetServerName(config.ServerName);
|
||
lpDBEngine->SetServerIndex(config.ServerIndex);
|
||
lpDBEngine->SetGateServiceAddress(config.GateAddr.szAddr, config.GateAddr.nPort);
|
||
//lpDBEngine->SetSessionServerAddress(config.SessionAddr.szAddr, config.SessionAddr.nPort);
|
||
lpDBEngine->SetNameSyncServerAddress(config.NameAddr.szAddr, config.NameAddr.nPort);
|
||
lpDBEngine->SetDataServiceAddress(config.DataAddr.szAddr, config.DataAddr.nPort);
|
||
lpDBEngine->SetSQLConfig(config.DbConf.szHost, config.DbConf.nPort, config.DbConf.szDbName, config.DbConf.szUser, config.DbConf.szPassWord,config.DbConf.boUtf8 > 0 ? true : false);
|
||
|
||
//启动日志客户端
|
||
lpDBEngine->SetLogServerAddress(config.LogAddr.szAddr, config.LogAddr.nPort);
|
||
//lpDBEngine->SetDBCenterAddress(config.DBCenterAddr.szAddr, config.DBCenterAddr.nPort);
|
||
lpDBEngine->LoadNameGroup(config.m_sPrefixList, /*config.m_sConnectList,*/ config.m_sStuffixList);
|
||
return TRUE;
|
||
}
|