loggerx/remove.go

package loggerx

import (
	"fmt"
	"os"
	"path/filepath"
	"time"
)

// 监听dir文件夹的所有文件，循环查找是否有超过days天的文件，删除掉

func (l *Logger) delete() {
	tick := time.NewTicker(time.Hour)
	defer tick.Stop()

	for {
		select {
		case <-tick.C:
			// 后台清理不能把整个进程带崩
			func() {
				defer func() {
					if r := recover(); r != nil {
						fmt.Println("清理日志文件异常:", r)
					}
				}()
				if err := l.walkAndDel(); err != nil {
					fmt.Println(err)
				}
			}()
		case <-l.ctx.Done():
			return
		case <-l.done:
			return
		}
	}
}

func (l *Logger) walkAndDel() error {
	// 没设置删除天数（<=0 表示不删除）
	if l.option.days <= 0 {
		return nil
	}

	deadline := time.Now().AddDate(0, 0, -l.option.days)

	err := filepath.Walk(l.option.dir, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			// 单个文件出错不应该中断整个遍历
			fmt.Println(err)
			return nil
		}

		// 最后修改时间在保留期内，跳过
		if info.ModTime().After(deadline) {
			return nil
		}

		if info.IsDir() {
			if !isEmptyDir(path) {
				return nil
			}
		} else if filepath.Ext(path) != ".log" {
			return nil
		}

		// 删除文件 / 空目录
		if rerr := os.Remove(path); rerr != nil {
			// 正被写入的文件在 Windows 上删不掉，等下一轮再试，不要终止遍历
			fmt.Println("删除文件失败", path, rerr)
			return nil
		}
		fmt.Println("删除文件", path)
		return nil
	})

	return err
}

// 判断空文件夹
func isEmptyDir(path string) bool {
	files, err := os.ReadDir(path)
	if err != nil {
		fmt.Println(err)
		return false
	}
	return len(files) == 0 // 文件夹为空
}