77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
|
package logger
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"log/slog"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
type LogService interface {
|
||
|
|
Error(msg string, args ...any)
|
||
|
|
Info(msg string, args ...any)
|
||
|
|
}
|
||
|
|
|
||
|
|
type loggers struct {
|
||
|
|
infoLogger *slog.Logger
|
||
|
|
errorLogger *slog.Logger
|
||
|
|
}
|
||
|
|
|
||
|
|
func SetupLogger(env string) (LogService, error) {
|
||
|
|
var infoHandler slog.Handler
|
||
|
|
var errorHandler slog.Handler
|
||
|
|
|
||
|
|
if env == "test" {
|
||
|
|
infoHandler = slog.NewTextHandler(io.Discard, &slog.HandlerOptions{Level: slog.LevelInfo})
|
||
|
|
errorHandler = slog.NewTextHandler(io.Discard, &slog.HandlerOptions{Level: slog.LevelError})
|
||
|
|
} else {
|
||
|
|
err := os.MkdirAll("logs", 0755)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
infoFile, err := os.OpenFile("logs/Info.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
|
||
|
|
if err != nil {
|
||
|
|
if os.IsNotExist(err) {
|
||
|
|
infoFile, err = os.Create("logs/Info.log")
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
errorFile, err := os.OpenFile("logs/Error.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
|
||
|
|
if err != nil {
|
||
|
|
infoFile.Close()
|
||
|
|
if os.IsNotExist(err) {
|
||
|
|
errorFile, err = os.Create("logs/Error.log")
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
infoHandler = slog.NewTextHandler(infoFile, &slog.HandlerOptions{Level: slog.LevelInfo})
|
||
|
|
errorHandler = slog.NewTextHandler(errorFile, &slog.HandlerOptions{Level: slog.LevelError})
|
||
|
|
}
|
||
|
|
|
||
|
|
infoLogger := slog.New(infoHandler)
|
||
|
|
errorLogger := slog.New(errorHandler)
|
||
|
|
|
||
|
|
return &loggers{
|
||
|
|
infoLogger: infoLogger,
|
||
|
|
errorLogger: errorLogger,
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *loggers) Error(msg string, args ...any) {
|
||
|
|
l.errorLogger.Error(msg, args)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *loggers) Info(msg string, args ...any) {
|
||
|
|
l.infoLogger.Info(msg, args)
|
||
|
|
}
|