github.com/kim0/docker@v0.6.2-0.20161130212042-4addda3f07e7/cli/command/service/list.go (about)

     1  package service
     2  
     3  import (
     4  	"fmt"
     5  	"io"
     6  	"strings"
     7  	"text/tabwriter"
     8  
     9  	"github.com/docker/docker/api/types"
    10  	"github.com/docker/docker/api/types/filters"
    11  	"github.com/docker/docker/api/types/swarm"
    12  	"github.com/docker/docker/cli"
    13  	"github.com/docker/docker/cli/command"
    14  	"github.com/docker/docker/opts"
    15  	"github.com/docker/docker/pkg/stringid"
    16  	"github.com/spf13/cobra"
    17  	"golang.org/x/net/context"
    18  )
    19  
    20  const (
    21  	listItemFmt = "%s\t%s\t%s\t%s\t%s\n"
    22  )
    23  
    24  type listOptions struct {
    25  	quiet  bool
    26  	filter opts.FilterOpt
    27  }
    28  
    29  func newListCommand(dockerCli *command.DockerCli) *cobra.Command {
    30  	opts := listOptions{filter: opts.NewFilterOpt()}
    31  
    32  	cmd := &cobra.Command{
    33  		Use:     "ls [OPTIONS]",
    34  		Aliases: []string{"list"},
    35  		Short:   "List services",
    36  		Args:    cli.NoArgs,
    37  		RunE: func(cmd *cobra.Command, args []string) error {
    38  			return runList(dockerCli, opts)
    39  		},
    40  	}
    41  
    42  	flags := cmd.Flags()
    43  	flags.BoolVarP(&opts.quiet, "quiet", "q", false, "Only display IDs")
    44  	flags.VarP(&opts.filter, "filter", "f", "Filter output based on conditions provided")
    45  
    46  	return cmd
    47  }
    48  
    49  func runList(dockerCli *command.DockerCli, opts listOptions) error {
    50  	ctx := context.Background()
    51  	client := dockerCli.Client()
    52  	out := dockerCli.Out()
    53  
    54  	services, err := client.ServiceList(ctx, types.ServiceListOptions{Filter: opts.filter.Value()})
    55  	if err != nil {
    56  		return err
    57  	}
    58  
    59  	if len(services) > 0 && !opts.quiet {
    60  		// only non-empty services and not quiet, should we call TaskList and NodeList api
    61  		taskFilter := filters.NewArgs()
    62  		for _, service := range services {
    63  			taskFilter.Add("service", service.ID)
    64  		}
    65  
    66  		tasks, err := client.TaskList(ctx, types.TaskListOptions{Filter: taskFilter})
    67  		if err != nil {
    68  			return err
    69  		}
    70  
    71  		nodes, err := client.NodeList(ctx, types.NodeListOptions{})
    72  		if err != nil {
    73  			return err
    74  		}
    75  
    76  		PrintNotQuiet(out, services, nodes, tasks)
    77  	} else if !opts.quiet {
    78  		// no services and not quiet, print only one line with columns ID, NAME, REPLICAS...
    79  		PrintNotQuiet(out, services, []swarm.Node{}, []swarm.Task{})
    80  	} else {
    81  		PrintQuiet(out, services)
    82  	}
    83  
    84  	return nil
    85  }
    86  
    87  // PrintNotQuiet shows service list in a non-quiet way.
    88  // Besides this, command `docker stack services xxx` will call this, too.
    89  func PrintNotQuiet(out io.Writer, services []swarm.Service, nodes []swarm.Node, tasks []swarm.Task) {
    90  	activeNodes := make(map[string]struct{})
    91  	for _, n := range nodes {
    92  		if n.Status.State != swarm.NodeStateDown {
    93  			activeNodes[n.ID] = struct{}{}
    94  		}
    95  	}
    96  
    97  	running := map[string]int{}
    98  	for _, task := range tasks {
    99  		if _, nodeActive := activeNodes[task.NodeID]; nodeActive && task.Status.State == "running" {
   100  			running[task.ServiceID]++
   101  		}
   102  	}
   103  
   104  	printTable(out, services, running)
   105  }
   106  
   107  func printTable(out io.Writer, services []swarm.Service, running map[string]int) {
   108  	writer := tabwriter.NewWriter(out, 0, 4, 2, ' ', 0)
   109  
   110  	// Ignore flushing errors
   111  	defer writer.Flush()
   112  
   113  	fmt.Fprintf(writer, listItemFmt, "ID", "NAME", "REPLICAS", "IMAGE", "COMMAND")
   114  	for _, service := range services {
   115  		replicas := ""
   116  		if service.Spec.Mode.Replicated != nil && service.Spec.Mode.Replicated.Replicas != nil {
   117  			replicas = fmt.Sprintf("%d/%d", running[service.ID], *service.Spec.Mode.Replicated.Replicas)
   118  		} else if service.Spec.Mode.Global != nil {
   119  			replicas = "global"
   120  		}
   121  		fmt.Fprintf(
   122  			writer,
   123  			listItemFmt,
   124  			stringid.TruncateID(service.ID),
   125  			service.Spec.Name,
   126  			replicas,
   127  			service.Spec.TaskTemplate.ContainerSpec.Image,
   128  			strings.Join(service.Spec.TaskTemplate.ContainerSpec.Args, " "))
   129  	}
   130  }
   131  
   132  // PrintQuiet shows service list in a quiet way.
   133  // Besides this, command `docker stack services xxx` will call this, too.
   134  func PrintQuiet(out io.Writer, services []swarm.Service) {
   135  	for _, service := range services {
   136  		fmt.Fprintln(out, service.ID)
   137  	}
   138  }