74 lines
1.5 KiB
Go
74 lines
1.5 KiB
Go
package fileservice
|
||
|
||
import (
|
||
"fmt"
|
||
"go-admin/app/admin/models"
|
||
"os"
|
||
"path/filepath"
|
||
"strconv"
|
||
"time"
|
||
|
||
"github.com/go-admin-team/go-admin-core/sdk/config"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
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 := 7
|
||
var sysConfig models.SysConfig
|
||
|
||
orm.Model(&sysConfig).Where("config_key = ?", "log_expirate_date").First(&sysConfig)
|
||
|
||
if sysConfig.ConfigValue != "" {
|
||
day, _ := strconv.Atoi(sysConfig.ConfigValue)
|
||
|
||
if day > 0 {
|
||
expirateDay = day
|
||
}
|
||
}
|
||
|
||
// 遍历指定文件夹中的所有文件
|
||
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)
|
||
}
|
||
}
|