feat: task manager

internal/task/manager.go

@@ -0,0 +1,61 @@
+package task
+
+import (
+	"sync/atomic"
+
+	"github.com/alist-org/alist/v3/pkg/generic_sync"
+)
+
+func NewTaskManager() *TaskManager {
+	return &TaskManager{
+		tasks: generic_sync.MapOf[int64, *Task]{},
+		curID: 0,
+	}
+}
+
+type TaskManager struct {
+	curID int64
+	tasks generic_sync.MapOf[int64, *Task]
+}
+
+func (tm *TaskManager) AddTask(task *Task) {
+	task.ID = tm.curID
+	atomic.AddInt64(&tm.curID, 1)
+	tm.tasks.Store(task.ID, task)
+}
+
+func (tm *TaskManager) GetAll() []*Task {
+	return tm.tasks.Values()
+}
+
+func (tm *TaskManager) Get(id int64) (*Task, bool) {
+	return tm.tasks.Load(id)
+}
+
+func (tm *TaskManager) Remove(id int64) {
+	tm.tasks.Delete(id)
+}
+
+func (tm *TaskManager) RemoveFinished() {
+	tasks := tm.GetAll()
+	for _, task := range tasks {
+		if task.Status == FINISHED {
+			tm.Remove(task.ID)
+		}
+	}
+}
+
+func (tm *TaskManager) RemoveError() {
+	tasks := tm.GetAll()
+	for _, task := range tasks {
+		if task.Error != nil {
+			tm.Remove(task.ID)
+		}
+	}
+}
+
+func (tm *TaskManager) Add(name string, f Func) {
+	task := NewTask(name, f)
+	tm.AddTask(task)
+	go task.Run()
+}

internal/task/task.go

@@ -1,10 +1,36 @@
 // manage task, such as file upload, file copy between accounts, offline download, etc.
 package task
 
+type Func func(task *Task) error
+
+var (
+	PENDING  = "pending"
+	RUNNING  = "running"
+	FINISHED = "finished"
+)
+
 type Task struct {
-	Name     string
-	Status   string
-	Error    error
-	Finish   bool
-	Children []*Task
+	ID     int64
+	Name   string
+	Status string
+	Error  error
+	Func   Func
+}
+
+func NewTask(name string, func_ Func) *Task {
+	return &Task{
+		Name:   name,
+		Status: PENDING,
+		Func:   func_,
+	}
+}
+
+func (t *Task) SetStatus(status string) {
+	t.Status = status
+}
+
+func (t *Task) Run() {
+	t.Status = RUNNING
+	t.Error = t.Func(t)
+	t.Status = FINISHED
 }