jdysya
97ce74bba5
- 在 VuePress 配置中启用 hotReload 功能，以便实时查看更改 - 重构日志功能，实现日志文件按天轮转 - 优化日志初始化过程，确保每天创建新的日志文件 - 添加检查，每次记录访问日志前确认是否需要轮转日志文件
90 lines
1.9 KiB
Go
90 lines
1.9 KiB
Go
package utils
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
Logger = logrus.New()
|
|
currentLogFile *os.File
|
|
currentLogDay string
|
|
)
|
|
|
|
func InitLogger() {
|
|
if err := os.MkdirAll("log", 0755); err != nil {
|
|
panic(fmt.Sprintf("创建日志目录失败: %v", err))
|
|
}
|
|
|
|
// 初始化日志文件
|
|
rotateLog()
|
|
}
|
|
|
|
// 添加新的rotateLog函数用于日志文件轮转
|
|
func rotateLog() {
|
|
today := time.Now().Format("2006-01-02")
|
|
|
|
// 如果是同一天的日志文件，直接返回
|
|
if today == currentLogDay && currentLogFile != nil {
|
|
return
|
|
}
|
|
|
|
// 关闭之前的日志文件
|
|
if currentLogFile != nil {
|
|
currentLogFile.Close()
|
|
}
|
|
|
|
// 创建新的日志文件
|
|
logFileName := filepath.Join("log", today+".log")
|
|
logFile, err := os.OpenFile(logFileName, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("打开日志文件失败: %v", err))
|
|
}
|
|
|
|
// 更新当前日志文件和日期
|
|
currentLogFile = logFile
|
|
currentLogDay = today
|
|
|
|
Logger.SetFormatter(&logrus.JSONFormatter{})
|
|
Logger.SetOutput(io.MultiWriter(os.Stdout, logFile))
|
|
}
|
|
|
|
func LogAccess(ip, path, method string, userInfo map[string]interface{}) {
|
|
// 检查是否需要轮转日志文件
|
|
rotateLog()
|
|
|
|
// 设置东八区时区
|
|
loc, err := time.LoadLocation("Asia/Shanghai")
|
|
if err != nil {
|
|
Logger.WithError(err).Error("Failed to load timezone")
|
|
return
|
|
}
|
|
|
|
// 获取当前时间并格式化为东八区时间，使用中国人的习惯格式
|
|
now := time.Now().In(loc)
|
|
timestamp := now.Format("2006-01-02 15:04:05")
|
|
|
|
// 构建日志字段
|
|
fields := logrus.Fields{
|
|
"ip": ip,
|
|
"path": path,
|
|
"method": method,
|
|
"timestamp": timestamp,
|
|
}
|
|
|
|
// 合并用户信息到日志字段中
|
|
if userInfo != nil {
|
|
for k, v := range userInfo {
|
|
fields[k] = v
|
|
}
|
|
}
|
|
|
|
// 记录日志
|
|
Logger.WithFields(fields).Info("Page accessed")
|
|
}
|