Fields as array

Author: calmh

main.go

@@ -10,28 +10,43 @@ import (
 	"time"
 )
 
-type recordMap map[string]interface{}
+type InterpretedRecord struct {
+	ExportTime uint32               `json:"exportTime"`
+	TemplateId uint16               `json:"templateId"`
+	Fields     []myInterpretedField `json:"fields"`
+}
+
+// Because we want to control JSON serialization
+type myInterpretedField struct {
+	Name         string      `json:"name"`
+	EnterpriseId uint32      `json:"enterprise,omitempty"`
+	FieldId      uint16      `json:"field"`
+	Value        interface{} `json:"value,omitempty"`
+	RawValue     []byte      `json:"raw,omitempty"`
+}
 
-func messagesGenerator(s *ipfix.Session) <-chan []recordMap {
-	c := make(chan []recordMap)
+func messagesGenerator(s *ipfix.Session) <-chan []InterpretedRecord {
+	c := make(chan []InterpretedRecord)
 
 	go func() {
 		for {
-			sets := make([]recordMap, 0, 32)
 			msg, err := s.ReadMessage()
 			if err != nil {
 				panic(err)
 			}
 
-			for _, record := range msg.DataRecords {
-				set := make(map[string]interface{})
-				set["templateId"] = record.TemplateId
-				set["exportTime"] = msg.Header.ExportTime
-				set["elements"] = s.Interpret(&record)
-				sets = append(sets, set)
+			irecs := make([]InterpretedRecord, len(msg.DataRecords))
+			for i, record := range msg.DataRecords {
+				ifs := s.Interpret(&record)
+				mfs := make([]myInterpretedField, len(ifs))
+				for i, iif := range ifs {
+					mfs[i] = myInterpretedField{iif.Name, iif.EnterpriseId, iif.FieldId, iif.Value, iif.RawValue}
+				}
+				ir := InterpretedRecord{msg.Header.ExportTime, record.TemplateId, mfs}
+				irecs[i] = ir
 			}
 
-			c <- sets
+			c <- irecs
 		}
 	}()
 
@@ -74,18 +89,18 @@ func main() {
 	tick := time.Tick(time.Duration(*statsIntv) * time.Second)
 	for {
 		select {
-		case sets := <-msgs:
+		case irecs := <-msgs:
 			if *messageStats {
-				accountMsgStats(sets)
+				accountMsgStats(irecs)
 			}
 
-			for _, set := range sets {
+			for _, rec := range irecs {
 				if *trafficStats {
-					accountTraffic(set)
+					accountTraffic(rec)
 				}
 
 				if *output {
-					bs, _ := json.Marshal(set)
+					bs, _ := json.Marshal(rec)
 					fmt.Println(string(bs))
 				}
 			}

msgstats.go

@@ -11,9 +11,9 @@ var (
 	msgs    int
 )
 
-func accountMsgStats(sets []recordMap) {
+func accountMsgStats(recs []InterpretedRecord) {
 	msgs++
-	records += len(sets)
+	records += len(recs)
 }
 
 type MsgStats struct {

trafficstats.go

@@ -16,15 +16,12 @@ type TrafStats struct {
 	OutKbps int
 }
 
-func accountTraffic(set recordMap) {
-	if elems, ok := set["elements"]; ok {
-		// Account the in/out traffic rate if we have Procera Networks vendor fields available.
-		elems := elems.(map[string]interface{})
-		if in, ok := elems["proceraIncomingOctets"]; ok {
-			inOctets += in.(uint64)
-		}
-		if out, ok := elems["proceraOutgoingOctets"]; ok {
-			outOctets += out.(uint64)
+func accountTraffic(rec InterpretedRecord) {
+	for _, f := range rec.Fields {
+		if f.Name == "proceraIncomingOctets" {
+			inOctets += f.Value.(uint64)
+		} else if f.Name == "proceraOutgoingOctets" {
+			outOctets += f.Value.(uint64)
 		}
 	}
 }