mirror/netsampler-goflow2
1
0
Fork 0
mirror of https://github.com/netsampler/goflow2.git synced 2024-05-06 15:54:52 +00:00
Code Issues Projects Releases Wiki Activity

producer: add datetime render (#225)

Browse Source
This commit is contained in:
Louis
2023-12-06 14:03:45 +01:00
committed by GitHub
parent 5ba02103eb
commit da044ac77a
2 changed files with 47 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/goflow2/mapping.yaml
View File

@ -43,6 +43,8 @@ formatter:
- name: csum
index: 999
type: varint
render:
time_received_ns: datetimenano
# Decoder mappings
ipfix:
mapping:

33
producer/proto/render.go
View File

@ -5,6 +5,7 @@ import (
"encoding/hex"
"net"
"net/netip"
"time"
)
type RenderFunc func(msg *ProtoProducerMessage, fieldName string, data interface{}) interface{}
@ -19,6 +20,8 @@ const (
RendererProto RendererID = "proto"
RendererType RendererID = "type"
RendererNetwork RendererID = "network"
RendererDateTime RendererID = "datetime"
RendererDateTimeNano RendererID = "datetimenano"
)
var (
@ -28,6 +31,8 @@ var (
RendererMac: MacRenderer,
RendererEtype: EtypeRenderer,
RendererProto: ProtoRenderer,
RendererDateTime: DateTimeRenderer,
RendererDateTimeNano: DateTimeNanoRenderer,
}
defaultRenderers = map[string]RenderFunc{
@ -90,6 +95,34 @@ func NilRenderer(msg *ProtoProducerMessage, fieldName string, data interface{})
return data
}
func DateTimeRenderer(msg *ProtoProducerMessage, fieldName string, data interface{}) interface{} {
if dataC, ok := data.(uint64); ok {
ts := time.Unix(int64(dataC), 0).UTC()
return ts.Format(time.RFC3339Nano)
} else if dataC, ok := data.(int64); ok {
ts := time.Unix(dataC, 0).UTC()
return ts.Format(time.RFC3339Nano)
} else if dataC, ok := data.(uint32); ok {
ts := time.Unix(int64(dataC), 0).UTC()
return ts.Format(time.RFC3339Nano)
} else if dataC, ok := data.(int32); ok {
ts := time.Unix(int64(dataC), 0).UTC()
return ts.Format(time.RFC3339Nano)
}
return NilRenderer(msg, fieldName, data)
}
func DateTimeNanoRenderer(msg *ProtoProducerMessage, fieldName string, data interface{}) interface{} {
if dataC, ok := data.(uint64); ok {
ts := time.Unix(int64(dataC)/1e9, int64(dataC)%1e9).UTC()
return ts.Format(time.RFC3339Nano)
} else if dataC, ok := data.(int64); ok {
ts := time.Unix(dataC/1e9, dataC%1e9).UTC()
return ts.Format(time.RFC3339Nano)
}
return NilRenderer(msg, fieldName, data)
}
func MacRenderer(msg *ProtoProducerMessage, fieldName string, data interface{}) interface{} {
if dataC, ok := data.(uint64); ok {
var mac [8]byte