DockerCLI/cli/command/stack/kubernetes/ps.go

package kubernetes

import (
	"fmt"
	"sort"

	"github.com/docker/cli/cli/command"
	"github.com/docker/cli/cli/command/formatter"
	"github.com/docker/cli/cli/command/stack/options"
	"github.com/docker/cli/cli/command/task"
	"github.com/docker/docker/api/types/swarm"
	apiv1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/fields"
	corev1 "k8s.io/client-go/kubernetes/typed/core/v1"
	"k8s.io/kubernetes/pkg/api"
)

// RunPS is the kubernetes implementation of docker stack ps
func RunPS(dockerCli *KubeCli, options options.PS) error {
	namespace := options.Namespace

	// Initialize clients
	client, err := dockerCli.composeClient()
	if err != nil {
		return err
	}
	stacks, err := client.Stacks(false)
	if err != nil {
		return err
	}
	podsClient := client.Pods()

	// Fetch pods
	if _, err := stacks.Get(namespace); err != nil {
		return fmt.Errorf("nothing found in stack: %s", namespace)
	}

	pods, err := fetchPods(namespace, podsClient)
	if err != nil {
		return err
	}

	if len(pods) == 0 {
		return fmt.Errorf("nothing found in stack: %s", namespace)
	}

	format := options.Format
	if len(format) == 0 {
		format = task.DefaultFormat(dockerCli.ConfigFile(), options.Quiet)
	}
	nodeResolver := makeNodeResolver(options.NoResolve, client.Nodes())

	tasks := make([]swarm.Task, len(pods))
	for i, pod := range pods {
		tasks[i] = podToTask(pod)
	}
	return print(dockerCli, namespace, tasks, pods, nodeResolver, !options.NoTrunc, options.Quiet, format)
}

type idResolver func(name string) (string, error)

func print(dockerCli command.Cli, namespace string, tasks []swarm.Task, pods []apiv1.Pod, nodeResolver idResolver, trunc, quiet bool, format string) error {
	sort.Stable(tasksBySlot(tasks))

	names := map[string]string{}
	nodes := map[string]string{}

	tasksCtx := formatter.Context{
		Output: dockerCli.Out(),
		Format: formatter.NewTaskFormat(format, quiet),
		Trunc:  trunc,
	}

	for i, task := range tasks {
		nodeValue, err := nodeResolver(pods[i].Spec.NodeName)
		if err != nil {
			return err
		}

		names[task.ID] = fmt.Sprintf("%s_%s", namespace, pods[i].Name)
		nodes[task.ID] = nodeValue
	}

	return formatter.TaskWrite(tasksCtx, tasks, names, nodes)
}

func makeNodeResolver(noResolve bool, nodes corev1.NodeInterface) func(string) (string, error) {
	// Here we have a name and we need to resolve its identifier. To mimic swarm behavior
	// we need to resolve the id when noresolve is set, otherwise we return the name.
	if noResolve {
		return func(name string) (string, error) {
			n, err := nodes.List(metav1.ListOptions{
				FieldSelector: fields.OneTermEqualSelector(api.ObjectNameField, name).String(),
			})
			if err != nil {
				return "", err
			}
			if len(n.Items) != 1 {
				return "", fmt.Errorf("could not find node '%s'", name)
			}
			return string(n.Items[0].UID), nil
		}
	}
	return func(name string) (string, error) { return name, nil }
}