wip: refactory log with zap

pkg/log/default.go

@@ -1,67 +0,0 @@
-package log
-
-import (
-	"fmt"
-	"os"
-	"sync"
-)
-
-var (
-	nilLogger    = func(prefix, timestamp, msg string, data ...any) {}
-	normalLogger = func(prefix, timestamp, msg string, data ...any) {
-		fmt.Printf(prefix+"| "+timestamp+" | "+msg+"\n", data...)
-	}
-
-	panicLogger = func(prefix, timestamp, msg string, data ...any) {
-		panic(fmt.Sprintf(prefix+"| "+timestamp+" | "+msg+"\n", data...))
-	}
-
-	fatalLogger = func(prefix, timestamp, msg string, data ...any) {
-		fmt.Printf(prefix+"| "+timestamp+" | "+msg+"\n", data...)
-		os.Exit(1)
-	}
-
-	DefaultLogger = &logger{
-		Mutex:      sync.Mutex{},
-		timeFormat: "2006-01-02T15:04:05",
-		writer:     os.Stdout,
-		level:      LogLevelInfo,
-		debug:      nilLogger,
-		info:       normalLogger,
-		warn:       normalLogger,
-		error:      normalLogger,
-		panic:      panicLogger,
-		fatal:      fatalLogger,
-	}
-)
-
-func SetTimeFormat(format string) {
-	DefaultLogger.SetTimeFormat(format)
-}
-
-func SetLogLevel(level LogLevel) {
-	DefaultLogger.SetLogLevel(level)
-}
-
-func Debug(msg string, data ...any) {
-	DefaultLogger.Debug(msg, data...)
-}
-func Info(msg string, data ...any) {
-	DefaultLogger.Info(msg, data...)
-}
-
-func Warn(msg string, data ...any) {
-	DefaultLogger.Warn(msg, data...)
-}
-
-func Error(msg string, data ...any) {
-	DefaultLogger.Error(msg, data...)
-}
-
-func Panic(msg string, data ...any) {
-	DefaultLogger.Panic(msg, data...)
-}
-
-func Fatal(msg string, data ...any) {
-	DefaultLogger.Fatal(msg, data...)
-}

pkg/log/log.go

@@ -1,115 +1,36 @@
 package log
 
 import (
-	"github.com/fatih/color"
-	"io"
 	"sync"
-	"time"
+
+	"go.uber.org/zap"
 )
 
-type LogLevel uint32
-
-const (
-	LogLevelDebug = iota
-	LogLevelInfo
-	LogLevelWarn
-	LogLevelError
-	LogLevelPanic
-	LogLevelFatal
-)
-
-type logger struct {
-	sync.Mutex
-	timeFormat string
-	writer     io.Writer
-	level      LogLevel
-	debug      func(prefix, timestamp, msg string, data ...any)
-	info       func(prefix, timestamp, msg string, data ...any)
-	warn       func(prefix, timestamp, msg string, data ...any)
-	error      func(prefix, timestamp, msg string, data ...any)
-	panic      func(prefix, timestamp, msg string, data ...any)
-	fatal      func(prefix, timestamp, msg string, data ...any)
-}
-
 var (
-	red    = color.New(color.FgRed)
-	hired  = color.New(color.FgHiRed)
-	green  = color.New(color.FgGreen)
-	yellow = color.New(color.FgYellow)
-	white  = color.New(color.FgWhite)
+	Logger *zap.Logger
+	locker = &sync.Mutex{}
 )
 
-func (l *logger) SetTimeFormat(format string) {
-	l.Lock()
-	defer l.Unlock()
-	l.timeFormat = format
-}
-
-func (l *logger) SetLogLevel(level LogLevel) {
-	l.Lock()
-	defer l.Unlock()
-
-	if level > LogLevelDebug {
-		l.debug = nilLogger
-	} else {
-		l.debug = normalLogger
-	}
-
-	if level > LogLevelInfo {
-		l.info = nilLogger
-	} else {
-		l.info = normalLogger
-	}
-
-	if level > LogLevelWarn {
-		l.warn = nilLogger
-	} else {
-		l.warn = normalLogger
-	}
-
-	if level > LogLevelError {
-		l.error = nilLogger
-	} else {
-		l.error = normalLogger
-	}
-
-	if level > LogLevelPanic {
-		l.panic = nilLogger
-	} else {
-		l.panic = panicLogger
-	}
-
-	if level > LogLevelFatal {
-		l.fatal = nilLogger
-	} else {
-		l.fatal = fatalLogger
+func init() {
+	var err error
+	config := zap.NewProductionConfig()
+	Logger, err = config.Build()
+	if err != nil {
+		panic(err)
 	}
 }
 
-func (l *logger) Debug(msg string, data ...any) {
-	l.debug(white.Sprint("Debug "), time.Now().Format(l.timeFormat), msg, data...)
-}
+func Debug() {
+	var err error
 
-func (l *logger) Info(msg string, data ...any) {
-	l.info(green.Sprint("Info  "), time.Now().Format(l.timeFormat), msg, data...)
-}
+	config := zap.NewProductionConfig()
+	config.Level = zap.NewAtomicLevelAt(zap.DebugLevel)
 
-func (l *logger) Warn(msg string, data ...any) {
-	l.warn(yellow.Sprint("Warn  "), time.Now().Format(l.timeFormat), msg, data...)
-}
+	locker.Lock()
+	defer locker.Unlock()
 
-func (l *logger) Error(msg string, data ...any) {
-	l.error(red.Sprint("Error "), time.Now().Format(l.timeFormat), msg, data...)
-}
-
-func (l *logger) Panic(msg string, data ...any) {
-	l.panic(hired.Sprint("Panic "), time.Now().Format(l.timeFormat), msg, data...)
-}
-
-func (l *logger) Fatal(msg string, data ...any) {
-	l.fatal(hired.Sprint("Fatal "), time.Now().Format(l.timeFormat), msg, data...)
-}
-
-type WroteLogger interface {
-	Info(msg string, data ...any)
+	Logger, err = config.Build()
+	if err != nil {
+		panic(err)
+	}
 }

pkg/log/new.go

@@ -1,21 +0,0 @@
-package log
-
-import (
-	"os"
-	"sync"
-)
-
-func New() *logger {
-	return &logger{
-		Mutex:      sync.Mutex{},
-		timeFormat: "2006-01-02T15:04:05",
-		writer:     os.Stdout,
-		level:      LogLevelInfo,
-		debug:      nilLogger,
-		info:       normalLogger,
-		warn:       normalLogger,
-		error:      normalLogger,
-		panic:      panicLogger,
-		fatal:      fatalLogger,
-	}
-}