Feat(remotearia2): 转存

bootstrap/init.go

@@ -31,6 +31,7 @@ func Init(path string) {
 	} else {
 		if conf.SlaveConfig.Aria2 {
 			model.Init()
+			task.Init()
 			aria2.Init(false)
 		}
 	}

pkg/aria2/aria2.go

@@ -152,7 +152,15 @@ func SlaveInit(isReload bool) {
 
 	Instance = client
 
-	//	monitor
+	if !isReload {
+		// 从数据库中读取未完成任务，创建监控
+		unfinished := model.GetDownloadsByStatus(Ready, Paused, Downloading)
+
+		for i := 0; i < len(unfinished); i++ {
+			// 创建任务监控
+			NewMonitor(&unfinished[i])
+		}
+	}
 }
 
 // MasterInit 主机初始化

pkg/task/tranfer.go

@@ -3,6 +3,7 @@ package task
 import (
 	"context"
 	"encoding/json"
+	"github.com/cloudreve/Cloudreve/v3/pkg/conf"
 	"os"
 	"path"
 	"path/filepath"
@@ -84,6 +85,13 @@ func (job *TransferTask) GetError() *JobError {
 // Do 开始执行任务
 func (job *TransferTask) Do() {
 	defer job.Recycle()
+	if model.IsTrueVal(model.GetSettingByName("aria2_remote_enabled")) && conf.SystemConfig.Mode != "slave" {
+		return
+	}
+
+	if conf.SlaveConfig.SlaveId == 0 || model.GetIntSetting("aria2_remote_id", 0) != int(conf.SlaveConfig.SlaveId) {
+		return
+	}
 
 	// 创建文件系统
 	fs, err := filesystem.NewFileSystem(job.User)