Go语言自定义日志库实现：按小时滚动与多级别控制

ylog.SetLevel(LevelInfo)
ylog.Debug("I am a debug log.")
ylog.Info("I am a Info log.")

type FileLogger struct {
	lastHour int64       // 记录创建日志文件时的小时数，用于检测是否需要切换文件
	file     *os.File    // 当前的日志文件实例
	Level    LogLevel    // 当前生效的日志级别
	mu       sync.Mutex  // 互斥体，防止多goroutine并发写日志导致的重复创建问题
	iLogger  *log.Logger // 内嵌的标准日志库实例
	Path     string      // 日志保存的最上层目录路径

type LogLevel int
const (
	LevelTrace LogLevel = iota
	LevelDebug
	LevelInfo
	LevelWarn
	LevelError
	LevelFatal
)

func (l *FileLogger) CanInfo() bool {
	return l.Level <= LevelInfo 
}
func (l *FileLogger) Info(v ...any) {
	if l.CanInfo() {                // 1. 判断是否达到输出级别
		l.ensureFile()               // 2. 确保日志文件已就绪
		v = append([]any{"Info "}, v...) // 3. 将级别前缀加入消息
		l.iLogger.Output(2, fmt.Sprintln(v...)) // 4. 调用底层Output方法输出

func (l *FileLogger) ensureFile() (err error) {
	currentTime := time.Now()
	if l.file == nil {            // 首次启动或实例刚初始化
		l.mu.Lock()
		defer l.mu.Unlock()
		if l.file == nil {          // 双重检查锁模式
			l.file, err = createFile(&l.Path, &currentTime)
			l.iLogger.SetOutput(l.file)
			l.iLogger.SetFlags(log.Lshortfile | log.Ldate | log.Ltime | log.Lmicroseconds)
			l.lastHour = getTimeHour(&currentTime)
		}
		return
	}
	// 如果当前时间与上次创建文件的时间所在小时不同
	currentHour := getTimeHour(&currentTime)
	if l.lastHour != currentHour {
		l.mu.Lock()
		defer l.mu.Unlock()
		if l.lastHour != currentHour { // 再次确认未发生并发变更
			_ = l.file.Close()                   // 关闭旧文件
			l.file, err = createFile(&l.Path, &currentTime) // 创建新小时文件
			l.iLogger.SetOutput(l.file)
			l.iLogger.SetFlags(log.Llongfile | log.Ldate | log.Ltime)
			l.lastHour = currentHour             // 更新时钟状态
		}
	}
	return
}

var stdPath = "logs"
var std = NewFileLogger(LevelInfo, stdPath)

func Trace(v ...any) {
	if std.CanTrace() {
		std.ensureFile()
		v = append([]any{"Trace"}, v...)
		std.iLogger.Output(2, fmt.Sprintln(v...))
	}
}

ylog.SetLevel(LevelInfo)
ylog.Debug("I am a debug log.")
ylog.Info("I am a Info log.")