TextUnmarshaler for Options for env

By implementing the encoding.TextUnmarshaler, Options can be populated
by environment variables from env.

Author: oxzi

logging/config.go

@@ -5,12 +5,35 @@ import (
 	"github.com/pkg/errors"
 	"go.uber.org/zap/zapcore"
 	"os"
+	"strings"
 	"time"
 )
 
 // Options define child loggers with their desired log level.
 type Options map[string]zapcore.Level
 
+// UnmarshalText implements encoding.TextUnmarshaler to allow Options to be parsed by env.
+func (o *Options) UnmarshalText(text []byte) error {
+	optionsMap := make(map[string]zapcore.Level)
+
+	for _, entry := range strings.Split(string(text), ",") {
+		key, valueStr, found := strings.Cut(entry, ":")
+		if !found {
+			return fmt.Errorf("entry %q cannot be unmarshalled as an Option entry", entry)
+		}
+
+		valueLvl, err := zapcore.ParseLevel(valueStr)
+		if err != nil {
+			return fmt.Errorf("entry %q cannot be unmarshalled as level, %w", entry, err)
+		}
+
+		optionsMap[key] = valueLvl
+	}
+
+	*o = optionsMap
+	return nil
+}
+
 // Config defines Logger configuration.
 type Config struct {
 	// zapcore.Level at 0 is for info level.

logging/config_test.go

@@ -55,17 +55,9 @@ func TestConfig(t *testing.T) {
 			},
 		},
 		{
-			name: "options-as-ints",
-			opts: config.EnvOptions{Environment: map[string]string{"OPTIONS": "foo:-1,bar:0,buz:4"}},
-			expected: Config{
-				Output:   "console",
-				Interval: 20 * time.Second,
-				Options: map[string]zapcore.Level{
-					"foo": zapcore.DebugLevel,
-					"bar": zapcore.InfoLevel,
-					"buz": zapcore.PanicLevel,
-				},
-			},
+			name:  "options-invalid-levels",
+			opts:  config.EnvOptions{Environment: map[string]string{"OPTIONS": "foo:foo,bar:0"}},
+			error: true,
 		},
 	}