mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
112 lines
2.7 KiB
Go
112 lines
2.7 KiB
Go
package stack
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/swarm"
|
|
"github.com/docker/docker/cli"
|
|
"github.com/docker/docker/cli/command"
|
|
"github.com/spf13/cobra"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
type removeOptions struct {
|
|
namespace string
|
|
}
|
|
|
|
func newRemoveCommand(dockerCli *command.DockerCli) *cobra.Command {
|
|
var opts removeOptions
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "rm STACK",
|
|
Aliases: []string{"remove", "down"},
|
|
Short: "Remove the stack",
|
|
Args: cli.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
opts.namespace = args[0]
|
|
return runRemove(dockerCli, opts)
|
|
},
|
|
}
|
|
return cmd
|
|
}
|
|
|
|
func runRemove(dockerCli *command.DockerCli, opts removeOptions) error {
|
|
namespace := opts.namespace
|
|
client := dockerCli.Client()
|
|
ctx := context.Background()
|
|
|
|
services, err := getServices(ctx, client, namespace)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
networks, err := getStackNetworks(ctx, client, namespace)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
secrets, err := getStackSecrets(ctx, client, namespace)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(services)+len(networks)+len(secrets) == 0 {
|
|
fmt.Fprintf(dockerCli.Out(), "Nothing found in stack: %s\n", namespace)
|
|
return nil
|
|
}
|
|
|
|
hasError := removeServices(ctx, dockerCli, services)
|
|
hasError = removeSecrets(ctx, dockerCli, secrets) || hasError
|
|
hasError = removeNetworks(ctx, dockerCli, networks) || hasError
|
|
|
|
if hasError {
|
|
return fmt.Errorf("Failed to remove some resources")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func removeServices(
|
|
ctx context.Context,
|
|
dockerCli *command.DockerCli,
|
|
services []swarm.Service,
|
|
) bool {
|
|
var err error
|
|
for _, service := range services {
|
|
fmt.Fprintf(dockerCli.Err(), "Removing service %s\n", service.Spec.Name)
|
|
if err = dockerCli.Client().ServiceRemove(ctx, service.ID); err != nil {
|
|
fmt.Fprintf(dockerCli.Err(), "Failed to remove service %s: %s", service.ID, err)
|
|
}
|
|
}
|
|
return err != nil
|
|
}
|
|
|
|
func removeNetworks(
|
|
ctx context.Context,
|
|
dockerCli *command.DockerCli,
|
|
networks []types.NetworkResource,
|
|
) bool {
|
|
var err error
|
|
for _, network := range networks {
|
|
fmt.Fprintf(dockerCli.Err(), "Removing network %s\n", network.Name)
|
|
if err = dockerCli.Client().NetworkRemove(ctx, network.ID); err != nil {
|
|
fmt.Fprintf(dockerCli.Err(), "Failed to remove network %s: %s", network.ID, err)
|
|
}
|
|
}
|
|
return err != nil
|
|
}
|
|
|
|
func removeSecrets(
|
|
ctx context.Context,
|
|
dockerCli *command.DockerCli,
|
|
secrets []swarm.Secret,
|
|
) bool {
|
|
var err error
|
|
for _, secret := range secrets {
|
|
fmt.Fprintf(dockerCli.Err(), "Removing secret %s\n", secret.Spec.Name)
|
|
if err = dockerCli.Client().SecretRemove(ctx, secret.ID); err != nil {
|
|
fmt.Fprintf(dockerCli.Err(), "Failed to remove secret %s: %s", secret.ID, err)
|
|
}
|
|
}
|
|
return err != nil
|
|
}
|