add TimeNow in the exprlib helpers (#756)

* add TimeNow in the exprlib helpers
* add a default date when none is detected: when no date is recognised by ParseDate, then use time.Now()
This commit is contained in:
registergoofy 2021-04-16 19:13:48 +02:00 committed by GitHub
parent 88e1095478
commit 7e9ce901a4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 60 additions and 24 deletions

View file

@ -60,3 +60,9 @@ Returns the uppercase version of the string
Returns true if the IP `IPStr` is contained in the IP range `RangeStr` (uses `net.ParseCIDR`) Returns true if the IP `IPStr` is contained in the IP range `RangeStr` (uses `net.ParseCIDR`)
> IpInRange("1.2.3.4", "1.2.3.0/24") > IpInRange("1.2.3.4", "1.2.3.0/24")
## `TimeNow() string`
Return RFC3339 formatted time
> TimeNow()

View file

@ -9,6 +9,7 @@ import (
"regexp" "regexp"
"strconv" "strconv"
"strings" "strings"
"time"
"github.com/davecgh/go-spew/spew" "github.com/davecgh/go-spew/spew"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
@ -39,6 +40,7 @@ func GetExprEnv(ctx map[string]interface{}) map[string]interface{} {
"RegexpInFile": RegexpInFile, "RegexpInFile": RegexpInFile,
"Upper": Upper, "Upper": Upper,
"IpInRange": IpInRange, "IpInRange": IpInRange,
"TimeNow": TimeNow,
} }
for k, v := range ctx { for k, v := range ctx {
ExprLib[k] = v ExprLib[k] = v
@ -134,3 +136,7 @@ func IpInRange(ip string, ipRange string) bool {
} }
return false return false
} }
func TimeNow() string {
return time.Now().Format(time.RFC3339)
}

View file

@ -2,6 +2,7 @@ package exprhelpers
import ( import (
"fmt" "fmt"
"time"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
@ -367,8 +368,20 @@ func TestUpper(t *testing.T) {
expectedStr := "TEST" expectedStr := "TEST"
if Upper(testStr) != expectedStr { if Upper(testStr) != expectedStr {
t.Fatalf("Upper() should returned 1.5 as a float") t.Fatalf("Upper() should returned test in upper case")
} }
log.Printf("test 'Upper()' : OK") log.Printf("test 'Upper()' : OK")
} }
func TestTimeNow(t *testing.T) {
ti, err := time.Parse(time.RFC3339, TimeNow())
if err != nil {
t.Fatalf("Error parsing the return value of TimeNow: %s", err)
}
if -1*time.Until(ti) > time.Second {
t.Fatalf("TimeNow func should return time.Now()")
}
log.Printf("test 'TimeNow()' : OK")
}

View file

@ -50,8 +50,8 @@ func Loadplugin(path string) ([]EnricherCtx, error) {
} }
func GenDateParse(date string) (string, time.Time) { func GenDateParse(date string) (string, time.Time) {
var retstr string var (
var layouts = [...]string{ layouts = [...]string{
time.RFC3339, time.RFC3339,
"02/Jan/2006:15:04:05 -0700", "02/Jan/2006:15:04:05 -0700",
"Mon Jan 2 15:04:05 2006", "Mon Jan 2 15:04:05 2006",
@ -69,6 +69,7 @@ func GenDateParse(date string) (string, time.Time) {
"2006/01/02 15:04:05", "2006/01/02 15:04:05",
"2006-01-02 15:04:05", "2006-01-02 15:04:05",
} }
)
for _, dateFormat := range layouts { for _, dateFormat := range layouts {
t, err := time.Parse(dateFormat, date) t, err := time.Parse(dateFormat, date)
@ -85,17 +86,24 @@ func GenDateParse(date string) (string, time.Time) {
return string(retstr), t return string(retstr), t
} }
} }
return retstr, time.Time{}
now := time.Now()
retstr, err := now.MarshalText()
if err != nil {
log.Warningf("Failed marshaling current time")
return "", time.Time{}
}
return string(retstr), now
} }
func ParseDate(in string, p *types.Event, x interface{}) (map[string]string, error) { func ParseDate(in string, p *types.Event, x interface{}) (map[string]string, error) {
var ret map[string]string = make(map[string]string) var ret map[string]string = make(map[string]string)
tstr, tbin := GenDateParse(in) tstr, tbin := GenDateParse(in)
if !tbin.IsZero() { if !tbin.IsZero() {
ret["MarshaledTime"] = string(tstr) ret["MarshaledTime"] = string(tstr)
return ret, nil return ret, nil
} }
return nil, nil return nil, nil
} }

View file

@ -133,9 +133,12 @@ func (n *Node) ProcessStatics(statics []types.ExtraField, event *types.Event) er
} }
if value == "" { if value == "" {
//allow ParseDate to have empty input
if static.Method != "ParseDate" {
clog.Debugf("Empty value for %s, skip.", printStaticTarget(static)) clog.Debugf("Empty value for %s, skip.", printStaticTarget(static))
continue continue
} }
}
if static.Method != "" { if static.Method != "" {
processed := false processed := false