static void startServer() { gServer = std::make_shared<WrapServer>(); gServer->startWorkThread(ox_getcpunum(), [](EventLoop&){ }); /*开启对外客户端端口*/ gListenClient = std::make_shared<ListenThread>(); gListenClient->startListen(false, "127.0.0.1", portForClient, nullptr, nullptr, [&](int fd){ WrapAddNetSession(gServer, fd, std::make_shared<ConnectionClientSession>(), -1, 32 * 1024); }); /*开启对内逻辑服务器端口*/ gListenLogic = std::make_shared<ListenThread>(); gListenLogic->startListen(false, "127.0.0.1", portForLogicServer, nullptr, nullptr, [&](int fd){ WrapAddNetSession(gServer, fd, std::make_shared<LogicServerSession>(), 10000, 32 * 1024 * 1024); }); }
int main() { { srand(static_cast<unsigned int>(time(nullptr))); struct lua_State* L = nullptr; int listenPort; /*代理服务器的监听端口*/ ox_socket_init(); std::vector<std::tuple<int, string, int>> backendConfigs; try { struct msvalue_s config(true); L = luaL_newstate(); luaopen_base(L); luaL_openlibs(L); /*TODO::由启动参数指定配置路径*/ if (lua_tinker::dofile(L, "Config.lua")) { aux_readluatable_byname(L, "ProxyConfig", &config); } else { throw std::runtime_error("not found lua file"); } map<string, msvalue_s*>& allconfig = *config._map; listenPort = atoi(map_at(allconfig, string("listenPort"))->_str.c_str()); sharding_function = map_at(allconfig, string("sharding_function"))->_str; map<string, msvalue_s*>& backends = *map_at(allconfig, string("backends"))->_map; cout << "listen port:" << listenPort << endl; for (auto& v : backends) { map<string, msvalue_s*>& oneBackend = *(v.second)->_map; int id = atoi(map_at(oneBackend, string("id"))->_str.c_str()); string dbServerIP = map_at(oneBackend, string("ip"))->_str; int port = atoi(map_at(oneBackend, string("port"))->_str.c_str()); backendConfigs.push_back(std::make_tuple(id, dbServerIP, port)); cout << "backend :" << id << ", ip:" << dbServerIP << ", port:" << port << endl; } } catch (const std::exception& e) { cout << "exception:" << e.what() << endl; cin.get(); exit(-1); } //WrapLog::PTR gDailyLogger = std::make_shared<WrapLog>(); spdlog::set_level(spdlog::level::info); ox_dir_create("logs"); ox_dir_create("logs/DBProxyServer"); //gDailyLogger->setFile("", "logs/DBProxyServer/daily"); EventLoop mainLoop; WrapServer::PTR server = std::make_shared<WrapServer>(); ListenThread::PTR listenThread = std::make_shared<ListenThread>(); int netWorkerThreadNum = ox_getcpunum(); /*开启网络线程*/ server->startWorkThread(netWorkerThreadNum, nullptr); /*链接数据库服务器*/ for (auto& v : backendConfigs) { int id = std::get<0>(v); string ip = std::get<1>(v); int port = std::get<2>(v); //gDailyLogger->info("connec db server id:{}, address: {}:{}", id, ip, port); sock fd = ox_socket_connect(ip.c_str(), port); auto bserver = std::make_shared<BackendSession>(); bserver->setID(id); WrapAddNetSession(server, fd, bserver, -1, 32*1024*1024); } // gDailyLogger->info("listen proxy port:{}", listenPort); /*开启代理服务器监听*/ listenThread->startListen(listenPort, nullptr, nullptr, [&](int fd){ WrapAddNetSession(server, fd, make_shared<ClientSession>(), -1, 32 * 1024 * 1024); }); //gDailyLogger->warn("db proxy server start!"); while (true) { if (app_kbhit()) { break; } mainLoop.loop(1); } listenThread->closeListenThread(); server->getService()->closeService(); lua_close(L); L = nullptr; } return 0; }