feat: Support ts as time field (#27)

hedhyw · Sep 8, 2023 · 40a5d31 · 40a5d31
1 parent aa8e0d3
commit 40a5d31
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/internal/pkg/source/entry_test.go b/internal/pkg/source/entry_test.go
@@ -44,6 +44,14 @@ func TestParseLogEntry(t *testing.T) {
 			assert.Equal(t, "1", logEntry.Time)
 		},
 	}, {
+		Name: "ts_number",
+		JSON: `{"ts":1}`,
+		Assert: func(tb testing.TB, logEntry source.LogEntry) {
+			tb.Helper()
+
+			assert.Equal(t, "1", logEntry.Time)
+		},
+	}, {
 		Name: "time_text",
 		JSON: `{"time":"1970-01-01T00:00:00.00"}`,
 		Assert: func(tb testing.TB, logEntry source.LogEntry) {
@@ -60,6 +68,14 @@ func TestParseLogEntry(t *testing.T) {
 			assert.Equal(t, "1970-01-01T00:00:00.00", logEntry.Time)
 		},
 	}, {
+		Name: "ts_text",
+		JSON: `{"ts":"1970-01-01T00:00:00.00"}`,
+		Assert: func(tb testing.TB, logEntry source.LogEntry) {
+			tb.Helper()
+
+			assert.Equal(t, "1970-01-01T00:00:00.00", logEntry.Time)
+		},
+	}, {
 		Name: "message",
 		JSON: `{"message":"message"}`,
 		Assert: func(tb testing.TB, logEntry source.LogEntry) {

diff --git a/internal/pkg/source/helper.go b/internal/pkg/source/helper.go
@@ -8,7 +8,7 @@ import (
 )
 
 func extractTime(value *fastjson.Value) string {
-	timeValue := extractValue(value, "timestamp", "time", "t")
+	timeValue := extractValue(value, "timestamp", "time", "t", "ts")
 	if timeValue != "" {
 		return formatMessage(strings.TrimSpace(timeValue))
 	}