ChineseSubFinder/pkg/cache_center/task_queue_info.go

package cache_center

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

	"github.com/allanpk716/ChineseSubFinder/pkg"

	"github.com/allanpk716/ChineseSubFinder/pkg/cache_center/models"
)

func (c *CacheCenter) TaskQueueClear() error {

	// 没有必要删除 DB 中的数据，直接删除外部的缓存文件即可
	err := pkg.ClearFolder(c.taskQueueSaveRootPath)
	if err != nil {
		return err
	}
	return nil
}

func (c *CacheCenter) TaskQueueSave(taskPriority int, taskQueueBytes []byte) error {
	defer c.locker.Unlock()
	c.locker.Lock()

	var taskQueues []models.TaskQueueInfo
	c.db.Where("priority = ?", taskPriority).Find(&taskQueues)
	// 写入到本地存储
	saveFPath := filepath.Join(c.taskQueueSaveRootPath, fmt.Sprintf("%d", taskPriority)+".tq")
	err := pkg.WriteFile(saveFPath, taskQueueBytes)
	if err != nil {
		return err
	}
	relPath, err := filepath.Rel(c.taskQueueSaveRootPath, saveFPath)
	if err != nil {
		return err
	}

	if len(taskQueues) == 0 {
		// 不存在，需要新建
		nowTaskQueue := models.TaskQueueInfo{
			Priority: taskPriority,
			RelPath:  relPath,
		}
		c.db.Save(&nowTaskQueue)
	} else {
		// 存在，需要更新
		taskQueues[0].RelPath = relPath
		c.db.Save(&taskQueues[0])
	}

	return nil
}

func (c *CacheCenter) TaskQueueRead() (map[int][]byte, error) {
	defer c.locker.Unlock()
	c.locker.Lock()

	var taskQueues []models.TaskQueueInfo
	c.db.Find(&taskQueues)

	outTaskQueueBytes := make(map[int][]byte, 0)
	for _, taskQueue := range taskQueues {

		oneTaskQueueFPath := filepath.Join(c.taskQueueSaveRootPath, taskQueue.RelPath)
		if pkg.IsFile(oneTaskQueueFPath) == false {
			continue
		}
		bytes, err := os.ReadFile(oneTaskQueueFPath)
		if err != nil {
			return nil, err
		}

		outTaskQueueBytes[taskQueue.Priority] = bytes
	}

	return outTaskQueueBytes, nil
}