BBIT-Kai
98 lines
1.8 KiB
Go
98 lines
1.8 KiB
Go
package log
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
)
|
|
|
|
var logDir = "./logs" // 日志目录，可根据需要修改
|
|
|
|
// 初始化日志目录
|
|
func Init(dir string) {
|
|
if dir != "" {
|
|
logDir = dir
|
|
}
|
|
if err := os.MkdirAll(logDir, 0755); err != nil {
|
|
fmt.Println("create log dir failed:", err)
|
|
}
|
|
cleanupOldLogs()
|
|
}
|
|
|
|
// Cleanup 删除超过7天的日志文件
|
|
func cleanupOldLogs() {
|
|
files, err := os.ReadDir(logDir)
|
|
if err != nil {
|
|
return
|
|
}
|
|
cutoff := time.Now().AddDate(0, 0, -7)
|
|
for _, f := range files {
|
|
if f.IsDir() {
|
|
continue
|
|
}
|
|
info, err := f.Info()
|
|
if err != nil {
|
|
continue
|
|
}
|
|
if info.ModTime().Before(cutoff) {
|
|
_ = os.Remove(filepath.Join(logDir, f.Name()))
|
|
}
|
|
}
|
|
}
|
|
|
|
// log 内部写文件
|
|
func logToFile(level, msg string) {
|
|
fmt.Println(msg)
|
|
t := time.Now()
|
|
// 确保日志目录存在
|
|
if err := os.MkdirAll(logDir, 0755); err != nil {
|
|
fmt.Println("create log dir failed:", err)
|
|
return
|
|
}
|
|
|
|
filename := filepath.Join(logDir, t.Format("2006-01-02")+".log")
|
|
f, err := os.OpenFile(filename, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0644)
|
|
if err != nil {
|
|
fmt.Println("open log file failed:", err)
|
|
return
|
|
}
|
|
defer f.Close()
|
|
|
|
line := fmt.Sprintf("%s [%s] %s\n", t.Format("2006-01-02 15:04:05"), level, msg)
|
|
_, _ = f.WriteString(line)
|
|
}
|
|
|
|
// 对外接口
|
|
func Info(msg string) {
|
|
logToFile("INFO", msg)
|
|
}
|
|
|
|
// Println 支持多个参数拼接，写 INFO 日志
|
|
func Println(v ...interface{}) {
|
|
msg := fmt.Sprint(v...)
|
|
logToFile("INFO", msg)
|
|
}
|
|
func Warn(msg string) {
|
|
logToFile("WARN", msg)
|
|
}
|
|
|
|
func Error(msg string) {
|
|
logToFile("ERROR", msg)
|
|
}
|
|
func Fatal(err error) {
|
|
if err == nil {
|
|
return
|
|
}
|
|
logToFile("ERROR", err.Error())
|
|
}
|
|
|
|
// Fatal 打印错误日志并退出程序
|
|
func Fatalf(msg string, args ...interface{}) {
|
|
if len(args) > 0 {
|
|
msg = fmt.Sprintf(msg, args...)
|
|
}
|
|
logToFile("FATAL", msg)
|
|
os.Exit(1)
|
|
}
|