ProjectHentai/alist
6
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: set aria2 client and add url to aria2 api

Browse Source
This commit is contained in:
Noah Hsu
2022-06-29 17:31:37 +08:00
parent a09a1b814b
commit 8abee6504f
5 changed files with 102 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/aria2/aria2.go
View File

@ -2,6 +2,8 @@ package aria2
import (
"context"
"github.com/alist-org/alist/v3/internal/conf"
"github.com/alist-org/alist/v3/internal/setting"
"github.com/alist-org/alist/v3/pkg/aria2/rpc"
"github.com/alist-org/alist/v3/pkg/task"
"github.com/pkg/errors"
@ -13,6 +15,12 @@ var DownTaskManager = task.NewTaskManager[string](3)
var notify = NewNotify()
var client rpc.Client
func InitClient(timeout int) error {
uri := setting.GetByKey(conf.Aria2Uri)
secret := setting.GetByKey(conf.Aria2Secret)
return InitAria2Client(uri, secret, timeout)
}
func InitAria2Client(uri string, secret string, timeout int) error {
c, err := rpc.New(context.Background(), uri, secret, time.Duration(timeout)*time.Second, notify)
if err != nil {