73 lines
2.0 KiB
C++
73 lines
2.0 KiB
C++
#include "precompile.h"
|
|
|
|
#include "target_conn_mgr.h"
|
|
#include "target_conn.h"
|
|
#include "jsondatamgr.h"
|
|
#include "app.h"
|
|
|
|
void TargetConnMgr::Init()
|
|
{
|
|
#if MASTER_MODE
|
|
#else
|
|
auto target_server_cluster_conf = JsonDataMgr::Instance()->GetTargetServerClusterConf();
|
|
for (int i = 0; i < target_server_cluster_conf->Size(); ++i) {
|
|
auto target_server_conf = target_server_cluster_conf->At(i);
|
|
int instance_id = target_server_conf->At("instance_id")->AsXValue();
|
|
std::string remote_ip = target_server_conf->At("ip")->AsXValue();
|
|
int remote_port = target_server_conf->At("port")->AsXValue();
|
|
if (App::Instance()->HasFlag(3)) {
|
|
remote_ip = "127.0.0.1";
|
|
}
|
|
{
|
|
TargetConn* conn = new TargetConn();
|
|
conn->Init(instance_id, remote_ip, remote_port);
|
|
id_hash_[conn->instance_id] = conn;
|
|
conn->Open();
|
|
}
|
|
if (App::Instance()->HasFlag(3)) {
|
|
break;
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void TargetConnMgr::UnInit()
|
|
{
|
|
for (auto& pair : id_hash_) {
|
|
pair.second->UnInit();
|
|
delete pair.second;
|
|
}
|
|
}
|
|
|
|
TargetConn* TargetConnMgr::GetConnByKey(const std::string& key)
|
|
{
|
|
auto itr = key_hash_.find(key);
|
|
return itr != key_hash_.end() ? itr->second : nullptr;
|
|
}
|
|
|
|
TargetConn* TargetConnMgr::GetConnById(int instance_id)
|
|
{
|
|
auto itr = id_hash_.find(instance_id);
|
|
return itr != id_hash_.end() ? itr->second : nullptr;
|
|
}
|
|
|
|
TargetConn* TargetConnMgr::RecreateTargetConn(const std::string& host, int port)
|
|
{
|
|
std::string key = host + ":" + a8::XValue(port).GetString();
|
|
TargetConn* conn = GetConnByKey(key);
|
|
if (conn) {
|
|
return conn;
|
|
}
|
|
while (GetConnById(++curr_id_)) {};
|
|
int instance_id = curr_id_;
|
|
std::string remote_ip = host;
|
|
int remote_port = port;
|
|
|
|
conn = new TargetConn();
|
|
conn->Init(instance_id, remote_ip, remote_port);
|
|
id_hash_[conn->instance_id] = conn;
|
|
key_hash_[key] = conn;
|
|
conn->Open();
|
|
return conn;
|
|
}
|