82 lines
1.6 KiB
Go
82 lines
1.6 KiB
Go
|
package jobs
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"os"
|
|||
|
|
"path/filepath"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/go-admin-team/go-admin-core/logger"
|
|||
|
|
"github.com/go-admin-team/go-admin-core/sdk"
|
|||
|
|
"github.com/go-admin-team/go-admin-core/sdk/config"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type ClearLogJob struct {
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 删除过期日志
|
|||
|
|
func (t ClearLogJob) Exec(arg interface{}) error {
|
|||
|
|
str := time.Now().Format(timeFormat) + " [INFO] JobCore ClearLogJob exec success"
|
|||
|
|
defer logger.Info(str)
|
|||
|
|
var db *gorm.DB
|
|||
|
|
|
|||
|
|
for _, item := range sdk.Runtime.GetDb() {
|
|||
|
|
db = item
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ClearLogs(db)
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func ClearLogs(orm *gorm.DB) {
|
|||
|
|
dir := config.LoggerConfig.Path
|
|||
|
|
|
|||
|
|
if dir == "" {
|
|||
|
|
dir = "temp/logs"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查文件夹是否存在
|
|||
|
|
if _, err := os.Stat(dir); os.IsNotExist(err) {
|
|||
|
|
fmt.Printf("Directory %s does not exist, skipping cleanup.\n", dir)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取当前时间
|
|||
|
|
now := time.Now()
|
|||
|
|
expirateDay := 2
|
|||
|
|
|
|||
|
|
// 遍历指定文件夹中的所有文件
|
|||
|
|
err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 只处理普通文件
|
|||
|
|
if !info.IsDir() {
|
|||
|
|
// 获取文件的修改时间
|
|||
|
|
modTime := info.ModTime()
|
|||
|
|
|
|||
|
|
// 计算文件修改时间与当前时间的差值
|
|||
|
|
duration := now.Sub(modTime)
|
|||
|
|
|
|||
|
|
// 如果文件超过7天，则删除
|
|||
|
|
if duration > time.Duration(expirateDay)*24*time.Hour {
|
|||
|
|
fmt.Printf("Deleting file: %s (Last modified: %s)\n", path, modTime)
|
|||
|
|
err := os.Remove(path)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Printf("Error walking the path %v: %v\n", dir, err)
|
|||
|
|
}
|
|||
|
|
}
|