Merge pull request #202 from skruger/memory_usage_stats

Report memory usage and correct CPU percentage in heartbeat messages
myzhan · May 11, 2024 · 07b7994 · 07b7994
2 parents edd5204 + ce46146
commit 07b7994
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/runner.go b/runner.go
@@ -621,9 +621,11 @@ func (r *slaveRunner) run() {
 				}
 				// send client heartbeat message
 				CPUUsage := GetCurrentCPUUsage()
+				MemUsage := GetCurrentMemUsage()
 				data := map[string]interface{}{
-					"state":             r.state,
-					"current_cpu_usage": CPUUsage,
+					"state":                r.state,
+					"current_cpu_usage":    CPUUsage,
+					"current_memory_usage": MemUsage,
 				}
 				r.client.sendChannel() <- newGenericMessage("heartbeat", data, r.nodeID)
 			case <-r.shutdownChan:

diff --git a/utils.go b/utils.go
@@ -137,3 +137,9 @@ func GetCurrentCPUUsage() float64 {
 	}
 	return percent / float64(runtime.NumCPU())
 }
+
+func GetCurrentMemUsage() uint64 {
+	var m runtime.MemStats
+	runtime.ReadMemStats(&m)
+	return m.Alloc
+}