add new field to ipfix exporter for flowRtt

netobserv · Feb 20, 2024 · 4b4d3f9 · 4b4d3f9
1 parent a56fc72
commit 4b4d3f9
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/pkg/pipeline/write/write_ipfix.go b/pkg/pipeline/write/write_ipfix.go
@@ -85,6 +85,10 @@ func addKubeContextToTemplate(elements *[]entities.InfoElementWithValue, registr
 	if err != nil {
 		return err
 	}
+	err = addElementToTemplate("timeFlowRttNs", nil, elements, registryID)
+	if err != nil {
+		return err
+	}
 	return nil
 }
 
@@ -124,6 +128,11 @@ func loadCustomRegistry(EnterpriseID uint32) error {
 		ilog.WithError(err).Errorf("Failed to register element")
 		return err
 	}
+	err = registry.PutInfoElement((*entities.NewInfoElement("timeFlowRttNs", 7740, 13, EnterpriseID, 65535)), EnterpriseID)
+	if err != nil {
+		ilog.WithError(err).Errorf("Failed to register element")
+		return err
+	}
 	return nil
 }
 
@@ -445,7 +454,13 @@ func setKubeIEValue(record config.GenericMap, ieValPtr *entities.InfoElementWith
 		} else {
 			ieVal.SetStringValue("none")
 		}
-	}
+	case "timeFlowRttNs":
+        if record["TimeFlowRttNs"] != nil {
+			ieVal.SetUnsigned64Value(record["TimeFlowRttNs"].(uint64))
+        } else {
+            // nothing here for now
+        }
+    }
 }
 func setEntities(record config.GenericMap, enrichEnterpriseID uint32, elements *[]entities.InfoElementWithValue) error {
 	for _, ieVal := range *elements {