Added TimeEncoder flag and helper funcs

Author: bharathi-tenneti

pkg/log/zap/flags.go

@@ -41,6 +41,10 @@ var levelStrings = map[string]zapcore.Level{
 	"fatal":  zap.FatalLevel,
 }
 
+type encoderConfigFunc func(*zapcore.EncoderConfig)
+
+var ecfs []encoderConfigFunc
+
 type encoderFlag struct {
 	setFunc func(zapcore.Encoder)
 	value   string
@@ -60,26 +64,69 @@ func (ev *encoderFlag) Set(flagValue string) error {
 	val := strings.ToLower(flagValue)
 	switch val {
 	case "json":
-		ev.setFunc(newJSONEncoder())
+		ev.setFunc(newJSONEncoder(ecfs))
 	case "console":
-		ev.setFunc(newConsoleEncoder())
+		ev.setFunc(newConsoleEncoder(ecfs))
 	default:
 		return fmt.Errorf("invalid encoder value \"%s\"", flagValue)
 	}
 	ev.value = flagValue
 	return nil
 }
 
-func newJSONEncoder() zapcore.Encoder {
+func newJSONEncoder(ecfs []encoderConfigFunc) zapcore.Encoder {
 	encoderConfig := zap.NewProductionEncoderConfig()
+	for _, f := range ecfs {
+		f(&encoderConfig)
+	}
 	return zapcore.NewJSONEncoder(encoderConfig)
 }
 
-func newConsoleEncoder() zapcore.Encoder {
+func newConsoleEncoder(ecfs []encoderConfigFunc) zapcore.Encoder {
 	encoderConfig := zap.NewDevelopmentEncoderConfig()
+	for _, f := range ecfs {
+		f(&encoderConfig)
+	}
 	return zapcore.NewConsoleEncoder(encoderConfig)
 }
 
+type timeEncoderFlag struct {
+	setFunc func(zapcore.TimeEncoder)
+	value   string
+}
+
+var _ pflag.Value = &timeEncoderFlag{}
+
+func (ev *timeEncoderFlag) String() string {
+	return ev.value
+}
+
+func (ev *timeEncoderFlag) Type() string {
+	return "timeEncoder"
+}
+
+func (ev *timeEncoderFlag) Set(flagValue string) error {
+	val := strings.ToLower(flagValue)
+	var timeEnc zapcore.TimeEncoder
+	switch val {
+	case "iso8601", "rfc3339", "rfc3339nano", "millis", "nanos", "epoch":
+		if err := timeEnc.UnmarshalText([]byte(val)); err != nil {
+			return fmt.Errorf("not able to unmarshal time encoder value \"%s\"", val)
+		}
+		ev.setFunc(timeEnc)
+	default:
+		return fmt.Errorf("invalid time encoder value \"%s\"", flagValue)
+	}
+	ev.value = flagValue
+	return nil
+}
+
+func withTimeEncoding(te zapcore.TimeEncoder) encoderConfigFunc {
+	return func(ec *zapcore.EncoderConfig) {
+		ec.EncodeTime = te
+	}
+}
+
 type levelFlag struct {
 	setFunc func(zap.AtomicLevel)
 	value   string

pkg/log/zap/zap.go

@@ -30,6 +30,8 @@ import (
 	"go.uber.org/zap/zapcore"
 )
 
+var encVal encoderFlag
+
 // New returns a brand new Logger configured with Opts. It
 // uses KubeAwareEncoder which adds Type information and
 // Namespace/Name to the log.
@@ -132,6 +134,8 @@ type Options struct {
 	// Encoder configures how Zap will encode the output.  Defaults to
 	// console when Development is true and JSON otherwise
 	Encoder zapcore.Encoder
+	// TimeEncoder configures the zap time format. Defaults to epoch.
+	TimeEncoder zapcore.TimeEncoder
 	// DestWritter controls the destination of the log output.  Defaults to
 	// os.Stderr.
 	DestWritter io.Writer
@@ -155,8 +159,10 @@ func (o *Options) addDefaults() {
 
 	if o.Development {
 		if o.Encoder == nil {
-			encCfg := zap.NewDevelopmentEncoderConfig()
-			o.Encoder = zapcore.NewConsoleEncoder(encCfg)
+			if o.TimeEncoder != nil {
+				ecfs = append(ecfs, withTimeEncoding(o.TimeEncoder))
+			}
+			o.Encoder = newConsoleEncoder(ecfs)
 		}
 		if o.Level == nil {
 			lvl := zap.NewAtomicLevelAt(zap.DebugLevel)
@@ -170,8 +176,10 @@ func (o *Options) addDefaults() {
 
 	} else {
 		if o.Encoder == nil {
-			encCfg := zap.NewProductionEncoderConfig()
-			o.Encoder = zapcore.NewJSONEncoder(encCfg)
+			if o.TimeEncoder != nil {
+				ecfs = append(ecfs, withTimeEncoding(o.TimeEncoder))
+			}
+			o.Encoder = newJSONEncoder(ecfs)
 		}
 		if o.Level == nil {
 			lvl := zap.NewAtomicLevelAt(zap.InfoLevel)
@@ -223,8 +231,14 @@ func (o *Options) BindFlags(fs *flag.FlagSet) {
 		"Development Mode defaults(encoder=consoleEncoder,logLevel=Debug,stackTraceLevel=Warn). "+
 			"Production Mode defaults(encoder=jsonEncoder,logLevel=Info,stackTraceLevel=Error)")
 
+	// Set TimeEncoder value
+	var timeEncVal timeEncoderFlag
+	timeEncVal.setFunc = func(fromFlag zapcore.TimeEncoder) {
+		o.TimeEncoder = fromFlag
+	}
+	fs.Var(&timeEncVal, "zap-time-encoder", "Zap time encoding format('rfc3339', 'rfc339nanos', 'epoch', 'millis', 'nano', or 'iso8601')")
+
 	// Set Encoder value
-	var encVal encoderFlag
 	encVal.setFunc = func(fromFlag zapcore.Encoder) {
 		o.Encoder = fromFlag
 	}
@@ -255,6 +269,14 @@ func (o *Options) BindFlags(fs *flag.FlagSet) {
 func UseFlagOptions(in *Options) Opts {
 	return func(o *Options) {
 		*o = *in
+		if o.Encoder != nil && o.TimeEncoder != nil {
+			ecfs = append(ecfs, withTimeEncoding(o.TimeEncoder))
+			if encVal.value == "json" {
+				o.Encoder = newJSONEncoder(ecfs)
+			} else {
+				o.Encoder = newConsoleEncoder(ecfs)
+			}
+		}
 		o.addDefaults()
 	}
 }