cgroup v2 CPU & memory stats for correct chart generation (#65)

* fix reading of CPUSET in case cgroups v2 have been detected * add stat.file and events.oom_kill stats to the list of stats to be charted
f18m · Oct 29, 2024 · 51c5901 · 51c5901
1 parent ddd963d
commit 51c5901
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/collector/src/cgroups_cpuacct.cpp b/collector/src/cgroups_cpuacct.cpp
@@ -35,8 +35,12 @@
 
 bool CMonitorCgroups::read_cpuset_cpus(std::string kernelPath, std::set<uint64_t>& cpus)
 {
-    std::set<uint64_t> empty_set;
-    return read_integers_with_range_validation(kernelPath + "/cpuset.cpus", 0, INT32_MAX, cpus);
+    if (m_nCGroupsFound == CG_NONE)
+        return false;
+
+    std::string cpuset_path = kernelPath + ((m_nCGroupsFound == CG_VERSION2)? "/cpuset.cpus.effective" : "/cpuset.cpus");
+
+    return read_integers_with_range_validation(cpuset_path, 0, INT32_MAX, cpus);
 }
 
 bool CMonitorCgroups::read_cpuacct_line(FastFileReader& reader, std::vector<uint64_t>& valuesINT /* OUT */)

diff --git a/collector/src/cgroups_memory.cpp b/collector/src/cgroups_memory.cpp
@@ -192,7 +192,7 @@ void CMonitorCgroups::sample_memory(
     case CG_VERSION2: {
         key_value_map_t newEventsValues;
         if (sample_flat_keyed_file(m_cgroup_memory_v2_events, allowedStatsNames, "events.", newEventsValues)) {
-            if (print) {
+            if (print) { 
                 for (auto entry : newEventsValues) {
                     auto prevValue = m_memory_prev_values.v2_events.find(entry.first);
                     if (prevValue != m_memory_prev_values.v2_events.end())

diff --git a/collector/src/main.cpp b/collector/src/main.cpp
@@ -879,6 +879,8 @@ int CMonitorCollectorApp::run_main_loop()
         charted_stats_from_cgroup_memory_v1.insert("failcnt");
         // cgroups v2
         charted_stats_from_cgroup_memory_v2.insert("stat.anon");
+        charted_stats_from_cgroup_memory_v2.insert("stat.file");
+        charted_stats_from_cgroup_memory_v2.insert("events.oom_kill");
     }
     // else: leave empty