Stats API: added GetMemory()

Retrieved numbers relate to the statseg heap.

Change-Id: I72750183db3524481918c71c993b39e65c28ddb6
Signed-off-by: Vladimir Lavor <vlavor@cisco.com>

1 files changed, 27 insertions, 0 deletions

diff --git a/core/stats.go b/core/stats.go
index cd93cc1..f2da494 100644
--- a/core/stats.go
+++ b/core/stats.go
@@ -39,6 +39,10 @@ const (
 
 	CounterStatsPrefix = "/err/"
 
+	MemoryStatPrefix  = "/mem/statseg"
+	MemoryStats_Total = "total"
+	MemoryStats_Used  = "used"
+
 	InterfaceStatsPrefix         = "/if/"
 	InterfaceStats_Names         = InterfaceStatsPrefix + "names"
 	InterfaceStats_Drops         = InterfaceStatsPrefix + "drops"
@@ -80,6 +84,7 @@ type StatsConnection struct {
 	ifaceStatsData *adapter.StatDir
 	sysStatsData   *adapter.StatDir
 	bufStatsData   *adapter.StatDir
+	memStatsData   *adapter.StatDir
 }
 
 func newStatsConnection(stats adapter.StatsAPI) *StatsConnection {
@@ -471,3 +476,25 @@ func (c *StatsConnection) GetBufferStats(bufStats *api.BufferStats) (err error)
 
 	return nil
 }
+
+func (c *StatsConnection) GetMemoryStats(memStats *api.MemoryStats) (err error) {
+	if err := c.updateStats(&c.memStatsData, MemoryStatPrefix); err != nil {
+		return err
+	}
+
+	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
+		}
+	}
+	return nil
+}