Merge pull request cloudflare#37 from psvmcc/netflow_v5_flows_counter

Extend netflow v5 flows metrics
sahejsingh · Aug 17, 2019 · 04c3ca1 · 04c3ca1
2 parents ee3bcf9 + 4f9cca0
commit 04c3ca1
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