feat: create aria2 task in batch

models/group.go

@@ -33,7 +33,6 @@ type GroupOption struct {
 	Aria2Options    map[string]interface{} `json:"aria2_options,omitempty"` // 离线下载用户组配置
 	SourceBatchSize int                    `json:"source_batch,omitempty"`
 	Aria2BatchSize  int                    `json:"aria2_batch,omitempty"`
-	Aria2TaskSizeLimit uint64                 `json:"aria2_size,omitempty"`
 }
 
 // GetGroupByID 用ID获取用户组

pkg/serializer/error.go

@@ -82,6 +82,8 @@ const (
 	CodeInvalidContentLength = 400013
 	// CodeBatchSourceSize 超出批量获取外链限制
 	CodeBatchSourceSize = 40014
+	// CodeBatchAria2Size 超出最大 Aria2 任务数量限制
+	CodeBatchAria2Size = 40012
 	// CodeDBError 数据库操作失败
 	CodeDBError = 50001
 	// CodeEncryptError 加密失败

routers/controllers/aria2.go

@@ -11,7 +11,7 @@ import (
 
 // AddAria2URL 添加离线下载URL
 func AddAria2URL(c *gin.Context) {
-	var addService aria2.AddURLService
+	var addService aria2.BatchAddURLService
 	if err := c.ShouldBindJSON(&addService); err == nil {
 		res := addService.Add(c, common.URLTask)
 		c.JSON(200, res)
@@ -52,7 +52,7 @@ func AddAria2Torrent(c *gin.Context) {
 
 		if err := c.ShouldBindJSON(&addService); err == nil {
 			addService.URL = res.Data.(string)
-			res := addService.Add(c, common.URLTask)
+			res := addService.Add(c, nil, common.URLTask)
 			c.JSON(200, res)
 		} else {
 			c.JSON(200, ErrorResponse(err))

service/aria2/add.go

@@ -13,6 +13,51 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
+// AddURLService 添加URL离线下载服务
+type BatchAddURLService struct {
+	URLs []string `json:"url" binding:"required"`
+	Dst  string   `json:"dst" binding:"required,min=1"`
+}
+
+// Add 主机批量创建新的链接离线下载任务
+func (service *BatchAddURLService) Add(c *gin.Context, taskType int) serializer.Response {
+	// 创建文件系统
+	fs, err := filesystem.NewFileSystemFromContext(c)
+	if err != nil {
+		return serializer.Err(serializer.CodePolicyNotAllowed, err.Error(), err)
+	}
+	defer fs.Recycle()
+
+	// 检查用户组权限
+	if !fs.User.Group.OptionsSerialized.Aria2 {
+		return serializer.Err(serializer.CodeGroupNotAllowed, "当前用户组无法进行此操作", nil)
+	}
+
+	// 存放目录是否存在
+	if exist, _ := fs.IsPathExist(service.Dst); !exist {
+		return serializer.Err(serializer.CodeNotFound, "存放路径不存在", nil)
+	}
+
+	// 检查批量任务数量
+	limit := fs.User.Group.OptionsSerialized.Aria2BatchSize
+	if limit > 0 && len(service.URLs) > limit {
+		return serializer.Err(serializer.CodeBatchAria2Size, "Exceed aria2 batch size", nil)
+	}
+
+	res := make([]serializer.Response, 0, len(service.URLs))
+	for _, target := range service.URLs {
+		subService := &AddURLService{
+			URL: target,
+			Dst: service.Dst,
+		}
+
+		addRes := subService.Add(c, fs, taskType)
+		res = append(res, addRes)
+	}
+
+	return serializer.Response{Data: res}
+}
+
 // AddURLService 添加URL离线下载服务
 type AddURLService struct {
 	URL string `json:"url" binding:"required"`
@@ -20,9 +65,11 @@ type AddURLService struct {
 }
 
 // Add 主机创建新的链接离线下载任务
-func (service *AddURLService) Add(c *gin.Context, taskType int) serializer.Response {
+func (service *AddURLService) Add(c *gin.Context, fs *filesystem.FileSystem, taskType int) serializer.Response {
+	if fs == nil {
+		var err error
 		// 创建文件系统
-	fs, err := filesystem.NewFileSystemFromContext(c)
+		fs, err = filesystem.NewFileSystemFromContext(c)
 		if err != nil {
 			return serializer.Err(serializer.CodePolicyNotAllowed, err.Error(), err)
 		}
@@ -37,6 +84,13 @@ func (service *AddURLService) Add(c *gin.Context, taskType int) serializer.Respo
 		if exist, _ := fs.IsPathExist(service.Dst); !exist {
 			return serializer.Err(serializer.CodeNotFound, "存放路径不存在", nil)
 		}
+	}
+
+	downloads := model.GetDownloadsByStatusAndUser(0, fs.User.ID, common.Downloading, common.Paused, common.Ready)
+	limit := fs.User.Group.OptionsSerialized.Aria2BatchSize
+	if limit > 0 && len(downloads)+1 > limit {
+		return serializer.Err(serializer.CodeBatchAria2Size, "Exceed aria2 batch size", nil)
+	}
 
 	// 创建任务
 	task := &model.Download{