mirror of
https://github.com/woodchen-ink/webp_server_go.git
synced 2025-07-18 05:32:02 +08:00
126 lines
3.0 KiB
Go
126 lines
3.0 KiB
Go
package schedule
|
||
|
||
import (
|
||
"os"
|
||
"path/filepath"
|
||
"time"
|
||
"webp_server_go/config"
|
||
|
||
log "github.com/sirupsen/logrus"
|
||
)
|
||
|
||
func getDirSize(path string) (int64, error) {
|
||
// Check if path is a directory and exists
|
||
if _, err := os.Stat(path); os.IsNotExist(err) {
|
||
return 0, nil
|
||
}
|
||
var size int64
|
||
err := filepath.Walk(path, func(path string, info os.FileInfo, err error) error {
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if !info.IsDir() {
|
||
size += info.Size()
|
||
}
|
||
return nil
|
||
})
|
||
return size, err
|
||
}
|
||
|
||
// Delete the oldest file in the given path
|
||
func clearDirForOldestFiles(path string) error {
|
||
oldestFile := ""
|
||
oldestModTime := time.Now()
|
||
|
||
err := filepath.Walk(path, func(path string, info os.FileInfo, err error) error {
|
||
if err != nil {
|
||
log.Errorf("Error accessing path %s: %s\n", path, err.Error())
|
||
return nil
|
||
}
|
||
|
||
if !info.IsDir() && info.ModTime().Before(oldestModTime) {
|
||
oldestFile = path
|
||
oldestModTime = info.ModTime()
|
||
}
|
||
return nil
|
||
})
|
||
|
||
if err != nil {
|
||
log.Errorf("遍历目录时出错: %s\n", err.Error())
|
||
return err
|
||
}
|
||
|
||
if oldestFile != "" {
|
||
err := os.Remove(oldestFile)
|
||
if err != nil {
|
||
log.Errorf("删除文件时出错 %s: %s\n", oldestFile, err.Error())
|
||
return err
|
||
}
|
||
log.Infof("删除了最旧的文件: %s\n", oldestFile)
|
||
} else {
|
||
log.Infoln("目录中没有找到文件.")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// Clear cache, size is in bytes that needs to be cleared out
|
||
// Will delete oldest files first, then second oldest, etc.
|
||
// Until all files size are less than maxCacheSizeBytes
|
||
func clearCacheFiles(path string, maxCacheSizeBytes int64) error {
|
||
dirSize, err := getDirSize(path)
|
||
if err != nil {
|
||
log.Errorf("获取目录大小时出错: %s\n", err.Error())
|
||
return err
|
||
}
|
||
|
||
for dirSize > maxCacheSizeBytes {
|
||
err := clearDirForOldestFiles(path)
|
||
if err != nil {
|
||
log.Errorf("清除目录时出错: %s\n", err.Error())
|
||
return err
|
||
}
|
||
dirSize, err = getDirSize(path)
|
||
if err != nil {
|
||
log.Errorf("获取目录大小时出错: %s\n", err.Error())
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func CleanCache() {
|
||
log.Info("MaxCacheSize不为0，启动缓存清理服务...")
|
||
ticker := time.NewTicker(1 * time.Minute)
|
||
defer ticker.Stop()
|
||
|
||
for range ticker.C {
|
||
// MB to bytes
|
||
maxCacheSizeBytes := int64(config.Config.MaxCacheSize) * 1024 * 1024
|
||
|
||
if err := clearCacheFiles(config.Config.RemoteRawPath, maxCacheSizeBytes); err != nil {
|
||
log.Warnf("清除远程原始缓存失败: %v", err)
|
||
}
|
||
|
||
if err := clearCacheFiles(config.Config.ExhaustPath, maxCacheSizeBytes); err != nil && err != os.ErrNotExist {
|
||
log.Warnf("清除优化图像缓存失败: %v", err)
|
||
}
|
||
|
||
if err := clearCacheFiles(config.Config.MetadataPath, maxCacheSizeBytes); err != nil && err != os.ErrNotExist {
|
||
log.Warnf("清除元数据缓存失败: %v", err)
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
var cleanupDelay = 5 * time.Minute
|
||
|
||
func ScheduleCleanup(filePath string) {
|
||
time.AfterFunc(cleanupDelay, func() {
|
||
if err := os.Remove(filePath); err != nil {
|
||
log.Warnf("清理原始文件失败: %s, 错误: %v", filePath, err)
|
||
} else {
|
||
log.Infof("成功清理原始文件: %s", filePath)
|
||
}
|
||
})
|
||
}
|