Merge pull request #1415 from ulyssessouza/add-restart

Add restart command

cli/cmd/compose/compose.go

@@ -127,6 +127,7 @@ func Command(contextType string) *cobra.Command {
 		upCommand(&opts, contextType),
 		downCommand(&opts, contextType),
 		startCommand(&opts),
+		restartCommand(&opts),
 		stopCommand(&opts),
 		psCommand(&opts),
 		listCommand(contextType),

cli/cmd/compose/restart.go

@@ -0,0 +1,70 @@
+/*
+   Copyright 2020 Docker Compose CLI authors
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+*/
+
+package compose
+
+import (
+	"context"
+	"time"
+
+	"github.com/spf13/cobra"
+
+	"github.com/docker/compose-cli/api/client"
+	"github.com/docker/compose-cli/api/compose"
+	"github.com/docker/compose-cli/api/progress"
+)
+
+type restartOptions struct {
+	*projectOptions
+	timeout int
+}
+
+func restartCommand(p *projectOptions) *cobra.Command {
+	opts := restartOptions{
+		projectOptions: p,
+	}
+	restartCmd := &cobra.Command{
+		Use:   "restart",
+		Short: "Restart containers",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return runRestart(cmd.Context(), opts, args)
+		},
+	}
+	flags := restartCmd.Flags()
+	flags.IntVarP(&opts.timeout, "timeout", "t", 10, "Specify a shutdown timeout in seconds")
+
+	return restartCmd
+}
+
+func runRestart(ctx context.Context, opts restartOptions, services []string) error {
+	c, err := client.New(ctx)
+	if err != nil {
+		return err
+	}
+
+	project, err := opts.toProject(services)
+	if err != nil {
+		return err
+	}
+
+	timeout := time.Duration(opts.timeout) * time.Second
+	_, err = progress.Run(ctx, func(ctx context.Context) (string, error) {
+		return "", c.ComposeService().Restart(ctx, project, compose.RestartOptions{
+			Timeout: &timeout,
+		})
+	})
+	return err
+}