From 4a4094e6cdc7f5d9e5a470ccf82df1c780c7e224 Mon Sep 17 00:00:00 2001
From: Vladimir Lavor <vlavor@cisco.com>
Date: Mon, 2 May 2022 13:58:04 +0200
Subject: [PATCH] read memory stats from per-heap couters

Signed-off-by: Vladimir Lavor <vlavor@cisco.com>
Change-Id: Ic4a500ea83b392eb9720575d95597bc48b44c759
---
 api/stats.go  | 19 ++++++++++++++++++-
 core/stats.go | 60 ++++++++++++++++++++++++++++++++++++++++++++---------------
 2 files changed, 63 insertions(+), 16 deletions(-)

diff --git a/api/stats.go b/api/stats.go
index 3a72c6c..d5665b7 100644
--- a/api/stats.go
+++ b/api/stats.go
@@ -117,6 +117,23 @@ type BufferPool struct {
 
 // MemoryStats represents memory stats segment counters.
 type MemoryStats struct {
+	// Deprecated: /mem/statseg total memory directory
 	Total float64
-	Used  float64
+	// Deprecated: /mem/statseg used memory directory
+	Used float64
+
+	// stat/main memory usage per-heap
+	Stat map[int]MemoryCounters
+	Main map[int]MemoryCounters
+}
+
+// MemoryCounters represents values of various memory usage
+type MemoryCounters struct {
+	Total      uint64
+	Used       uint64
+	Free       uint64
+	UsedMMap   uint64
+	TotalAlloc uint64
+	FreeChunks uint64
+	Releasable uint64
 }
diff --git a/core/stats.go b/core/stats.go
index 3218f1e..4033981 100644
--- a/core/stats.go
+++ b/core/stats.go
@@ -39,9 +39,11 @@ const (
 
 	CounterStatsPrefix = "/err/"
 
-	MemoryStatPrefix  = "/mem/statseg"
-	MemoryStats_Total = "total"
-	MemoryStats_Used  = "used"
+	MemoryStatSegPrefix = "/mem/statseg"
+	MemoryStatPrefix    = "/mem/stat"
+	MemoryMainPrefix    = "/mem/main"
+	MemoryStats_Total   = "total"
+	MemoryStats_Used    = "used"
 
 	InterfaceStatsPrefix         = "/if/"
 	InterfaceStats_Names         = InterfaceStatsPrefix + "names"
@@ -545,22 +547,50 @@ func (c *StatsConnection) GetBufferStats(bufStats *api.BufferStats) (err error)
 }
 
 func (c *StatsConnection) GetMemoryStats(memStats *api.MemoryStats) (err error) {
-	if err := c.updateStats(&c.memStatsData, MemoryStatPrefix); err != nil {
+	if err := c.updateStats(&c.memStatsData, MemoryStatSegPrefix, MemoryStatPrefix, MemoryMainPrefix); err != nil {
 		return err
 	}
+	convertStats := func(stats []adapter.Counter) api.MemoryCounters {
+		memUsg := make([]adapter.Counter, 7)
+		copy(memUsg, stats)
+		return api.MemoryCounters{
+			Total: uint64(memUsg[0]), Used: uint64(memUsg[1]), Free: uint64(memUsg[2]), UsedMMap: uint64(memUsg[3]),
+			TotalAlloc: uint64(memUsg[4]), FreeChunks: uint64(memUsg[5]), Releasable: uint64(memUsg[6]),
+		}
+	}
 
 	for _, stat := range c.memStatsData.Entries {
-		_, f := path.Split(string(stat.Name))
-		var val float64
-		m, ok := stat.Data.(adapter.ScalarStat)
-		if ok {
-			val = float64(m)
-		}
-		switch f {
-		case MemoryStats_Total:
-			memStats.Total = val
-		case MemoryStats_Used:
-			memStats.Used = val
+		if strings.Contains(string(stat.Name), MemoryStatSegPrefix) {
+			_, f := path.Split(string(stat.Name))
+			var val float64
+			m, ok := stat.Data.(adapter.ScalarStat)
+			if ok {
+				val = float64(m)
+			}
+			switch f {
+			case MemoryStats_Total:
+				memStats.Total = val
+			case MemoryStats_Used:
+				memStats.Used = val
+			}
+		} else if strings.Contains(string(stat.Name), MemoryStatPrefix) {
+			if perHeapStats, ok := stat.Data.(adapter.SimpleCounterStat); ok {
+				if memStats.Stat == nil {
+					memStats.Stat = make(map[int]api.MemoryCounters)
+				}
+				for heap, stats := range perHeapStats {
+					memStats.Stat[heap] = convertStats(stats)
+				}
+			}
+		} else if strings.Contains(string(stat.Name), MemoryMainPrefix) {
+			if perHeapStats, ok := stat.Data.(adapter.SimpleCounterStat); ok {
+				if memStats.Main == nil {
+					memStats.Main = make(map[int]api.MemoryCounters)
+				}
+				for heap, stats := range perHeapStats {
+					memStats.Main[heap] = convertStats(stats)
+				}
+			}
 		}
 	}
 	return nil
-- 
2.16.6