aiforge/services/cloudbrain/clear.go

package cloudbrain

import (
	"io/ioutil"
	"os"
	"sort"
	"time"

	"code.gitea.io/gitea/models"
	"code.gitea.io/gitea/modules/log"
	"code.gitea.io/gitea/modules/setting"
	"code.gitea.io/gitea/modules/storage"
)

func ClearCloudbrainResultSpace() {

	tasks, err := models.GetCloudBrainOneStoppedJobDaysAgo(setting.ClearStrategy.ResultSaveDays, setting.ClearStrategy.ResultBatchSize)

	if err != nil {
		log.Warn("Failed to get cloudbrain, clear result failed.", err)
		return
	}
	var ids []int64
	for _, task := range tasks {
		err := DeleteCloudbrainOneJobStorage(task.JobName)
		if err == nil {
			ids = append(ids, task.ID)
		}
	}

	err = models.UpdateCloudBrainRecordsCleared(ids)
	if err != nil {
		log.Warn("Failed to set cloudbrain cleared status", err)
	}
	if len(tasks) < setting.ClearStrategy.ResultBatchSize { //如果云脑表处理完了，通过遍历minio对象处理历史垃圾数据，如果存在
		files, err := ioutil.ReadDir(setting.JobPath)
		processCount:=0
		if err != nil {
			log.Warn("Can not browser local job path.")
		} else {
			SortModTimeAscend(files)
			for _, file := range files {
				//清理n天前的历史垃圾数据，清理job目录
				if file.ModTime().Before(time.Now().AddDate(0, 0, -setting.ClearStrategy.TrashSaveDays)) {
					os.RemoveAll(setting.JobPath + file.Name())
					processCount++
				}
				if processCount==setting.ClearStrategy.ResultBatchSize{
					break
				}
			}

		}

		minioFiles, err := storage.GetAllObjectByBucketAndPrefixMinio(setting.Attachment.Minio.Bucket, setting.CBCodePathPrefix)
		processCount=0
		if err != nil {
			log.Warn("Can not browser minio job path.")
		} else {
			SortModTimeAscendForMinio(minioFiles)
			for _, file := range minioFiles {

				timeConvert, err := time.Parse("2006-01-02 15:04:05", file.ModTime)
				if err == nil && timeConvert.Before(time.Now().AddDate(0, 0, -setting.ClearStrategy.TrashSaveDays)) {
					dirPath := setting.CBCodePathPrefix + file.FileName + "/"
					log.Info(dirPath)
					storage.Attachments.DeleteDir(dirPath)
					processCount++
					if processCount==setting.ClearStrategy.ResultBatchSize{
						break
					}
				}
			}

		}

	}

}

func SortModTimeAscend(files []os.FileInfo) {
	sort.Slice(files, func(i, j int) bool {
		return files[i].ModTime().Before(files[j].ModTime())
	})
}
func SortModTimeAscendForMinio(files []storage.FileInfo) {
	sort.Slice(files, func(i, j int) bool {
		timeI, _ := time.Parse("2006-01-02 15:04:05", files[i].ModTime)
		timeJ, _ := time.Parse("2006-01-02 15:04:05", files[i].ModTime)
		return timeI.Before(timeJ)
	})
}

func DeleteCloudbrainOneJobStorage(jobName string) error {
	//delete local
	localJobPath := setting.JobPath + jobName
	err := os.RemoveAll(localJobPath)
	if err != nil {
		log.Error("RemoveAll(%s) failed:%v", localJobPath, err)
	}

	dirPath := setting.CBCodePathPrefix + jobName + "/"
	err1 := storage.Attachments.DeleteDir(dirPath)

	if err1 != nil {
		log.Error("DeleteDir(%s) failed:%v", localJobPath, err)
	}
	if err == nil {
		err = err1
	}

	return err
}