307 lines
No EOL
9.3 KiB
C++
307 lines
No EOL
9.3 KiB
C++
// DNDBServer.cpp : Defines the entry point for the console application.
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "IniFile.h"
|
|
#include "DNIocpManager.h"
|
|
#include "DNThreadManager.h"
|
|
#include "DNSQLConnectionManager.h"
|
|
#include "DNConfig.h"
|
|
#include "Log.h"
|
|
#include "ExceptionReport.h"
|
|
#include "DnServiceConnection.h"
|
|
#include "Version.h"
|
|
#if !defined( _FINAL_BUILD )
|
|
#include "DNQueryTest.h"
|
|
#endif // #if !defined( _FINAL_BUILD )
|
|
#include "ServiceUtil.h"
|
|
#include "DNConnectionManager.h"
|
|
|
|
#if defined(_KRAZ)
|
|
#include "DNActozShield.h"
|
|
#endif // #if defined(_KRAZ)
|
|
|
|
#include <iostream>
|
|
|
|
//#include "..//../Server/RLKT_LICENSE/license_rlkt.h"
|
|
|
|
TLogConfig g_Config;
|
|
|
|
bool LoadConfig(int argc, TCHAR * argv[])
|
|
{
|
|
memset(&g_Config, 0, sizeof(TLogConfig));
|
|
WCHAR wszBuf[128] = { 0, };
|
|
|
|
//Version
|
|
strcpy_s(g_Config.szVersion, szLogVersion);
|
|
strcpy_s(g_Config.szResVersion, "Unknown Version");
|
|
|
|
g_Config.nThreadMax = THREADMAX;
|
|
g_Config.bUseCmd = argc >= 2 ? IsUseCmd(argv[1]) : false;
|
|
|
|
if (g_Config.bUseCmd)
|
|
{
|
|
GetDefaultInfo(argv[1], g_Config.nManagedID, g_Config.szResourcePath, g_Config.szResVersion, g_Config.ServiceInfo.szIP, g_Config.ServiceInfo.nPort);
|
|
g_Log.Log(LogType::_FILELOG, L"ManagedID(SID) [%d]\n", g_Config.nManagedID);
|
|
g_Log.Log(LogType::_FILELOG, L"ResourcePath [%S] \nResourceRevision [%S]\n", g_Config.szResourcePath.c_str(), g_Config.szResVersion);
|
|
g_Log.Log(LogType::_FILELOG, L"ServiceManager [IP:%S][Port:%d]\n", g_Config.ServiceInfo.szIP, g_Config.ServiceInfo.nPort);
|
|
|
|
std::wstring wstrTempConfig;
|
|
std::string strTempConfig;
|
|
|
|
if (GetFirstRightValue(L"lgp", argv[1], wstrTempConfig))
|
|
g_Config.nAcceptPort = _wtoi(wstrTempConfig.c_str());
|
|
else _ASSERT(0);
|
|
g_Log.Log(LogType::_FILELOG, L"LogServer Port [%d]\n", g_Config.nAcceptPort);
|
|
|
|
if (GetFirstRightValue(L"title", argv[1], wstrTempConfig))
|
|
SetConsoleTitleW(wstrTempConfig.c_str());
|
|
|
|
GetLogdbInfo(argv[1], &g_Config.ServerLogDB);
|
|
g_Log.Log(LogType::_FILELOG, L"ServerLogDB Connect Info(Ip:%S, Port:%d DBID:%s DBName:%s) Success\r\n", g_Config.ServerLogDB.szIP, g_Config.ServerLogDB.nPort, g_Config.ServerLogDB.wszDBID, g_Config.ServerLogDB.wszDBName);
|
|
}
|
|
else
|
|
{
|
|
if (!g_IniFile.Open(L"./Config/DNLog.ini")){
|
|
g_Log.Log(LogType::_FILELOG, L"DNLog.ini File not Found!!\r\n");
|
|
return false;
|
|
}
|
|
|
|
g_IniFile.GetValue(L"Info", L"ThreadMax", &g_Config.nThreadMax);
|
|
g_IniFile.GetValue(L"Connection", L"AcceptPort", &g_Config.nAcceptPort);
|
|
|
|
WCHAR wszBuf[128] = { 0, };
|
|
|
|
g_IniFile.GetValue(L"MasterLogDB", L"DBIP", wszBuf);
|
|
WideCharToMultiByte(CP_ACP, NULL, wszBuf, -1, g_Config.MasterLogDB.szIP, sizeof(g_Config.MasterLogDB.szIP), NULL, NULL);
|
|
g_IniFile.GetValue(L"MasterLogDB", L"DBPort", &g_Config.MasterLogDB.nPort);
|
|
g_IniFile.GetValue(L"MasterLogDB", L"DBID", g_Config.MasterLogDB.wszDBID);
|
|
g_IniFile.GetValue(L"MasterLogDB", L"DBName", g_Config.MasterLogDB.wszDBName);
|
|
|
|
g_IniFile.GetValue(L"ServerLogDB", L"DBIP", wszBuf);
|
|
WideCharToMultiByte(CP_ACP, NULL, wszBuf, -1, g_Config.ServerLogDB.szIP, sizeof(g_Config.ServerLogDB.szIP), NULL, NULL);
|
|
g_IniFile.GetValue(L"ServerLogDB", L"DBPort", &g_Config.ServerLogDB.nPort);
|
|
g_IniFile.GetValue(L"ServerLogDB", L"DBID", g_Config.ServerLogDB.wszDBID);
|
|
g_IniFile.GetValue(L"ServerLogDB", L"DBName", g_Config.ServerLogDB.wszDBName);
|
|
|
|
g_IniFile.GetValue(L"LogDB", L"LogDBCount", &g_Config.nLogDBCount);
|
|
|
|
int nWorldID = 0;
|
|
WCHAR wszData[128] = { 0, };
|
|
for (int i = 0; i < g_Config.nLogDBCount; i++){
|
|
swprintf(wszData, L"Log%dWorldID", i + 1);
|
|
g_IniFile.GetValue(L"LogDB", wszData, &nWorldID);
|
|
if (nWorldID <= 0) continue;
|
|
|
|
g_Config.LogDB[i].nWorldSetID = nWorldID;
|
|
swprintf(wszData, L"DBIP%d", i + 1);
|
|
g_IniFile.GetValue(L"LogDB", wszData, wszBuf);
|
|
WideCharToMultiByte(CP_ACP, NULL, wszBuf, -1, g_Config.LogDB[i].szIP, sizeof(g_Config.LogDB[i].szIP), NULL, NULL);
|
|
swprintf(wszData, L"DBPort%d", i + 1);
|
|
g_IniFile.GetValue(L"LogDB", wszData, &g_Config.LogDB[i].nPort);
|
|
swprintf(wszData, L"Log%dDBID", i + 1);
|
|
g_IniFile.GetValue(L"LogDB", wszData, g_Config.LogDB[i].wszDBID);
|
|
swprintf(wszData, L"Log%dDBName", i + 1);
|
|
g_IniFile.GetValue(L"LogDB", wszData, g_Config.LogDB[i].wszDBName);
|
|
}
|
|
|
|
g_IniFile.GetValue(L"ServerManagerEx", L"sid", &g_Config.nManagedID);
|
|
g_IniFile.GetValue(L"ServerManagerEx", L"ip", wszBuf);
|
|
WideCharToMultiByte(CP_ACP, NULL, wszBuf, -1, g_Config.ServiceInfo.szIP, sizeof(g_Config.ServiceInfo.szIP), NULL, NULL);
|
|
g_IniFile.GetValue(L"ServerManagerEx", L"port", &g_Config.ServiceInfo.nPort);
|
|
}
|
|
|
|
if (g_Config.nThreadMax > THREADMAX) g_Config.nThreadMax = THREADMAX;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool CheckWorkingFolder(const WCHAR* szFolderName)
|
|
{
|
|
WCHAR szCurrentDir[MAX_PATH] = L"";
|
|
GetCurrentDirectory(MAX_PATH, szCurrentDir);
|
|
std::wstring wszCurrentDir;
|
|
wszCurrentDir = szCurrentDir;
|
|
std::transform(wszCurrentDir.begin(), wszCurrentDir.end(), wszCurrentDir.begin(), towlower);
|
|
if ( wszCurrentDir.find(szFolderName) > wszCurrentDir.size() )
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool InitApp(int argc, TCHAR * argv[])
|
|
{
|
|
//if (!CheckSerial()) return false;
|
|
|
|
#if defined(_DEBUG) && defined(_WORK)
|
|
if ( CheckWorkingFolder(L"out") == false )
|
|
{
|
|
wprintf(L"累诀 叼泛配府甫 汲沥秦林技夸!!!! setworking folder error " );
|
|
}
|
|
#endif
|
|
|
|
#if !defined( _FINAL_BUILD )
|
|
CDNQueryTest::CreateInstance();
|
|
#endif // #if !defined( _FINAL_BUILD )
|
|
|
|
WCHAR wszLogName[128];
|
|
memset(wszLogName, 0, sizeof(wszLogName));
|
|
|
|
if (argc >= 2)
|
|
{
|
|
if (IsUseCmd(argv[1]))
|
|
{
|
|
int nSID = 0;
|
|
std::wstring wstrTempArgv;
|
|
|
|
if (GetFirstRightValue(L"sid", argv[1], wstrTempArgv))
|
|
nSID = _wtoi(wstrTempArgv.c_str());
|
|
swprintf(wszLogName, L"LogServer_%d", nSID);
|
|
}
|
|
else
|
|
swprintf(wszLogName, L"LogServer");
|
|
}
|
|
else
|
|
swprintf(wszLogName, L"LogServer");
|
|
|
|
//g_Log.Init(L"LogServer", LOGTYPE_CRT_FILE_DAY);
|
|
g_Log.Init(wszLogName, LOGTYPE_CRT_FILE_DAY);
|
|
|
|
if (!LoadConfig(argc, argv)){
|
|
g_Log.Log(LogType::_FILELOG, L"LoadConfig Failed\r\n");
|
|
return false;
|
|
}
|
|
g_Log.SetServerID(g_Config.nManagedID);
|
|
|
|
g_pConnectionManager = new CDNConnectionManager;
|
|
if (!g_pConnectionManager) return false;
|
|
|
|
g_pIocpManager = new CDNIocpManager;
|
|
if (!g_pIocpManager) return false;
|
|
|
|
#if defined(_WORK)
|
|
if (g_pIocpManager->Init(500) < 0){
|
|
#else // #if defined(_WORK)
|
|
if (g_pIocpManager->Init(100) < 0){
|
|
#endif // #if defined(_WORK)
|
|
g_Log.Log(LogType::_FILELOG, L"Iocp Initialize Fail(%d)\r\n", 100);
|
|
return false;
|
|
}
|
|
else {
|
|
g_Log.Log(LogType::_FILELOG, L"## Iocp Initialize Success(%d)\r\n", 100);
|
|
}
|
|
|
|
if (!g_ThreadManager.Init(g_Config.nThreadMax)){
|
|
g_Log.Log(LogType::_FILELOG, L"ThreadManager Initialize Fail\r\n");
|
|
return false;
|
|
}
|
|
else {
|
|
g_Log.Log(LogType::_FILELOG, L"## ThreadManager Initialize Success(%d)\r\n", g_Config.nThreadMax);
|
|
}
|
|
|
|
if (!g_SQLConnectionManager.CreateDB()){
|
|
g_Log.Log(LogType::_FILELOG, L"CreateDB Fail\r\n");
|
|
return false;
|
|
}
|
|
else {
|
|
g_Log.Log(LogType::_FILELOG, L"## CreateDB OK!!\r\n");
|
|
}
|
|
|
|
if (g_pIocpManager->AddAcceptConnection(CONNECTIONKEY_DEFAULT, g_Config.nAcceptPort, 1000) < 0){
|
|
g_Log.Log(LogType::_FILELOG, L"AcceptPort Fail(%d)\r\n", g_Config.nAcceptPort);
|
|
return false;
|
|
}
|
|
else {
|
|
g_Log.Log(LogType::_FILELOG, L"## AcceptPort (%d)\r\n", g_Config.nAcceptPort);
|
|
}
|
|
g_pIocpManager->CreateUpdateThread();
|
|
|
|
// TODO(Cussrro): 优化配置项启动无法连接服务管理器
|
|
if (g_Config.nManagedID > 0)
|
|
{
|
|
g_pIocpManager->CreateThread();
|
|
g_pServiceConnection = new CDNServiceConnection(g_Config.nManagedID);
|
|
if (!g_pServiceConnection)
|
|
return false;
|
|
g_pServiceConnection->SetIp(g_Config.ServiceInfo.szIP);
|
|
g_pServiceConnection->SetPort(g_Config.ServiceInfo.nPort);
|
|
}
|
|
|
|
#if defined(_KRAZ) && defined(_FINAL_BUILD)
|
|
g_pActozShield = new CDNActozShield;
|
|
if (!g_pActozShield)
|
|
return false;
|
|
if (!g_pActozShield->Init())
|
|
return false;
|
|
#endif // #if defined(_KRAZ)
|
|
|
|
g_Log.Log(LogType::_FILEDBLOG, L"Application Initialize Success\r\n");
|
|
return true;
|
|
}
|
|
|
|
void ClearApp()
|
|
{
|
|
#if defined(_KRAZ)
|
|
SAFE_DELETE(g_pActozShield);
|
|
#endif // #if defined(_KRAZ)
|
|
|
|
#if !defined( _FINAL_BUILD )
|
|
if( CDNQueryTest::IsActive() )
|
|
CDNQueryTest::DestroyInstance();
|
|
#endif // #if !defined( _FINAL_BUILD )
|
|
|
|
if(g_pIocpManager)
|
|
{
|
|
g_pIocpManager->CloseAcceptors();
|
|
g_pIocpManager->ThreadStop();
|
|
}
|
|
SAFE_DELETE(g_pServiceConnection);
|
|
SAFE_DELETE(g_pIocpManager);
|
|
}
|
|
|
|
|
|
int _tmain(int argc, TCHAR* argv[])
|
|
{
|
|
#if defined(_CH)
|
|
setlocale(LC_ALL, "chinese-simplified");
|
|
#elif defined(_TW)
|
|
setlocale(LC_ALL, "chinese-traditional");
|
|
#elif defined(_JP)
|
|
setlocale(LC_ALL, "japanese");
|
|
#elif defined(_US)
|
|
setlocale(LC_ALL, "us");
|
|
#else
|
|
setlocale(LC_ALL, "Korean");
|
|
#endif
|
|
|
|
// 抗寇 贸府磊 霖厚
|
|
DWORD dwRetVal = CExceptionReport::GetInstancePtr()->Open(_T(".\\"), TRUE, TRUE); // Release 葛靛 哪颇老 矫 C4744 版绊啊 惯积窍咯 Singleton 备泅 函版, CExceptionReport::GetInstancePtr() 阑 inline 拳 窍瘤 臼澜 (曼绊 : http://msdn.microsoft.com/ko-kr/library/a7za416f.aspx)
|
|
if (NOERROR != dwRetVal) {
|
|
DWORD dwErrNo = ::GetLastError();
|
|
DN_RETURN(dwErrNo);
|
|
}
|
|
|
|
if (!InitApp(argc, argv)){
|
|
ClearApp();
|
|
return 0;
|
|
}
|
|
|
|
wprintf(L"exit 疙飞阑 摹搁 辆丰\r\n");
|
|
|
|
char szCmd[256] = {0};
|
|
while (1)
|
|
{
|
|
#if !defined( _FINAL_BUILD )
|
|
CDNQueryTest::GetInstance().ProcessQueryTest( szCmd );
|
|
#endif // #if !defined( _FINAL_BUILD )
|
|
|
|
if (strcmp(szCmd, "exit") == 0) break;
|
|
|
|
printf("CMD>");
|
|
std::cin >> szCmd;
|
|
}
|
|
|
|
ClearApp();
|
|
|
|
return 0;
|
|
} |