Golang学习笔记-自定义日志轮转及输出

package pkg

import (
	"fmt"
	"log"
	"log/slog"
	"os"
	"time"
)

var controlLogger *slog.Logger
var fileLogger *slog.Logger

const (
	timeFormat = "2006-01-02"
)

func InitLog(filepath string) {
	options := slog.HandlerOptions{
		// 是否打印文件名称及行号
		AddSource: false,
		Level:     slog.LevelDebug,
		ReplaceAttr: func(groups []string, a slog.Attr) slog.Attr {
			if a.Key == slog.TimeKey {
				a.Value = slog.AnyValue(time.Now().Format("2006-01-02 15:04:05"))
			}
			return a
		},
	}
	// 打开文件
	file, err := os.OpenFile(filepath, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)
	if err != nil {
		fmt.Println(err)
	}
	// 文件写入
	fileLogger = slog.New(slog.NewJSONHandler(file, &options))
	// 控制台输出
	controlLogger = slog.New(slog.NewTextHandler(os.Stdout, &options))

	// 启动定时器，每天零点进行日志轮转
	startTimer(file, filepath)
}

// 计时器
func startTimer(fileObj *os.File, filepath string) {
	now := time.Now()
	//// 计算下一分钟的开始时间
	//next := now.Truncate(time.Minute).Add(time.Minute)
	// 计算下一个零点时间
	next := now.Add(time.Hour * 24)
	next = time.Date(next.Year(), next.Month(), next.Day(), 0, 0, 0, 0, next.Location())
	// 计算当前时间到下一个零点的时间间隔
	timer := time.NewTimer(next.Sub(now))
	// 每秒钟
	//timer := time.NewTicker(time.Second)

	go func() {
		<-timer.C
		rotateLogs(fileObj, filepath)
		startTimer(fileObj, filepath)
	}()
}

// 日志轮转函数
func rotateLogs(fileObj *os.File, filepath string) {
	// 关闭当前的日志文件
	fileObj.Close()

	// 生成新的日志文件名
	newPath := fmt.Sprintf("%s_%s", filepath, time.Now().Format(timeFormat))

	// 将当前日志文件重命名为新的文件名
	err := os.Rename(filepath, newPath)
	if err != nil {
		log.Fatal("无法重命名日志文件:", err)
	}

	// 创建一个新的日志文件
	writer, err := os.OpenFile(filepath, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)
	if err != nil {
		log.Fatal("无法打开日志文件:", err)
	}
	options := slog.HandlerOptions{
		// 是否打印文件名称及行号
		AddSource: false,
		Level:     slog.LevelDebug,
		ReplaceAttr: func(groups []string, a slog.Attr) slog.Attr {
			if a.Key == slog.TimeKey {
				a.Value = slog.AnyValue(time.Now().Format("2006-01-02 15:04:05"))
			}
			return a
		},
	}
	// 设置日志输出到新的日志文件
	fileLogger = slog.New(slog.NewJSONHandler(writer, &options))
	// 控制台输出
	controlLogger = slog.New(slog.NewTextHandler(os.Stdout, &options))
}

func Info(msg string, args ...any) {
	controlLogger.Info(msg, args...)
	fileLogger.Info(msg, args...)
}

func Error(msg string, args ...any) {
	controlLogger.Error(msg, args...)
	fileLogger.Error(msg, args...)
}

func Debug(msg string, args ...any) {
	controlLogger.Debug(msg, args...)
	fileLogger.Debug(msg, args...)
}

func Warn(msg string, args ...any) {
	controlLogger.Warn(msg, args...)
	fileLogger.Warn(msg, args...)
}