364 lines
8.0 KiB
Go
364 lines
8.0 KiB
Go
package logger
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"io"
|
||
"log"
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
|
||
"git.apinb.com/bsm-sdk/core/conf"
|
||
"git.apinb.com/bsm-sdk/core/utils"
|
||
"git.apinb.com/bsm-sdk/core/vars"
|
||
)
|
||
|
||
// Logger 日志器结构
|
||
type Logger struct {
|
||
level vars.LogLevel
|
||
infoLogger *log.Logger
|
||
warnLogger *log.Logger
|
||
errorLogger *log.Logger
|
||
fatalLogger *log.Logger
|
||
debugLogger *log.Logger
|
||
|
||
fileWriter io.Writer
|
||
consoleWriter io.Writer
|
||
|
||
mu sync.RWMutex
|
||
name string
|
||
logDir string
|
||
currentDate string
|
||
onRemote bool
|
||
endpoint string
|
||
}
|
||
|
||
var (
|
||
globalLogger *Logger
|
||
once sync.Once
|
||
)
|
||
|
||
// 初始化Logger配置
|
||
func New(cfg *conf.LogConf) {
|
||
if cfg == nil {
|
||
cfg = &conf.LogConf{
|
||
Name: strings.ToLower(vars.ServiceKey),
|
||
Level: vars.LogLevel(vars.DEBUG),
|
||
Dir: "./logs/",
|
||
Endpoint: "",
|
||
Console: true,
|
||
File: true,
|
||
Remote: false,
|
||
}
|
||
}
|
||
|
||
InitLogger(cfg)
|
||
}
|
||
|
||
// InitLogger 初始化全局日志器
|
||
func InitLogger(cfg *conf.LogConf) error {
|
||
var err error
|
||
once.Do(func() {
|
||
globalLogger, err = NewLogger(cfg)
|
||
})
|
||
return err
|
||
}
|
||
|
||
// NewLogger 创建新的日志器
|
||
func NewLogger(cfg *conf.LogConf) (*Logger, error) {
|
||
// 确保日志目录存在
|
||
if err := os.MkdirAll(cfg.Dir, 0755); err != nil {
|
||
return nil, fmt.Errorf("创建日志目录失败: %v", err)
|
||
}
|
||
|
||
// 控制台输出
|
||
consoleWriter := os.Stdout
|
||
|
||
// 文件输出
|
||
fileWriter, err := createLogFile(cfg.Dir, cfg.Name)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("创建日志文件失败: %v", err)
|
||
}
|
||
|
||
// 创建多输出写入器
|
||
multiWriter := io.MultiWriter(consoleWriter, fileWriter)
|
||
|
||
logger := &Logger{
|
||
level: cfg.Level,
|
||
fileWriter: fileWriter,
|
||
consoleWriter: consoleWriter,
|
||
logDir: cfg.Dir,
|
||
name: strings.ToLower(cfg.Name),
|
||
currentDate: time.Now().Format("2006-01-02"),
|
||
onRemote: cfg.Remote,
|
||
endpoint: cfg.Endpoint,
|
||
}
|
||
|
||
// 创建不同级别的日志器
|
||
logger.infoLogger = log.New(multiWriter, "[INFO] ", log.LstdFlags)
|
||
logger.warnLogger = log.New(multiWriter, "[WARN] ", log.LstdFlags)
|
||
logger.errorLogger = log.New(multiWriter, "[ERROR] ", log.LstdFlags)
|
||
logger.fatalLogger = log.New(multiWriter, "[FATAL] ", log.LstdFlags)
|
||
logger.debugLogger = log.New(multiWriter, "[DEBUG] ", log.LstdFlags)
|
||
|
||
return logger, nil
|
||
}
|
||
|
||
// createLogFile 创建日志文件
|
||
func createLogFile(logDir, name string) (io.Writer, error) {
|
||
filename := fmt.Sprintf("%s_%s.log", name, time.Now().Format("2006-01-02"))
|
||
filepath := filepath.Join(logDir, filename)
|
||
|
||
file, err := os.OpenFile(filepath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return file, nil
|
||
}
|
||
|
||
// checkAndRotateLog 检查并轮转日志文件
|
||
func (l *Logger) checkAndRotateLog() error {
|
||
today := time.Now().Format("2006-01-02")
|
||
if l.currentDate != today {
|
||
l.mu.Lock()
|
||
defer l.mu.Unlock()
|
||
|
||
if l.currentDate != today {
|
||
// 关闭旧文件
|
||
if closer, ok := l.fileWriter.(io.Closer); ok {
|
||
closer.Close()
|
||
}
|
||
|
||
// 创建新文件
|
||
newFileWriter, err := createLogFile(l.logDir, l.name)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
l.fileWriter = newFileWriter
|
||
l.currentDate = today
|
||
|
||
// 重新创建多输出写入器
|
||
multiWriter := io.MultiWriter(l.consoleWriter, l.fileWriter)
|
||
|
||
l.infoLogger = log.New(multiWriter, "[INFO] ", log.LstdFlags)
|
||
l.warnLogger = log.New(multiWriter, "[WARN] ", log.LstdFlags)
|
||
l.errorLogger = log.New(multiWriter, "[ERROR] ", log.LstdFlags)
|
||
l.fatalLogger = log.New(multiWriter, "[FATAL] ", log.LstdFlags)
|
||
l.debugLogger = log.New(multiWriter, "[DEBUG] ", log.LstdFlags)
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (l *Logger) sendToRemote(level, name, out string) {
|
||
if l.endpoint == "" {
|
||
return
|
||
}
|
||
data := map[string]any{
|
||
"level": level,
|
||
"name": name,
|
||
"out": out,
|
||
}
|
||
jsonBytes, _ := json.Marshal(data)
|
||
utils.HttpPost(l.endpoint, nil, jsonBytes)
|
||
}
|
||
|
||
func (l *Logger) loggerFor(level vars.LogLevel) (*log.Logger, string) {
|
||
switch level {
|
||
case vars.DEBUG:
|
||
return l.debugLogger, "DEBUG"
|
||
case vars.INFO:
|
||
return l.infoLogger, "INFO"
|
||
case vars.WARN:
|
||
return l.warnLogger, "WARN"
|
||
case vars.ERROR:
|
||
return l.errorLogger, "ERROR"
|
||
default:
|
||
return l.fatalLogger, "FATAL"
|
||
}
|
||
}
|
||
|
||
func (l *Logger) output(level vars.LogLevel, out string, fatal bool) {
|
||
if !fatal && l.level > level {
|
||
return
|
||
}
|
||
|
||
_ = l.checkAndRotateLog()
|
||
logger, levelName := l.loggerFor(level)
|
||
if fatal {
|
||
logger = l.fatalLogger
|
||
levelName = "FATAL"
|
||
}
|
||
if l.onRemote {
|
||
go l.sendToRemote(levelName, l.name, out)
|
||
}
|
||
_ = logger.Output(3, out)
|
||
|
||
if fatal {
|
||
os.Exit(1)
|
||
}
|
||
}
|
||
|
||
func (l *Logger) outputf(level vars.LogLevel, format string, v ...any) {
|
||
l.output(level, fmt.Sprintf(format, v...), false)
|
||
}
|
||
|
||
// Debug 输出调试信息
|
||
func (l *Logger) Debug(v ...any) {
|
||
l.output(vars.DEBUG, fmt.Sprint(v...), false)
|
||
}
|
||
|
||
// Debugf 格式化输出调试信息
|
||
func (l *Logger) Debugf(format string, v ...any) {
|
||
l.outputf(vars.DEBUG, format, v...)
|
||
}
|
||
|
||
// Info 输出信息
|
||
func (l *Logger) Info(v ...any) {
|
||
l.output(vars.INFO, fmt.Sprint(v...), false)
|
||
}
|
||
|
||
// Infof 格式化输出信息
|
||
func (l *Logger) Infof(format string, v ...any) {
|
||
l.outputf(vars.INFO, format, v...)
|
||
}
|
||
|
||
// Warn 输出警告
|
||
func (l *Logger) Warn(v ...any) {
|
||
l.output(vars.WARN, fmt.Sprint(v...), false)
|
||
}
|
||
|
||
// Warnf 格式化输出警告
|
||
func (l *Logger) Warnf(format string, v ...any) {
|
||
l.outputf(vars.WARN, format, v...)
|
||
}
|
||
|
||
// Error 输出错误
|
||
func (l *Logger) Error(v ...any) {
|
||
l.output(vars.ERROR, fmt.Sprint(v...), false)
|
||
}
|
||
|
||
// Errorf 格式化输出错误
|
||
func (l *Logger) Errorf(format string, v ...any) {
|
||
l.outputf(vars.ERROR, format, v...)
|
||
}
|
||
|
||
// Fatal 输出致命错误并退出程序
|
||
func (l *Logger) Fatal(v ...any) {
|
||
l.output(vars.ERROR, fmt.Sprint(v...), true)
|
||
}
|
||
|
||
// Fatalf 格式化输出致命错误并退出程序
|
||
func (l *Logger) Fatalf(format string, v ...any) {
|
||
l.output(vars.ERROR, fmt.Sprintf(format, v...), true)
|
||
}
|
||
|
||
// Print 输出信息（兼容标准log包）
|
||
func (l *Logger) Print(v ...any) {
|
||
l.Info(v...)
|
||
}
|
||
|
||
// Printf 格式化输出信息（兼容标准log包）
|
||
func (l *Logger) Printf(format string, v ...any) {
|
||
l.Infof(format, v...)
|
||
}
|
||
|
||
// Println 输出信息并换行（兼容标准log包）
|
||
func (l *Logger) Println(v ...any) {
|
||
l.Info(v...)
|
||
}
|
||
|
||
// SetLevel 设置日志级别
|
||
func (l *Logger) SetLevel(level vars.LogLevel) {
|
||
l.mu.Lock()
|
||
defer l.mu.Unlock()
|
||
l.level = level
|
||
}
|
||
|
||
// GetLevel 获取日志级别
|
||
func (l *Logger) GetLevel() vars.LogLevel {
|
||
l.mu.RLock()
|
||
defer l.mu.RUnlock()
|
||
return l.level
|
||
}
|
||
|
||
// Close 关闭日志器
|
||
func (l *Logger) Close() error {
|
||
if closer, ok := l.fileWriter.(io.Closer); ok {
|
||
return closer.Close()
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 全局日志函数（兼容标准log包）
|
||
|
||
// Global logger functions.
|
||
|
||
func withGlobalLogger(fn func(*Logger)) {
|
||
if globalLogger != nil {
|
||
fn(globalLogger)
|
||
}
|
||
}
|
||
|
||
func Debug(v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Debug(v...) })
|
||
}
|
||
|
||
func Debugf(format string, v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Debugf(format, v...) })
|
||
}
|
||
|
||
func Info(v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Info(v...) })
|
||
}
|
||
|
||
func Infof(format string, v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Infof(format, v...) })
|
||
}
|
||
|
||
func Warn(v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Warn(v...) })
|
||
}
|
||
|
||
func Warnf(format string, v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Warnf(format, v...) })
|
||
}
|
||
|
||
func Error(v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Error(v...) })
|
||
}
|
||
|
||
func Errorf(format string, v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Errorf(format, v...) })
|
||
}
|
||
|
||
func Fatal(v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Fatal(v...) })
|
||
}
|
||
|
||
func Fatalf(format string, v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Fatalf(format, v...) })
|
||
}
|
||
|
||
func Print(v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Print(v...) })
|
||
}
|
||
|
||
func Printf(format string, v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Printf(format, v...) })
|
||
}
|
||
|
||
func Println(v ...any) {
|
||
withGlobalLogger(func(l *Logger) { l.Println(v...) })
|
||
}
|
||
|
||
// GetLogger 获取全局日志器实例
|
||
func GetLogger() *Logger {
|
||
return globalLogger
|
||
}
|