add netflow v5 flows counter for prometheus stats

sahejsingh · Aug 13, 2019 · 4f9cca0 · 4f9cca0
1 parent ee3bcf9
commit 4f9cca0
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/utils/nflegacy.go b/utils/nflegacy.go
@@ -39,14 +39,21 @@ func (s *StateNFLegacy) DecodeFlow(msg interface{}) error {
 		return err
 	}
 
-	switch msgDec.(type) {
+	switch msgDecConv := msgDec.(type) {
 	case netflowlegacy.PacketNetFlowV5:
 		NetFlowStats.With(
 			prometheus.Labels{
 				"router":  key,
 				"version": "5",
 			}).
 			Inc()
+		NetFlowSetStatsSum.With(
+			prometheus.Labels{
+				"router":  key,
+				"version": "5",
+				"type":    "DataFlowSet",
+			}).
+			Add(float64(msgDecConv.Count))
 	}
 
 	var flowMessageSet []*flowmessage.FlowMessage