mirror of
https://github.com/psviderski/uncloud.git
synced 2026-08-26 19:13:34 +00:00
74 lines
2.2 KiB
Go
74 lines
2.2 KiB
Go
//nolint:dupl
|
|
package service
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/docker/compose/v2/pkg/progress"
|
|
"github.com/docker/docker/api/types/container"
|
|
"github.com/psviderski/uncloud/internal/cli"
|
|
"github.com/psviderski/uncloud/internal/cli/completion"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type stopOptions struct {
|
|
services []string
|
|
signal string
|
|
timeout int
|
|
}
|
|
|
|
func NewStopCommand(groupID string) *cobra.Command {
|
|
opts := stopOptions{}
|
|
cmd := &cobra.Command{
|
|
Use: "stop SERVICE [SERVICE...]",
|
|
Short: "Stop one or more services.",
|
|
Long: `Stop one or more running services.
|
|
|
|
Gracefully stops all containers of the specified service(s) across all machines in the cluster.
|
|
Services can be specified by name or ID. Stopped services can be restarted with 'uc start'.`,
|
|
Args: cobra.MinimumNArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
uncli := cmd.Context().Value("cli").(*cli.CLI)
|
|
opts.services = args
|
|
return stop(cmd.Context(), uncli, opts)
|
|
},
|
|
GroupID: groupID,
|
|
ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]cobra.Completion, cobra.ShellCompDirective) {
|
|
uncli := cmd.Context().Value("cli").(*cli.CLI)
|
|
return completion.Services(cmd.Context(), uncli, args, toComplete)
|
|
},
|
|
}
|
|
cmd.Flags().StringVarP(&opts.signal, "signal", "s", "",
|
|
"Signal to send to each container's main process.\n"+
|
|
"Can be a signal name (SIGTERM, SIGINT, SIGHUP, etc.) or a number. (default SIGTERM)")
|
|
cmd.Flags().IntVarP(&opts.timeout, "timeout", "t", 10,
|
|
"Seconds to wait for each container to stop gracefully before forcibly killing it with SIGKILL.\n"+
|
|
"Use -1 to wait indefinitely.")
|
|
return cmd
|
|
}
|
|
|
|
func stop(ctx context.Context, uncli *cli.CLI, opts stopOptions) error {
|
|
client, err := uncli.ConnectCluster(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("connect to cluster: %w", err)
|
|
}
|
|
defer client.Close()
|
|
|
|
stopOpts := container.StopOptions{
|
|
Signal: opts.signal,
|
|
Timeout: &opts.timeout,
|
|
}
|
|
|
|
for _, s := range opts.services {
|
|
err = progress.RunWithTitle(ctx, func(ctx context.Context) error {
|
|
if err = client.StopService(ctx, s, stopOpts); err != nil {
|
|
return fmt.Errorf("stop service '%s': %w", s, err)
|
|
}
|
|
return nil
|
|
}, uncli.ProgressOut(), "Stopping service "+s)
|
|
}
|
|
|
|
return err
|
|
}
|