Add parallels flag

daaru00 · Mar 15, 2021 · 744d637 · 744d637
1 parent 9191955
commit 744d637
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 6 deletions.
diff --git a/cmd/deploy/deploy.go b/cmd/deploy/deploy.go
@@ -55,6 +55,11 @@ func NewCommand(globalFlags []cli.Flag) *cli.Command {
 				Aliases: []string{"a"},
 				Usage:   "Select all documents",
 			},
+			&cli.IntFlag{
+				Name:  "parallels",
+				Usage: "Set max deploy executed in parallel",
+				Value: 5,
+			},
 		}...),
 		Action:    Action,
 		ArgsUsage: "[path...]",
@@ -90,12 +95,12 @@ func Action(c *cli.Context) error {
 
 	// Setup max parallels
 	allDocuments := *documents
-	parallel := 10
+	parallels := c.Int("parallels")
 
 	// Start parallel deploy
-	for start := 0; start < len(allDocuments); start += parallel {
+	for start := 0; start < len(allDocuments); start += parallels {
 		// Update chunk start and end
-		end := start + parallel
+		end := start + parallels
 		if end > len(allDocuments) {
 			end = len(allDocuments)
 		}

diff --git a/cmd/remove/remove.go b/cmd/remove/remove.go
@@ -49,6 +49,11 @@ func NewCommand(globalFlags []cli.Flag) *cli.Command {
 				Aliases: []string{"a"},
 				Usage:   "Select all documents",
 			},
+			&cli.IntFlag{
+				Name:  "parallels",
+				Usage: "Set max remove executed in parallel",
+				Value: 5,
+			},
 		}...),
 		Action:    Action,
 		ArgsUsage: "[path...]",
@@ -92,12 +97,12 @@ func Action(c *cli.Context) error {
 
 	// Setup max parallels
 	allDocuments := *documents
-	parallel := 10
+	parallels := c.Int("parallels")
 
 	// Start parallel deploy
-	for start := 0; start < len(allDocuments); start += parallel {
+	for start := 0; start < len(allDocuments); start += parallels {
 		// Update chunk start and end
-		end := start + parallel
+		end := start + parallels
 		if end > len(allDocuments) {
 			end = len(allDocuments)
 		}