kapitan/IXWebSocket
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

(cobra to statsd bot) add ability to extract a numerical value and send a gauge event to statsd

Browse Source
This commit is contained in:
Benjamin Sergeant
2020-03-29 19:31:49 -07:00
parent cfa5718e40
commit f8bf1fe7cd
6 changed files with 58 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
ixbots/ixbots/IXCobraToStatsdBot.cpp
View File

@ -40,7 +40,7 @@ namespace ix
// Extract an attribute from a Json Value.
// extractAttr("foo.bar", {"foo": {"bar": "baz"}}) => baz
//
std::string extractAttr(const std::string& attr, const Json::Value& jsonValue)
Json::Value extractAttr(const std::string& attr, const Json::Value& jsonValue)
{
// Split by .
std::string token;
@ -53,7 +53,7 @@ namespace ix
val = val[token];
}
return val.asString();
return val;
}
int cobra_to_statsd_bot(const ix::CobraConfig& config,
@ -62,6 +62,7 @@ namespace ix
const std::string& position,
StatsdClient& statsdClient,
const std::string& fields,
const std::string& gauge,
bool verbose,
size_t maxQueueSize,
bool enableHeartbeat,
@ -124,7 +125,7 @@ namespace ix
std::thread t2(heartbeat);
auto statsdSender = [&statsdClient, &queueManager, &sentCount, &tokens, &stop] {
auto statsdSender = [&statsdClient, &queueManager, &sentCount, &tokens, &stop, &gauge, &fatalCobraError] {
while (true)
{
Json::Value msg = queueManager.pop();
@ -136,10 +137,51 @@ namespace ix
for (auto&& attr : tokens)
{
id += ".";
id += extractAttr(attr, msg);
auto val = extractAttr(attr, msg);
id += val.asString();
}
if (gauge.empty())
{
statsdClient.count(id, 1);
}
else
{
// spdlog::info("{} - {} -> {}", id, gauge, x);
auto val = extractAttr(gauge, msg);
if (val.isInt())
{
auto x = val.asInt();
statsdClient.gauge(id, (size_t) x);
}
else if (val.isInt64())
{
auto x = val.asInt64();
statsdClient.gauge(id, (size_t) x);
}
else if (val.isUInt())
{
auto x = val.asUInt();
statsdClient.gauge(id, (size_t) x);
}
else if (val.isUInt64())
{
auto x = val.asUInt64();
statsdClient.gauge(id, (size_t) x);
}
else if (val.isDouble())
{
auto x = val.asUInt64();
statsdClient.gauge(id, (size_t) x);
}
else
{
spdlog::error("Gauge {} is not a numberic type", gauge);
fatalCobraError = true;
}
}
statsdClient.count(id, 1);
sentCount += 1;
}
};

1
ixbots/ixbots/IXCobraToStatsdBot.h
View File

@ -18,6 +18,7 @@ namespace ix
const std::string& position,
StatsdClient& statsdClient,
const std::string& fields,
const std::string& gauge,
bool verbose,
size_t maxQueueSize,
bool enableHeartbeat,