完善日志输出配置

server/server.go

@@ -12,6 +12,9 @@ import (
 	"sync"
 	"syscall"
 	"time"
+
+	"github.com/lmittmann/tint"
+	"github.com/mattn/go-colorable"
 )
 
 type Context struct {
@@ -75,23 +78,25 @@ func Start() {
 }
 
 func initLog() {
-	// writer := colorable.NewColorable(os.Stdout)
-	// logger := slog.New(tint.NewHandler(writer, &tint.Options{
-	// 	Level:      slog.LevelDebug,
-	// 	TimeFormat: time.RFC3339,
-	// 	ReplaceAttr: func(_ []string, attr slog.Attr) slog.Attr {
-	// 		err, ok := attr.Value.Any().(error)
-	// 		if !ok {
-	// 			return attr
-	// 		}
-	// 		return tint.Err(err)
-	// 	},
-	// }))
-	// slog.SetDefault(logger)
-	handler := slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
-		Level: slog.LevelDebug,
-	})
-	slog.SetDefault(slog.New(handler))
+	switch env.AppLogMode {
+	case "dev":
+		writer := colorable.NewColorable(os.Stdout)
+		logger := slog.New(tint.NewHandler(writer, &tint.Options{
+			Level:      slog.LevelDebug,
+			TimeFormat: time.RFC3339,
+			ReplaceAttr: func(_ []string, attr slog.Attr) slog.Attr {
+				err, ok := attr.Value.Any().(error)
+				if ok {
+					return tint.Err(err)
+				}
+				return attr
+			},
+		}))
+		slog.SetDefault(logger)
+	case "test":
+		logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{}))
+		slog.SetDefault(logger)
+	}
 }
 
 func startFwdServer(ctx context.Context) error {