smpp-otp/pkg/lib/logger/logger.go

45 lines
1.1 KiB
Go

package logger
import (
"io"
"log/slog"
"os"
)
type Loggers struct {
InfoLogger *slog.Logger
ErrorLogger *slog.Logger
}
func SetupLogger(env string) (*Loggers, 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 {
infoFile, err := os.OpenFile("logs/Info.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
if err != nil {
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()
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
}