feat: update

Signed-off-by: Gaius <gaius.qi@gmail.com>

Author: gaius-qi

manager/job/preheat.go

@@ -124,8 +124,12 @@ func (p *preheat) CreatePreheat(ctx context.Context, schedulers []models.Schedul
 			return nil, err
 		}
 	case PreheatFileType:
-		// File preheat supports to preheat multiple files and single file.
-		if json.URL == "" && len(json.URLs) == 0 {
+		urls := json.URLs
+		if json.URL != "" {
+			urls = append(urls, json.URL)
+		}
+
+		if len(urls) == 0 {
 			return nil, errors.New("invalid params: url or urls is required")
 		}
 
@@ -135,7 +139,7 @@ func (p *preheat) CreatePreheat(ctx context.Context, schedulers []models.Schedul
 		}
 
 		files = append(files, internaljob.PreheatRequest{
-			URLs:                append(json.URLs, json.URL),
+			URLs:                urls,
 			PieceLength:         json.PieceLength,
 			Tag:                 json.Tag,
 			Application:         json.Application,

scheduler/job/job.go

@@ -296,7 +296,7 @@ func (j *job) preheatV2SingleSeedPeer(ctx context.Context, req *internaljob.Preh
 	eg, _ := errgroup.WithContext(ctx)
 	eg.SetLimit(int(req.ConcurrentCount))
 
-	var preheatResp *internaljob.PreheatResponse
+	preheatResp := &internaljob.PreheatResponse{}
 	for _, url := range req.URLs {
 		eg.Go(func() error {
 			resp, err := j.preheatV2SingleSeedPeerByURL(ctx, url, req, log)

test/e2e/manager/task.go

@@ -57,7 +57,7 @@ var _ = Describe("GetTask and DeleteTask with Manager", func() {
 				Type: internaljob.PreheatJob,
 				Args: types.PreheatArgs{
 					Type: "file",
-					URL:  testFile.GetDownloadURL(),
+					URLs: []string{testFile.GetDownloadURL()},
 				},
 				SchedulerClusterIDs: []uint{1},
 			})
@@ -163,7 +163,7 @@ var _ = Describe("GetTask and DeleteTask with Manager", func() {
 				Type: internaljob.PreheatJob,
 				Args: types.PreheatArgs{
 					Type: "file",
-					URL:  testFile.GetDownloadURL(),
+					URLs: []string{testFile.GetDownloadURL()},
 				},
 				SchedulerClusterIDs: []uint{1},
 			})