github
/
alist
mirror of https://github.com/Xhofe/alist
1
0
Fork 0
Code Issues Releases Wiki Activity

chore: change task.ID to uint64

refactor/fs
Noah Hsu 2022-06-17 21:52:31 +08:00
parent d73a9e4734
commit 68ca2abd0c
2 changed files with 19 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
pkg/task/manager.go
View File

@ -6,21 +6,21 @@ import (
"github.com/alist-org/alist/v3/pkg/generic_sync" "github.com/alist-org/alist/v3/pkg/generic_sync"
) )
func NewTaskManager() *Manager {
return &Manager{
tasks: generic_sync.MapOf[int64, *Task]{},
curID: 0,
}
}
type Manager struct { type Manager struct {
curID int64 curID uint64
tasks generic_sync.MapOf[int64, *Task] tasks generic_sync.MapOf[uint64, *Task]
} }
func (tm *Manager) AddTask(task *Task) { func (tm *Manager) Add(name string, f Func) uint64 {
task := newTask(name, f)
tm.addTask(task)
go task.Run()
return task.ID
}
func (tm *Manager) addTask(task *Task) {
task.ID = tm.curID task.ID = tm.curID
atomic.AddInt64(&tm.curID, 1) atomic.AddUint64(&tm.curID, 1)
tm.tasks.Store(task.ID, task) tm.tasks.Store(task.ID, task)
} }
@ -28,11 +28,11 @@ func (tm *Manager) GetAll() []*Task {
return tm.tasks.Values() return tm.tasks.Values()
} }
func (tm *Manager) Get(id int64) (*Task, bool) { func (tm *Manager) Get(id uint64) (*Task, bool) {
return tm.tasks.Load(id) return tm.tasks.Load(id)
} }
func (tm *Manager) Remove(id int64) { func (tm *Manager) Remove(id uint64) {
tm.tasks.Delete(id) tm.tasks.Delete(id)
} }
@ -54,8 +54,9 @@ func (tm *Manager) RemoveError() {
} }
} }
func (tm *Manager) Add(name string, f Func) { func NewTaskManager() *Manager {
task := newTask(name, f) return &Manager{
tm.AddTask(task) tasks: generic_sync.MapOf[uint64, *Task]{},
go task.Run() curID: 0,
}
} }

2
pkg/task/task.go
View File

@ -17,7 +17,7 @@ var (
type Func func(task *Task) error type Func func(task *Task) error
type Task struct { type Task struct {
ID int64 ID uint64
Name string Name string
Status string Status string
Error error Error error