| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791 |
- #include "pch.h"
- #include "PluginInstance.h"
- #define REDIS_TIMER_ID (1)
- CPluginInstance::CPluginInstance()
- : m_enable(false)
- , m_redis_error_notified(false)
- {
- }
- CPluginInstance::~CPluginInstance()
- {
- Stop();
- }
- void CPluginInstance::Release()
- {
- delete this;
- }
- MTAPIRES CPluginInstance::Start(IMTServerAPI * server)
- {
- MTAPIRES ret = MT_RET_OK;
- if (!server) return MT_RET_ERR_PARAMS;
- m_api = server;
- if ((m_config = m_api->PluginCreate()) == nullptr)
- return MT_RET_ERR_MEM;
- ret = m_api->About(m_info);
- if (ret != MT_RET_OK)
- m_api->LoggerOut(MTLogOK, L"Server info failed [%d]", ret);
- if ((ret = m_api->PluginSubscribe(this)) != MT_RET_OK)
- {
- m_api->LoggerOut(MTLogAtt, L"Plugin subscribe failed [%d]", ret);
- return ret;
- }
- if (ret = m_api->OrderSubscribe(this) != MT_RET_OK)
- {
- m_api->LoggerOut(MTLogAtt, L"Order subscribe failed [%d]", ret);
- return ret;
- }
- if (ret = m_api->DealSubscribe(this) != MT_RET_OK)
- {
- m_api->LoggerOut(MTLogAtt, L"Deal subscribe failed [%d]", ret);
- return ret;
- }
- if ((ret = LoadParam()) != MT_RET_OK)
- {
- m_api->LoggerOut(MTLogAtt, L"Load param failed [%d]", ret);
- return ret;
- }
- return MT_RET_OK;
- }
- MTAPIRES CPluginInstance::Stop()
- {
- MTAPIRES ret = MT_RET_OK;
- std::lock_guard<decltype(m_lock)> lk(m_lock);
- m_followers.clear();
- if (m_api == nullptr)
- return MT_RET_OK;
- if (m_config != nullptr)
- {
- m_config->Release();
- m_config = nullptr;
- }
- if ((ret = m_api->PluginUnsubscribe(this)) != MT_RET_OK && ret != MT_RET_ERR_NOTFOUND)
- m_api->LoggerOut(MTLogErr, L"failed to unsubscribe from plugin config updates [%s (%u)]",
- SMTFormat::FormatError(ret), ret);
- if ((ret = m_api->OrderUnsubscribe(this)) != MT_RET_OK && ret != MT_RET_ERR_NOTFOUND)
- m_api->LoggerOut(MTLogErr, L"failed to unsubscribe order [%s (%u)]",
- SMTFormat::FormatError(ret), ret);
- if ((ret = m_api->DealUnsubscribe(this)) != MT_RET_OK && ret != MT_RET_ERR_NOTFOUND)
- m_api->LoggerOut(MTLogErr, L"failed to unsubscribe deal [%s (%u)]",
- SMTFormat::FormatError(ret), ret);
- m_api = nullptr;
- m_enable = false;
- stop_redis();
- if (m_work_thread.joinable())
- m_work_thread.join();
- return MT_RET_OK;
- }
- void CPluginInstance::OnPluginUpdate(const IMTConPlugin * plugin)
- {
- int ret = MT_RET_OK;
- if ((ret = LoadParam()) != MT_RET_OK)
- {
- m_api->LoggerOut(MTLogAtt, L"Load param failed [%d]", ret);
- }
- }
- //void CPluginInstance::OnOrderAdd(const IMTOrder * order)
- //{
- // if (order == nullptr) return;
- //
- // std::lock_guard<decltype(m_lock)> lk(m_lock);
- // if (!m_enable) return;
- // if (order->Login() != m_trader) return;
- //
- // m_api->LoggerOut(MTLogOK, L"OnOrderAdd, login: %lld, ord: %lld, pos: %lld, state: %lld, vol_init: %lld, vol_cur: %lld",
- // order->Login(), order->Order(), order->PositionID(), order->State(), order->VolumeInitial(), order->VolumeCurrent());
- //}
- //void CPluginInstance::OnOrderUpdate(const IMTOrder * order)
- //{
- // if (order == nullptr) return;
- //
- // std::lock_guard<decltype(m_lock)> lk(m_lock);
- // if (!m_enable) return;
- // if (order->Login() != m_trader) return;
- //
- // m_api->LoggerOut(MTLogOK, L"OnOrderAdd, login: %lld, ord: %lld, pos: %lld, state: %lld, vol_init: %lld, vol_cur: %lld",
- // order->Login(), order->Order(), order->PositionID(), order->State(), order->VolumeInitial(), order->VolumeCurrent());
- //}
- void CPluginInstance::OnOrderDelete(const IMTOrder * order)
- {
- if (order == nullptr) return;
- std::lock_guard<decltype(m_lock)> lk(m_lock);
- if (!m_enable) return;
- if (order->Login() != m_trader) return;
- m_api->LoggerOut(MTLogOK, L"OnOrderDelete, login: %lld, ord: %lld, pos: %lld, state: %d, vol_init: %lld, vol_cur: %lld",
- order->Login(), order->Order(), order->PositionID(), order->State(), order->VolumeInitial(), order->VolumeCurrent());
- if (order->Type() != IMTOrder::OP_BUY
- && order->Type() != IMTOrder::OP_SELL)
- {
- // 不是buy或者sell,不跟,直接退出
- return;
- }
-
- char order_buf[128];
- sprintf(order_buf, "%lld", order->Order());
- IMTAccount* account = m_api->UserCreateAccount();
- IMTOrder* new_order = m_api->OrderCreate();
- ScopeGuard guard([account, new_order]
- {
- account->Release();
- new_order->Release();
- });
- // 订单在进入filled状态时,也会产生一个order delete回调
- if (order->Order() == order->PositionID())
- {
- // 如果是新建订单,那么写入新纪录
- char login_buf[128];
- for (auto login : m_followers)
- {
- if (m_api->UserAccountGet(login, account) != MT_RET_OK)
- {
- // TODO: 这里失败直接跳过没有做其他处理,下一次进来可能依然会遇到一样的问题
- continue;
- }
- sprintf(login_buf, "%lld", login);
- UINT level = (UINT)(account->Balance() / m_step);
- UINT volume = round((double)level * order->VolumeCurrent() / 10000) * 100;
- UINT volume_ext = round((double)level * order->VolumeCurrentExt() / 10000) * 100;
- UINT init_volume = round((double)level * order->VolumeInitial() / 10000) * 100;
- UINT init_volume_ext = round((double)level * order->VolumeInitialExt() / 10000) * 100;
- UINT64 new_order_id = 0;
- // 尚未完成建仓,目前position id不填,仅完成订单后记录
- new_order->Clear();
- new_order->VolumeInitial(init_volume);
- new_order->VolumeCurrent(volume);
- new_order->Login(login);
- new_order->Symbol(order->Symbol());
- new_order->Type(order->Type());
- new_order->Digits(order->Digits());
- new_order->DigitsCurrency(order->DigitsCurrency());
- //new_order->ContractSize(order->ContractSize());
- new_order->PriceOrder(order->PriceOrder());
- //new_order->PriceCurrent(order->PriceCurrent());
- // State不能填PARTIAL FILLED REJECTED EXPIRED
- new_order->StateSet(IMTOrder::ORDER_STATE_STARTED);
- new_order->TimeSetup(m_api->TimeCurrent());
- new_order->TimeSetupMsc(m_api->TimeCurrentMsc());
- // 一定不能填入time done
- //new_order->TimeDone(order->TimeDone());
- //new_order->TimeDoneMsc(order->TimeDoneMsc());
- // --
- //new_order->PriceSL(order->PriceSL());
- //new_order->PriceTP(order->PriceTP());
- //new_order->Comment(order->Comment());
- //new_order->ActivationFlags(order->ActivationFlags());
- //new_order->ActivationMode(order->ActivationMode());
- //new_order->ActivationPrice(order->ActivationPrice());
- //new_order->ActivationTime(order->ActivationTime());
- //new_order->PriceTrigger(order->PriceTrigger());
- //new_order->RateMargin(order->RateMargin()); //
- //new_order->ReasonSet(order->Reason()); //
- //new_order->TypeFill(order->TypeFill()); //
- //new_order->TypeTime(order->TypeTime()); //
- MTAPIRES ret = m_api->OrderAdd(new_order);
- if (ret != MT_RET_OK)
- {
- m_api->LoggerOut(MTLogErr, L"%lld failed to add order, original order #%lld [%d]", login, order->Login(), ret);
- }
- else
- {
- new_order_id = new_order->Order();
- }
- // TODO: 现在做法是,如果level为0,后面都不管
- // 如果跟单失败,那么position id也为0,所以后面也不应该处理
- position_context context;
- context.level = level;
- context.cur_ord = new_order_id;
- context.position_id = new_order_id; // 建仓,order id = position id
- int direction = 1;
- if (order->Type() == IMTOrder::OP_SELL)
- direction = -1;
- context.volume = direction * order->VolumeInitial();
- auto fut = m_redis_client->hset(order_buf, login_buf, std::string((char*)&context, sizeof(context)));
- m_redis_client->sync_commit();
- auto rep = fut.get();
- if (rep.ko())
- {
- // FIXME: 错误处理
- }
- }
- }
- else
- {
- // 如果不是新建订单,先使用position id先检查记录是否存在
- char login_buf[128];
- for (auto login : m_followers)
- {
- sprintf(login_buf, "%lld", login);
- auto fut = m_redis_client->hget(order_buf, login_buf);
- m_redis_client->sync_commit();
- auto reply = fut.get();
- // 如果不存在,忽略
- if (reply.ko()) continue;
- if (reply.is_null()) continue;
- // 获取context
- position_context context;
- memcpy(&context, reply.as_string().c_str(), sizeof(position_context));
- // 按建仓时的叙述,如果level或者position为0,亦忽略该记录
- if (context.level == 0) continue;
- if (context.position_id == 0) continue;
- // 如果存在,则继续操作
- UINT volume = round((double)context.level * order->VolumeCurrent() / 10000) * 100;
- UINT volume_ext = round((double)context.level * order->VolumeCurrentExt() / 10000) * 100;
- UINT init_volume = round((double)context.level * order->VolumeInitial() / 10000) * 100;
- UINT init_volume_ext = round((double)context.level * order->VolumeInitialExt() / 10000) * 100;
- UINT64 new_order_id = 0;
- new_order->Clear();
- new_order->VolumeInitial(init_volume);
- new_order->VolumeCurrent(volume);
- new_order->Login(login);
- new_order->Symbol(order->Symbol());
- new_order->Type(order->Type());
- new_order->Digits(order->Digits());
- new_order->DigitsCurrency(order->DigitsCurrency());
- //new_order->ContractSize(order->ContractSize());
- new_order->PriceOrder(order->PriceOrder());
- //new_order->PriceCurrent(order->PriceCurrent());
- // State不能填PARTIAL FILLED REJECTED EXPIRED
- new_order->StateSet(IMTOrder::ORDER_STATE_STARTED);
- new_order->TimeSetup(m_api->TimeCurrent());
- new_order->TimeSetupMsc(m_api->TimeCurrentMsc());
- new_order->PositionID(context.position_id);
- MTAPIRES ret = m_api->OrderAdd(new_order);
- if (ret != MT_RET_OK)
- {
- m_api->LoggerOut(MTLogErr, L"%lld failed to add order, original order #%lld [%d]", login, order->Order(), ret);
- // FIXME: 如果做单失败该怎么办
- continue;
- }
- new_order_id = new_order->Order();
- m_api->LoggerOut(MTLogOK, L"%lld add order #%lld, original order #%lld [%d]", login, new_order_id, order->Order());
- // 完成之后,写入新纪录
- context.cur_ord = new_order_id;
- int direction = 1;
- if (order->Type() == IMTOrder::OP_SELL)
- direction = -1;
- context.volume += direction * order->VolumeInitial();
- auto wfut = m_redis_client->hset(order_buf, login_buf, std::string((char*)&context, sizeof(context)));
- m_redis_client->sync_commit();
- auto wrep = fut.get();
- if (wrep.ko())
- {
- // FIXME: 错误处理
- }
- }
- }
- }
- //void CPluginInstance::OnOrderClean(const UINT64 login)
- //{
- // std::lock_guard<decltype(m_lock)> lk(m_lock);
- // if (!m_enable) return;
- // if (login != m_trader) return;
- //
- // m_api->LoggerOut(MTLogOK, L"OnOrderClean, Login: %d", login);
- //}
- //void CPluginInstance::OnDealAdd(const IMTDeal * deal)
- //{
- // if (deal == nullptr) return;
- //
- // std::lock_guard<decltype(m_lock)> lk(m_lock);
- // if (!m_enable) return;
- // if (deal->Login() != m_trader) return;
- //
- // m_api->LoggerOut(MTLogOK, L"OnDealAdd, login: %lld, deal: %lld, ord: %lld, pos: %lld, vol: %lld, volc: %lld",
- // deal->Login(), deal->Deal(), deal->Order(), deal->PositionID(), deal->Volume(), deal->VolumeClosed());
- //}
- //void CPluginInstance::OnDealUpdate(const IMTDeal * deal)
- //{
- // if (deal == nullptr) return;
- //
- // std::lock_guard<decltype(m_lock)> lk(m_lock);
- // if (!m_enable) return;
- // if (deal->Login() != m_trader) return;
- //
- // m_api->LoggerOut(MTLogOK, L"OnDealUpdate, login: %lld, deal: %lld, ord: %lld, pos: %lld, vol: %lld, volc: %lld",
- // deal->Login(), deal->Deal(), deal->Order(), deal->PositionID(), deal->Volume(), deal->VolumeClosed());
- //}
- //void CPluginInstance::OnDealDelete(const IMTDeal * deal)
- //{
- // if (deal == nullptr) return;
- //
- // std::lock_guard<decltype(m_lock)> lk(m_lock);
- // if (!m_enable) return;
- // if (deal->Login() != m_trader) return;
- //
- // m_api->LoggerOut(MTLogOK, L"OnDealDelete, login: %lld, deal: %lld, ord: %lld, pos: %lld, vol: %lld, volc: %lld",
- // deal->Login(), deal->Deal(), deal->Order(), deal->PositionID(), deal->Volume(), deal->VolumeClosed());
- //}
- //void CPluginInstance::OnDealClean(const UINT64 login)
- //{
- // std::lock_guard<decltype(m_lock)> lk(m_lock);
- // if (!m_enable) return;
- // if (login != m_trader) return;
- //
- // m_api->LoggerOut(MTLogOK, L"OndealClean, Login: %d", login);
- //}
- void CPluginInstance::OnDealPerform(const IMTDeal * deal, IMTAccount * account, IMTPosition * position)
- {
- // position为nullptr时,说明该deal不是由交易本身触发
- // account是deal完成后的用户状况
- // position是交易完成后的持仓状况
- // 对于deal是关闭一个持仓的情况,该position的volume会是0
- if (position == nullptr
- || deal == nullptr
- || account == nullptr)
- return;
- std::lock_guard<decltype(m_lock)> lk(m_lock);
- if (!m_enable) return;
- if (deal->Login() != m_trader) return;
- //m_api->LoggerOut(MTLogOK, L"OnDealPerform, login: %lld, deal: %lld, ord: %lld, pos: %lld, vol: %lld, volc: %lld",
- // deal->Login(), deal->Deal(), deal->Order(), deal->PositionID(), deal->Volume(), deal->VolumeClosed());
- // FIXME: 需要检查现在的deal是否和之前的order对应。如果没有记录到,则没法根本无法正常跟单以及平仓
- char order_buf[128];
- sprintf(order_buf, "%lld", deal->Order());
- IMTDeal* new_deal = m_api->DealCreate();
- std::vector<std::string>* fields = nullptr;
- ScopeGuard guard([new_deal, fields]
- {
- new_deal->Release();
- if (fields)
- {
- fields->clear();
- delete fields;
- }
- });
- char login_buf[128];
- for (auto login : m_followers)
- {
- sprintf(login_buf, "%lld", login);
- auto fut = m_redis_client->hget(order_buf, login_buf);
- m_redis_client->sync_commit();
- auto reply = fut.get();
- m_api->LoggerOut(MTLogOK, L"%lld add deal, original deal #%lld", login, deal->Deal());
- // 如果不存在,忽略
- if (reply.ko()) continue;
- if (reply.is_null()) continue;
- // 获取context
- position_context context;
- memcpy(&context, reply.as_string().c_str(), sizeof(position_context));
- // 按建仓时的叙述,如果level或者position为0,亦忽略该记录
- if (context.level == 0) continue;
- if (context.position_id == 0) continue;
- uint64_t volume = context.level * deal->Volume() / 100;
- uint64_t volume_ext = context.level * deal->VolumeExt() / 100;
- uint64_t volume_closed = context.level * deal->VolumeClosed() / 100;
- uint64_t volume_closed_ext = context.level * deal->VolumeClosed() / 100;
- double prop = (double)volume / deal->Volume();
- double profit = deal->Profit() * prop;
- double commission = deal->Commission() * prop;
- double storage = deal->Storage() * prop;
- double raw_profit = deal->ProfitRaw() * prop;
- new_deal->Clear();
- new_deal->DealSet(0);
- new_deal->Volume(volume);
- new_deal->VolumeExt(volume_ext);
- new_deal->VolumeClosed(volume_closed);
- new_deal->VolumeClosedExt();
- new_deal->ProfitRaw(raw_profit);
- new_deal->Profit(profit);
- new_deal->Commission(commission);
- new_deal->Storage(storage);
- new_deal->Order(context.cur_ord);
- new_deal->PositionID(context.position_id);
- MTAPIRES ret = m_api->DealAdd(new_deal);
- if (ret != MT_RET_OK)
- {
- // TODO: 有没有更多的错误处理?
- m_api->LoggerOut(MTLogErr, L"%lld cannot add deal [%d], original deal: #%lld", login, ret, deal->Deal());
- continue;
- }
- m_api->LoggerOut(MTLogOK, L"add deal #%lld, original deal #%lld [%d]", new_deal->Deal(), deal->Deal());
- if (position->Volume() == 0)
- {
- if (fields == nullptr)
- fields = new(std::vector<std::string>);
- fields->push_back(login_buf);
- }
- }
- if (position->Volume() == 0)
- {
- // 如果平仓,则删除hash值
- if (position->Volume() == 0)
- {
- auto fut = m_redis_client->hdel(order_buf, *fields);
- m_redis_client->sync_commit();
- auto rep = fut.get();
- if (rep.ko())
- {
- // TODO 错误处理
- }
- }
- // TODO 当position中的volume为0时,持仓被彻底平调,被跟订单是否也该检查
- }
- }
- MTAPIRES CPluginInstance::LoadParam()
- {
- MTAPIRES res = MT_RET_OK;
- IMTConParam* param = NULL;
- CMTStr128 tmp;
- if (!m_api || !m_config) return MT_RET_ERR_PARAMS;
- if ((res = m_api->PluginCurrent(m_config)) != MT_RET_OK)
- {
- m_api->LoggerOut(MTLogErr, L"failed to get current plugin configuration [%s (%u)]",
- SMTFormat::FormatError(res), res);
- return res;
- }
- if ((param = m_api->PluginParamCreate()) == NULL)
- {
- m_api->LoggerOut(MTLogErr, L"failed to create plugin parameter object");
- return MT_RET_ERR_MEM;
- }
- ScopeGuard param_release([param]()
- {
- param->Release();
- });
- std::lock_guard<decltype(m_lock)> lk(m_lock);
- //m_api->LoggerOut(MTLogOK, L"Load redis server params");
- if ((res = m_config->ParameterGet(L"Redis Server", param)) != MT_RET_OK || param->Type() != IMTConParam::TYPE_STRING)
- {
- return(MT_RET_ERR_PARAMS);
- }
- std::string redis_server = ws2s(param->ValueString());
- if ((res = m_config->ParameterGet(L"Redis Port", param)) != MT_RET_OK || param->Type() != IMTConParam::TYPE_INT)
- {
- return(MT_RET_ERR_PARAMS);
- }
- int redis_port = param->ValueInt();
- //if ((res = m_config->ParameterGet(L"Redis User", param)) != MT_RET_OK || param->Type() != IMTConParam::TYPE_STRING)
- //{
- // return(MT_RET_ERR_PARAMS);
- //}
- //std::string redis_user = ws2s(param->ValueString());
- if ((res = m_config->ParameterGet(L"Redis Password", param)) != MT_RET_OK || param->Type() != IMTConParam::TYPE_STRING)
- {
- return(MT_RET_ERR_PARAMS);
- }
- std::string redis_password = ws2s(param->ValueString());
- if (m_redis_server != redis_server
- || m_redis_port != redis_port
- //|| m_redis_user != redis_user
- || m_redis_password != redis_password)
- {
- m_redis_server = redis_server;
- m_redis_port = redis_port;
- //m_redis_user = redis_user;
- m_redis_password = redis_password;
- stop_redis();
- if (start_redis())
- {
- if (!m_work_thread.joinable())
- {
- m_work_thread = std::thread(&CPluginInstance::keep_alive, this);
- }
- }
- else
- {
- m_api->LoggerOut(MTLogErr, L"failed to connect to redis server");
- return MT_RET_ERR_CONNECTION;
- }
- }
- //m_api->LoggerOut(MTLogOK, L"Load trade params");
- if ((res = m_config->ParameterGet(L"Trader", param)) != MT_RET_OK || param->Type() != IMTConParam::TYPE_INT)
- {
- return(MT_RET_ERR_PARAMS);
- }
- m_trader = param->ValueInt();
- if ((res = m_config->ParameterGet(L"Step", param)) != MT_RET_OK || param->Type() != IMTConParam::TYPE_INT)
- {
- return(MT_RET_ERR_PARAMS);
- }
- m_step = param->ValueInt();
- if ((res = m_config->ParameterGet(L"Tolerance", param)) != MT_RET_OK || param->Type() != IMTConParam::TYPE_INT)
- {
- return(MT_RET_ERR_PARAMS);
- }
- m_tolerance = param->ValueInt();
- if ((res = m_config->ParameterGet(L"Groups", param)) != MT_RET_OK || param->Type() != IMTConParam::TYPE_STRING)
- {
- return(MT_RET_ERR_PARAMS);
- }
- wcsncpy(m_groups, param->ValueString(), 1024);
- //if ((res = m_config->ParameterGet(L"Logins", param)) != MT_RET_OK || param->Type() != IMTConParam::TYPE_STRING)
- //{
- // return(MT_RET_ERR_PARAMS);
- //}
- //m_logins = param->ValueString();
- res = LoadLogins();
- if (res != MT_RET_OK)
- {
- m_api->LoggerOut(MTLogErr, L"failed to load clients [%d]", res);
- return res;
- }
- //m_api->LoggerOut(MTLogOK, L"Load param success");
- m_enable = true;
- return MT_RET_OK;
- }
- MTAPIRES CPluginInstance::LoadLogins()
- {
- m_followers.clear();
- IMTConGroup* group = m_api->GroupCreate();
- ScopeGuard group_guard([&]()
- {
- group->Release();
- });
- if (group == nullptr) return MT_RET_ERR_MEM;
- UINT total = m_api->GroupTotal();
-
- MTAPIRES ret = MT_RET_OK;
- std::vector<std::wstring> groups;
- for (int i = 0; i < total; i++)
- {
- ret = m_api->GroupNext(i, group);
- if (ret != MT_RET_OK) break;
- if (CheckGroup(m_groups, group->Group()) == FALSE) continue;
- m_api->LoggerOut(MTLogOK, L"group %s matched config", group->Group());
- groups.push_back(group->Group());
- }
- for (auto& group : groups)
- {
- UINT64* logins = nullptr;
- UINT total_users = 0;
- ret = m_api->UserLogins(group.c_str(), logins, total_users);
- m_api->LoggerOut(MTLogOK, L"add group %s, total users: %d", group.c_str(), total_users);
- if (ret == MT_RET_OK && logins != nullptr)
- {
- for (int i = 0; i < total_users; ++i)
- {
- m_api->LoggerOut(MTLogOK, L"add %d to list", logins[i]);
- m_followers.push_back(logins[i]);
- }
- }
- if (logins)
- {
- m_api->Free(logins);
- }
- }
- return ret;
- }
- bool CPluginInstance::start_redis()
- {
- try
- {
- m_redis_error_notified = false;
- if (m_redis_client)
- {
- if (m_redis_client->is_connected())
- return true;
- }
- m_redis_client.reset(new cpp_redis::client);
- m_redis_client->connect
- (
- m_redis_server,
- m_redis_port,
- [this](const std::string& host, std::size_t port, cpp_redis::connect_state status)
- {
- if (status == cpp_redis::connect_state::ok)
- {
- m_api->LoggerOut(MTLogOK, L"redis server connected");
- }
- },
- 500,
- 10000000,
- 1000
- );
- if (m_redis_password != "")
- {
- auto fut = m_redis_client->auth(m_redis_password);
- m_redis_client->sync_commit();
- auto reply = fut.get();
- if (reply.is_error())
- {
- m_api->LoggerOut(MTLogErr, L"connect: authentication failed");
- return false;
- }
- }
- }
- catch (tacopie::tacopie_error& e)
- {
- m_api->LoggerOut(MTLogErr, L"redis conn: %s", e.what());
- return false;
- }
- catch (std::exception& e)
- {
- m_api->LoggerOut(MTLogErr, L"redis conn: %s", e.what());
- return false;
- }
- catch (...)
- {
- m_api->LoggerOut(MTLogErr, L"failed to connect to server");
- return false;
- }
- return true;
- }
- void CPluginInstance::stop_redis()
- {
- try
- {
- if (m_redis_client)
- {
- m_redis_client->sync_commit();
- m_redis_client->disconnect();
- m_redis_client.reset();
- }
- }
- catch (tacopie::tacopie_error& e)
- {
- m_api->LoggerOut(MTLogErr, L"stop redis: %s", e.what());
- }
- catch (std::exception& e)
- {
- m_api->LoggerOut(MTLogErr, L"stop redis: %s", e.what());
- }
- catch (...)
- {
- m_api->LoggerOut(MTLogErr, L"stop redis: unknown error");
- }
- }
- void CPluginInstance::keep_alive()
- {
- using namespace std::chrono_literals;
- int counter = 500;
- while (m_enable)
- {
- if (counter-- <= 0)
- {
- counter = 500;
- std::lock_guard<decltype(m_lock)> lk(m_lock);
- if (m_redis_client)
- {
- if (m_redis_client->is_connected())
- {
- m_redis_client->ping([this](cpp_redis::reply& r)
- {
- if (r.ko())
- {
- m_redis_error_notified = true;
- if (!m_redis_error_notified)
- {
- m_api->LoggerOut(MTLogErr, L"redis: %s", r.error().c_str());
- }
- }
- });
- }
- else
- {
- m_redis_error_notified = true;
- if (!m_redis_error_notified)
- {
- m_api->LoggerOut(MTLogErr, L"redis server not connected");
- }
- stop_redis();
- start_redis();
- }
- }
- }
- std::this_thread::sleep_for(16ms);
- }
- }
|