70 lines
2.4 KiB
C++
70 lines
2.4 KiB
C++
#include "RequestMetaDataWork.h"
|
|
|
|
#include <common/toolkit/MessagingTk.h>
|
|
#include <common/net/message/nodes/HeartbeatRequestMsg.h>
|
|
#include <common/net/message/mon/RequestMetaDataMsg.h>
|
|
#include <common/net/message/mon/RequestMetaDataRespMsg.h>
|
|
#include <components/StatsCollector.h>
|
|
|
|
void RequestMetaDataWork::process(char* bufIn, unsigned bufInLen, char* bufOut, unsigned bufOutLen)
|
|
{
|
|
|
|
if (!node->getIsResponding())
|
|
{
|
|
HeartbeatRequestMsg heartbeatRequestMsg;
|
|
if(MessagingTk::requestResponse(*node, heartbeatRequestMsg,
|
|
NETMSGTYPE_Heartbeat))
|
|
{
|
|
LOG(GENERAL, DEBUG, "Node is responding again.",
|
|
("NodeID", node->getNodeIDWithTypeStr()));
|
|
node->setIsResponding(true);
|
|
}
|
|
}
|
|
|
|
Result result = {};
|
|
result.data.isResponding = false;
|
|
|
|
if (node->getIsResponding())
|
|
{
|
|
// generate the RequestDataMsg with the lastStatsTime
|
|
RequestMetaDataMsg requestDataMsg(node->getLastStatRequestTime().count());
|
|
auto respMsg = MessagingTk::requestResponse(*node, requestDataMsg,
|
|
NETMSGTYPE_RequestMetaDataResp);
|
|
|
|
if (!respMsg)
|
|
{
|
|
LOG(GENERAL, DEBUG, "Node is not responding.", ("NodeID", node->getNodeIDWithTypeStr()));
|
|
node->setIsResponding(false);
|
|
}
|
|
else
|
|
{
|
|
// get response and process it
|
|
auto metaRspMsg = static_cast<RequestMetaDataRespMsg*>(respMsg.get());
|
|
result.highResStatsList = std::move(metaRspMsg->getStatsList());
|
|
|
|
result.data.isResponding = true;
|
|
result.data.indirectWorkListSize = metaRspMsg->getIndirectWorkListSize();
|
|
result.data.directWorkListSize = metaRspMsg->getDirectWorkListSize();
|
|
result.data.sessionCount = metaRspMsg->getSessionCount();
|
|
result.data.hostnameid = metaRspMsg->gethostnameid();
|
|
|
|
if (!result.highResStatsList.empty())
|
|
{
|
|
auto lastStatsRequestTime = std::chrono::milliseconds(
|
|
result.highResStatsList.front().rawVals.statsTimeMS);
|
|
node->setLastStatRequestTime(lastStatsRequestTime);
|
|
}
|
|
|
|
if (collectClientOpsByNode)
|
|
result.ipOpsUnorderedMap = ClientOpsRequestor::request(*node, false);
|
|
|
|
if (collectClientOpsByUser)
|
|
result.userOpsUnorderedMap = ClientOpsRequestor::request(*node, true);
|
|
}
|
|
}
|
|
|
|
result.node = std::move(node);
|
|
|
|
statsCollector->insertMetaData(std::move(result));
|
|
}
|