add support for up --scale

Signed-off-by: Nicolas De Loof <[email protected]>

Author: ndeloof

cli/cmd/compose/up.go

@@ -22,6 +22,8 @@ import (
 	"os"
 	"os/signal"
 	"path/filepath"
+	"strconv"
+	"strings"
 	"syscall"
 
 	"github.com/docker/compose-cli/api/client"
@@ -54,6 +56,7 @@ type upOptions struct {
 	noStart       bool
 	cascadeStop   bool
 	exitCodeFrom  string
+	scale         []string
 }
 
 func (o upOptions) recreateStrategy() string {
@@ -98,6 +101,7 @@ func upCommand(p *projectOptions, contextType string) *cobra.Command {
 	flags.BoolVarP(&opts.Detach, "detach", "d", false, "Detached mode: Run containers in the background")
 	flags.BoolVar(&opts.Build, "build", false, "Build images before starting containers.")
 	flags.BoolVar(&opts.removeOrphans, "remove-orphans", false, "Remove containers for services not defined in the Compose file.")
+	flags.StringArrayVar(&opts.scale, "scale", []string{}, "Scale SERVICE to NUM instances. Overrides the `scale` setting in the Compose file if present.")
 
 	switch contextType {
 	case store.AciContextType:
@@ -119,6 +123,11 @@ func runUp(ctx context.Context, opts upOptions, services []string) error {
 		return err
 	}
 
+	err = applyScale(opts.scale, project)
+	if err != nil {
+		return err
+	}
+
 	_, err = progress.Run(ctx, func(ctx context.Context) (string, error) {
 		return "", c.ComposeService().Up(ctx, project, compose.UpOptions{
 			Detach: opts.Detach,
@@ -133,6 +142,11 @@ func runCreateStart(ctx context.Context, opts upOptions, services []string) erro
 		return err
 	}
 
+	err = applyScale(opts.scale, project)
+	if err != nil {
+		return err
+	}
+
 	if opts.exitCodeFrom != "" {
 		_, err := project.GetService(opts.exitCodeFrom)
 		if err != nil {
@@ -201,6 +215,38 @@ func runCreateStart(ctx context.Context, opts upOptions, services []string) erro
 	return err
 }
 
+func applyScale(opts []string, project *types.Project) error {
+SCALE:
+	for _, scale := range opts {
+		split := strings.Split(scale, "=")
+		if len(split) != 2 {
+			return fmt.Errorf("invalid --scale option %q. Should be SERVICE=NUM", scale)
+		}
+		name := split[0]
+		replicas, err := strconv.Atoi(split[1])
+		if err != nil {
+			return err
+		}
+		for i, s := range project.Services {
+			if s.Name == name {
+				service, err := project.GetService(name)
+				if err != nil {
+					return err
+				}
+				if service.Deploy == nil {
+					service.Deploy = &types.DeployConfig{}
+				}
+				count := uint64(replicas)
+				service.Deploy.Replicas = &count
+				project.Services[i] = service
+				continue SCALE
+			}
+		}
+		return fmt.Errorf("unknown service %q", name)
+	}
+	return nil
+}
+
 func setup(ctx context.Context, opts composeOptions, services []string) (*client.Client, *types.Project, error) {
 	c, err := client.NewWithDefaultLocalBackend(ctx)
 	if err != nil {