Skip to content

Commit 7b7c0ba

Browse files
zliang-scGitHub Enterprise
authored andcommitted
Merge pull request #224 from Snapchat/remove-dot-in-nodename-metrics
fix nodename metricsname
2 parents a55c687 + 98e306c commit 7b7c0ba

1 file changed

Lines changed: 11 additions & 8 deletions

File tree

src/modules/keydb_modstatsd/modmain.cpp

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ std::string m_strPrefix { "keydb" };
7676
const std::regex g_replica_or_db_info_regex { "^(slave|db)(\\d+)" };
7777
const char *g_string_counter_separator = "__";
7878
const uint64_t g_stats_buffer_size_bytes = 1600;
79-
utsname sysName;
79+
std::string nodeName;
8080
int unameResult;
8181

8282
enum class StatsD_Type {
@@ -600,7 +600,7 @@ void event_cron_handler(struct RedisModuleCtx *ctx, RedisModuleEvent eid, uint64
600600

601601
/* node name */
602602
if (unameResult == 0) {
603-
g_stats->increment("node_name" + std::string(g_string_counter_separator) + sysName.nodename);
603+
g_stats->increment("node_name" + std::string(g_string_counter_separator) + nodeName);
604604
}
605605

606606
/* Log INFO Fields */
@@ -672,17 +672,20 @@ extern "C" int RedisModule_OnLoad(RedisModuleCtx *ctx, RedisModuleString **argv,
672672
RedisModule_AutoMemory(ctx);
673673
/* Use pod name if available*/
674674
const char *podName = getenv("POD_NAME");
675+
utsname sysName;
675676
unameResult = uname(&sysName);
677+
if (unameResult == 0) {
678+
nodeName = std::string(sysName.nodename);
679+
std::replace(nodeName.begin(), nodeName.end(), '.', '-');
680+
}
676681
if (podName != nullptr) {
677682
m_strPrefix = podName;
683+
std::replace(m_strPrefix.begin(), m_strPrefix.end(), '.', '-');
678684
}
679-
else {
680-
if (unameResult == 0) {
681-
m_strPrefix = std::string(sysName.nodename);
682-
unameResult = 1;
683-
}
685+
else if (unameResult == 0) {
686+
m_strPrefix = nodeName;
687+
unameResult = 1;
684688
}
685-
std::replace(m_strPrefix.begin(), m_strPrefix.end(), '.', '-');
686689

687690
for (int iarg = 0; iarg < argc; ++iarg) {
688691
size_t len = 0;

0 commit comments

Comments
 (0)