feat(logging): support custom time, level and message keys (#1295)

flipt-io · Jan 30, 2023 · 338c495 · 338c495
1 parent d0b365b
commit 338c495
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 0 deletions.
diff --git a/cmd/flipt/main.go b/cmd/flipt/main.go
@@ -164,6 +164,9 @@ func main() {
 		cfg = res.Config
 		cfgWarnings = res.Warnings
 
+		loggerConfig.EncoderConfig.TimeKey = cfg.Log.Keys.Time
+		loggerConfig.EncoderConfig.LevelKey = cfg.Log.Keys.Level
+		loggerConfig.EncoderConfig.MessageKey = cfg.Log.Keys.Message
 		// log to file if enabled
 		if cfg.Log.File != "" {
 			loggerConfig.OutputPaths = []string{cfg.Log.File}

diff --git a/internal/config/config_test.go b/internal/config/config_test.go
@@ -168,6 +168,11 @@ func defaultConfig() *Config {
 			Level:     "INFO",
 			Encoding:  LogEncodingConsole,
 			GRPCLevel: "ERROR",
+			Keys: LogKeys{
+				Time:    "T",
+				Level:   "L",
+				Message: "M",
+			},
 		},
 
 		UI: UIConfig{
@@ -424,6 +429,11 @@ func TestLoad(t *testing.T) {
 					File:      "testLogFile.txt",
 					Encoding:  LogEncodingJSON,
 					GRPCLevel: "ERROR",
+					Keys: LogKeys{
+						Time:    "time",
+						Level:   "level",
+						Message: "msg",
+					},
 				}
 				cfg.Cors = CorsConfig{
 					Enabled:        true,

diff --git a/internal/config/log.go b/internal/config/log.go
@@ -16,13 +16,25 @@ type LogConfig struct {
 	File      string      `json:"file,omitempty" mapstructure:"file"`
 	Encoding  LogEncoding `json:"encoding,omitempty" mapstructure:"encoding"`
 	GRPCLevel string      `json:"grpcLevel,omitempty" mapstructure:"grpc_level"`
+	Keys      LogKeys     `json:"keys" mapstructure:"keys"`
+}
+
+type LogKeys struct {
+	Time    string `json:"time" mapstructure:"time"`
+	Level   string `json:"level" mapstructure:"level"`
+	Message string `json:"message" mapstructure:"message"`
 }
 
 func (c *LogConfig) setDefaults(v *viper.Viper) {
 	v.SetDefault("log", map[string]any{
 		"level":      "INFO",
 		"encoding":   "console",
 		"grpc_level": "ERROR",
+		"keys": map[string]any{
+			"time":    "T",
+			"level":   "L",
+			"message": "M",
+		},
 	})
 }
 

diff --git a/internal/config/testdata/advanced.yml b/internal/config/testdata/advanced.yml
@@ -2,6 +2,10 @@ log:
   level: WARN
   file: "testLogFile.txt"
   encoding: "json"
+  keys:
+    time: "time"
+    level: "level"
+    message: "msg"
 
 cors:
   enabled: true