chore(aria2): export task manager

This commit is contained in:
Noah Hsu
2022-06-23 21:24:23 +08:00
parent 92983aa185
commit 40b7ecc845
5 changed files with 17 additions and 10 deletions

View File

@ -92,7 +92,7 @@ func (m *Monitor) Update() (bool, error) {
}
}
var transferTaskManager = task.NewTaskManager[uint64](3, func(k *uint64) {
var TransferTaskManager = task.NewTaskManager[uint64](3, func(k *uint64) {
atomic.AddUint64(k, 1)
})
@ -119,7 +119,7 @@ func (m *Monitor) Complete() error {
}
}()
for _, file := range files {
transferTaskManager.Submit(task.WithCancelCtx[uint64](&task.Task[uint64]{
TransferTaskManager.Submit(task.WithCancelCtx[uint64](&task.Task[uint64]{
Name: fmt.Sprintf("transfer %s to [%s](%s)", file.Path, account.GetAccount().VirtualPath, dstDirActualPath),
Func: func(tsk *task.Task[uint64]) error {
defer wg.Done()