stats: golang vpp_if_stats_client

see README for details

Change-Id: Ida603ccaee21dabc903512699b5b355cebb70320
Signed-off-by: Koren Lev <korenlev@gmail.com>

1 files changed, 223 insertions, 0 deletions

diff --git a/extras/vpp_if_stats/vpp_if_stats.go b/extras/vpp_if_stats/vpp_if_stats.go
new file mode 100755
index 00000000000..48ce0853d1e
--- /dev/null
+++ b/extras/vpp_if_stats/vpp_if_stats.go
@@ -0,0 +1,223 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"git.fd.io/govpp.git"
+	"git.fd.io/govpp.git/adapter"
+	"git.fd.io/govpp.git/adapter/vppapiclient"
+	"git.fd.io/govpp.git/api"
+	"git.fd.io/govpp.git/core"
+	"git.fd.io/govpp.git/examples/bin_api/interfaces"
+	"git.fd.io/govpp.git/examples/bin_api/vpe"
+	"log"
+)
+
+//////////////////////////////////////
+/////////   Data structs   ///////////
+//////////////////////////////////////
+
+const defaultStatsSocketPath = "/run/vpp/stats.sock"
+const defaultShmPrefix = ""
+
+func parseMacAddress(l2Address []byte, l2AddressLength uint32) string {
+	var mac string
+	for i := uint32(0); i < l2AddressLength; i++ {
+		mac += fmt.Sprintf("%02x", l2Address[i])
+		if i < l2AddressLength-1 {
+			mac += ":"
+		}
+	}
+	return mac
+}
+
+type interfaceStats struct {
+	TxBytes   uint64
+	TxPackets uint64
+	TxErrors  uint64
+	RxBytes   uint64
+	RxPackets uint64
+	RxErrors  uint64
+	Drops     uint64
+	Punts     uint64
+}
+
+type vppInterface struct {
+	interfaces.SwInterfaceDetails
+	Stats interfaceStats
+}
+
+type vppConnector struct {
+	statsSocketPath string
+	shmPrefix       string
+
+	conn  *core.Connection
+	api   api.Channel
+	stats adapter.StatsAPI
+
+	VppDetails vpe.ShowVersionReply
+	Interfaces []*vppInterface
+}
+
+//////////////////////////////////////
+/////////   VPP workflow   ///////////
+//////////////////////////////////////
+
+func (v *vppConnector) getVppVersion() error {
+	if err := v.api.SendRequest(&vpe.ShowVersion{}).ReceiveReply(&v.VppDetails); err != nil {
+		return fmt.Errorf("failed to fetch vpp version: %v", err)
+	}
+	return nil
+}
+
+func (v *vppConnector) getInterfaces() error {
+	ifCtx := v.api.SendMultiRequest(&interfaces.SwInterfaceDump{})
+	for {
+		ifDetails := interfaces.SwInterfaceDetails{}
+		stop, err := ifCtx.ReceiveReply(&ifDetails)
+		if err != nil {
+			return fmt.Errorf("failed to fetch vpp interface: %v", err)
+		}
+		if stop {
+			break
+		}
+
+		v.Interfaces = append(v.Interfaces, &vppInterface{SwInterfaceDetails: ifDetails})
+	}
+	return nil
+}
+
+func (v *vppConnector) connect() (err error) {
+	if v.conn, err = govpp.Connect(v.shmPrefix); err != nil {
+		return fmt.Errorf("failed to connect to vpp: %v", err)
+	}
+
+	if v.api, err = v.conn.NewAPIChannel(); err != nil {
+		return fmt.Errorf("failed to create api channel: %v", err)
+	}
+
+	v.stats = vppapiclient.NewStatClient(v.statsSocketPath)
+	if err = v.stats.Connect(); err != nil {
+		return fmt.Errorf("failed to connect to Stats adapter: %v", err)
+	}
+
+	return
+}
+
+func (v *vppConnector) disconnect() {
+	if v.stats != nil {
+		v.stats.Disconnect()
+	}
+	if v.conn != nil {
+		v.conn.Disconnect()
+	}
+}
+
+func (v *vppConnector) reduceCombinedCounters(stat *adapter.StatEntry) *[]adapter.CombinedCounter {
+	counters := stat.Data.(adapter.CombinedCounterStat)
+	stats := make([]adapter.CombinedCounter, len(v.Interfaces))
+	for _, workerStats := range counters {
+		for i, interfaceStats := range workerStats {
+			stats[i].Bytes += interfaceStats.Bytes
+			stats[i].Packets += interfaceStats.Packets
+		}
+	}
+	return &stats
+}
+
+func (v *vppConnector) reduceSimpleCounters(stat *adapter.StatEntry) *[]adapter.Counter {
+	counters := stat.Data.(adapter.SimpleCounterStat)
+	stats := make([]adapter.Counter, len(v.Interfaces))
+	for _, workerStats := range counters {
+		for i, interfaceStats := range workerStats {
+			stats[i] += interfaceStats
+		}
+	}
+	return &stats
+}
+
+func (v *vppConnector) getStatsForAllInterfaces() error {
+	statsDump, err := v.stats.DumpStats("/if")
+	if err != nil {
+		return fmt.Errorf("failed to dump vpp Stats: %v", err)
+	}
+
+	stats := func(i int) *interfaceStats { return &v.Interfaces[uint32(i)].Stats }
+
+	for _, stat := range statsDump {
+		switch stat.Name {
+		case "/if/tx":
+			{
+				for i, counter := range *v.reduceCombinedCounters(stat) {
+					stats(i).TxBytes = uint64(counter.Bytes)
+					stats(i).TxPackets = uint64(counter.Packets)
+				}
+			}
+		case "/if/rx":
+			{
+				for i, counter := range *v.reduceCombinedCounters(stat) {
+					stats(i).RxBytes = uint64(counter.Bytes)
+					stats(i).RxPackets = uint64(counter.Packets)
+				}
+			}
+		case "/if/tx-error":
+			{
+				for i, counter := range *v.reduceSimpleCounters(stat) {
+					stats(i).TxErrors = uint64(counter)
+				}
+			}
+		case "/if/rx-error":
+			{
+				for i, counter := range *v.reduceSimpleCounters(stat) {
+					stats(i).RxErrors = uint64(counter)
+				}
+			}
+		case "/if/drops":
+			{
+				for i, counter := range *v.reduceSimpleCounters(stat) {
+					stats(i).Drops = uint64(counter)
+				}
+			}
+		case "/if/punt":
+			{
+				for i, counter := range *v.reduceSimpleCounters(stat) {
+					stats(i).Punts = uint64(counter)
+				}
+			}
+		}
+	}
+	return nil
+}
+
+//////////////////////////////////////
+
+func main() {
+	statsSocketPathPtr := flag.String("stats_socket_path", defaultStatsSocketPath, "Path to vpp stats socket")
+	shmPrefixPtr := flag.String("shm_prefix", defaultShmPrefix, "Shared memory prefix (advanced)")
+	flag.Parse()
+
+	vppConn := &vppConnector{statsSocketPath: *statsSocketPathPtr, shmPrefix: *shmPrefixPtr}
+	defer vppConn.disconnect()
+
+	if err := vppConn.connect(); err != nil {
+		log.Fatalln(err)
+	}
+
+	if err := vppConn.getVppVersion(); err != nil {
+		log.Fatalln(err)
+	}
+
+	if err := vppConn.getInterfaces(); err != nil {
+		log.Fatalln(err)
+	}
+
+	if err := vppConn.getStatsForAllInterfaces(); err != nil {
+		log.Fatalln(err)
+	}
+
+	jsonString, err := dumpToJSONString(vppConn)
+	if err != nil {
+		log.Fatalln(err)
+	}
+	fmt.Println(jsonString)
+}