rotation.go

package log

import (
	"fmt"
	"os"
	"strings"
	"sync"
	"time"
)

var registerLogPaths sync.Map

func timeToLogFile() {
	registerLogPaths.Range(func(l, _ any) bool {
		logPath := strings.TrimSuffix(l.(string), ".log")
		os.RemoveAll(logPath + ".log")
		os.Symlink(fmt.Sprintf("%s-%s.log", logPath, time.Now().Format("2006-01-02")), logPath+".log")
		return true
	})
}

func scheduleLogRotation() {
	for {
		now := time.Now()
		// 计算明天的时间
		next := now.Add(24 * time.Hour)
		next = time.Date(next.Year(), next.Month(), next.Day(), 0, 1, 0, 0, next.Location()) // 设置为第二天的 00:01:00

		// 如果当前时间已经过了今天的00:01，则等待到明天
		if next.Before(now) {
			next = next.Add(24 * time.Hour)
		}

		duration := next.Sub(now)
		// fmt.Printf("下次日志轮转时间: %v\n", next.Format("2006-01-02 15:04:05"))

		// 等待到设定时间
		time.Sleep(duration)

		// 执行日志文件切换
		timeToLogFile()
	}
}

// 启动日志轮转
func StartLogRotation() {
	timeToLogFile()
	go scheduleLogRotation()
}