Add HW 3

homework/03-http-server/tests/logs.go

@@ -0,0 +1,94 @@
+package hw3test
+
+import (
+	"fmt"
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+	"os"
+	"sync"
+)
+
+func logLevelFromEnv() zap.AtomicLevel {
+	levelStr := os.Getenv("LOG_LEVEL")
+	if levelStr == "" {
+		return zap.NewAtomicLevelAt(zap.DebugLevel)
+	}
+
+	level, err := zap.ParseAtomicLevel(levelStr)
+	if err != nil {
+		panic(fmt.Errorf("failed to parse log level: %w", err))
+	}
+	return level
+}
+
+var global *zap.Logger
+var globalOnce sync.Once
+
+func Global() *zap.Logger {
+	if global == nil {
+		globalOnce.Do(func() {
+			encoder := zapcore.EncoderConfig{
+				// Keys can be anything except the empty string.
+				TimeKey:        "",
+				LevelKey:       "L",
+				NameKey:        "_",
+				CallerKey:      "C",
+				FunctionKey:    "F",
+				MessageKey:     "M",
+				StacktraceKey:  "S",
+				LineEnding:     zapcore.DefaultLineEnding,
+				EncodeLevel:    zapcore.CapitalColorLevelEncoder,
+				EncodeTime:     zapcore.ISO8601TimeEncoder,
+				EncodeDuration: zapcore.StringDurationEncoder,
+				EncodeCaller:   zapcore.ShortCallerEncoder,
+			}
+			cfg := zap.Config{
+				Level:             logLevelFromEnv(),
+				Development:       true,
+				DisableCaller:     true,
+				DisableStacktrace: true,
+				Encoding:          "console",
+				EncoderConfig:     encoder,
+				OutputPaths:       []string{"stderr"},
+				ErrorOutputPaths:  []string{"stderr"},
+			}
+			options := []zap.Option{
+				zap.AddCallerSkip(1),
+			}
+
+			var err error
+			global, err = cfg.Build(options...)
+			if err != nil {
+				panic(fmt.Errorf("failed to initialize global logger: %w", err))
+			}
+		})
+	}
+
+	return global
+}
+
+func Logger(t *TC) *zap.Logger {
+	logger := Global()
+	logger = logger.Named(t.Name())
+	return logger
+}
+
+func Debug(t *TC, msg string, args ...zap.Field) {
+	Logger(t).Debug(msg, args...)
+	_ = Logger(t).Sync()
+}
+
+func Info(t *TC, msg string, args ...zap.Field) {
+	Logger(t).Info(msg, args...)
+	_ = Logger(t).Sync()
+}
+
+func Warn(t *TC, msg string, args ...zap.Field) {
+	Logger(t).Warn(msg, args...)
+	_ = Logger(t).Sync()
+}
+
+func Error(t *TC, msg string, args ...zap.Field) {
+	Logger(t).Error(msg, args...)
+	_ = Logger(t).Sync()
+}