summaryrefslogtreecommitdiffstats
path: root/examples/cmd/stats-client/stats_client.go
blob: f61f9755b4c6746a68aeb1c773f54d8831aab963 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
// Copyright (c) 2017 Cisco and/or its affiliates.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at:
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Binary stats-client is an example VPP management application that exercises the
// govpp API for interface counters together with asynchronous connection to VPP.
package main

import (
	"fmt"
	"log"
	"os"
	"os/signal"

	"git.fd.io/govpp.git"
	"git.fd.io/govpp.git/api"
	"git.fd.io/govpp.git/core"
	"git.fd.io/govpp.git/examples/bin_api/stats"
)

func main() {
	fmt.Println("Starting stats VPP client..")

	// async connect to VPP
	conn, statCh, err := govpp.AsyncConnect("")
	if err != nil {
		log.Fatalln("Error:", err)
	}
	defer conn.Disconnect()

	// create an API channel that will be used in the examples
	ch, err := conn.NewAPIChannel()
	if err != nil {
		log.Fatalln("Error:", err)
	}
	defer ch.Close()

	// create channel for Interrupt signal
	sigChan := make(chan os.Signal, 1)
	signal.Notify(sigChan, os.Interrupt)

	var notifChan chan api.Message
	var simpleSub api.SubscriptionCtx
	var combinedSub api.SubscriptionCtx

	// loop until Interrupt signal is received
loop:
	for {
		select {
		case connEvent := <-statCh:
			// VPP connection state change
			switch connEvent.State {
			case core.Connected:
				fmt.Println("VPP connected.")
				if notifChan == nil {
					simpleSub, combinedSub, notifChan = subscribeNotifications(ch)
				}
				requestStatistics(ch)

			case core.Disconnected:
				fmt.Println("VPP disconnected.")
			}

		case msg := <-notifChan:
			switch notif := msg.(type) {
			case *stats.VnetInterfaceSimpleCounters:
				// simple counter notification received
				processSimpleCounters(notif)
			case *stats.VnetInterfaceCombinedCounters:
				// combined counter notification received
				processCombinedCounters(notif)
			default:
				fmt.Println("Ignoring unknown VPP notification")
			}

		case <-sigChan:
			// interrupt received
			fmt.Println("Interrupt received, exiting.")
			break loop
		}
	}

	simpleSub.Unsubscribe()
	combinedSub.Unsubscribe()
}

// subscribeNotifications subscribes for interface counters notifications.
func subscribeNotifications(ch api.Channel) (api.SubscriptionCtx, api.SubscriptionCtx, chan api.Message) {
	notifChan := make(chan api.Message, 100)

	simpleSub, err := ch.SubscribeNotification(notifChan, &stats.VnetInterfaceSimpleCounters{})
	if err != nil {
		panic(err)
	}
	combinedSub, err := ch.SubscribeNotification(notifChan, &stats.VnetInterfaceCombinedCounters{})
	if err != nil {
		panic(err)
	}

	return simpleSub, combinedSub, notifChan
}

// requestStatistics requests interface counters notifications from VPP.
func requestStatistics(ch api.Channel) {
	if err := ch.SendRequest(&stats.WantStats{
		PID:           uint32(os.Getpid()),
		EnableDisable: 1,
	}).ReceiveReply(&stats.WantStatsReply{}); err != nil {
		panic(err)
	}
}

// processSimpleCounters processes simple counters received from VPP.
func processSimpleCounters(counters *stats.VnetInterfaceSimpleCounters) {
	fmt.Printf("SimpleCounters: %+v\n", counters)

	counterNames := []string{
		"Drop", "Punt",
		"IPv4", "IPv6",
		"RxNoBuf", "RxMiss",
		"RxError", "TxError",
		"MPLS",
	}

	for i := uint32(0); i < counters.Count; i++ {
		fmt.Printf("Interface '%d': %s = %d\n",
			counters.FirstSwIfIndex+i, counterNames[counters.VnetCounterType], counters.Data[i])
	}
}

// processCombinedCounters processes combined counters received from VPP.
func processCombinedCounters(counters *stats.VnetInterfaceCombinedCounters) {
	fmt.Printf("CombinedCounters: %+v\n", counters)

	counterNames := []string{"Rx", "Tx"}

	for i := uint32(0); i < counters.Count; i++ {
		if len(counterNames) <= int(counters.VnetCounterType) {
			continue
		}
		fmt.Printf("Interface '%d': %s packets = %d, %s bytes = %d\n",
			counters.FirstSwIfIndex+i,
			counterNames[counters.VnetCounterType], counters.Data[i].Packets,
			counterNames[counters.VnetCounterType], counters.Data[i].Bytes)
	}
}