github.com/portworx/docker@v1.12.1/api/client/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/client"
    10  	"github.com/docker/docker/cli"
    11  	"github.com/docker/docker/opts"
    12  	"github.com/docker/docker/pkg/stringid"
    13  	"github.com/docker/engine-api/types"
    14  	"github.com/docker/engine-api/types/filters"
    15  	"github.com/docker/engine-api/types/swarm"
    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 *client.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 *client.DockerCli, opts listOptions) error {
    50  	ctx := context.Background()
    51  	client := dockerCli.Client()
    52  
    53  	services, err := client.ServiceList(ctx, types.ServiceListOptions{Filter: opts.filter.Value()})
    54  	if err != nil {
    55  		return err
    56  	}
    57  
    58  	out := dockerCli.Out()
    59  	if opts.quiet {
    60  		printQuiet(out, services)
    61  	} else {
    62  		taskFilter := filters.NewArgs()
    63  		for _, service := range services {
    64  			taskFilter.Add("service", service.ID)
    65  		}
    66  
    67  		tasks, err := client.TaskList(ctx, types.TaskListOptions{Filter: taskFilter})
    68  		if err != nil {
    69  			return err
    70  		}
    71  
    72  		nodes, err := client.NodeList(ctx, types.NodeListOptions{})
    73  		if err != nil {
    74  			return err
    75  		}
    76  		activeNodes := make(map[string]struct{})
    77  		for _, n := range nodes {
    78  			if n.Status.State == swarm.NodeStateReady {
    79  				activeNodes[n.ID] = struct{}{}
    80  			}
    81  		}
    82  
    83  		running := map[string]int{}
    84  		for _, task := range tasks {
    85  			if _, nodeActive := activeNodes[task.NodeID]; nodeActive && task.Status.State == "running" {
    86  				running[task.ServiceID]++
    87  			}
    88  		}
    89  
    90  		printTable(out, services, running)
    91  	}
    92  	return nil
    93  }
    94  
    95  func printTable(out io.Writer, services []swarm.Service, running map[string]int) {
    96  	writer := tabwriter.NewWriter(out, 0, 4, 2, ' ', 0)
    97  
    98  	// Ignore flushing errors
    99  	defer writer.Flush()
   100  
   101  	fmt.Fprintf(writer, listItemFmt, "ID", "NAME", "REPLICAS", "IMAGE", "COMMAND")
   102  	for _, service := range services {
   103  		replicas := ""
   104  		if service.Spec.Mode.Replicated != nil && service.Spec.Mode.Replicated.Replicas != nil {
   105  			replicas = fmt.Sprintf("%d/%d", running[service.ID], *service.Spec.Mode.Replicated.Replicas)
   106  		} else if service.Spec.Mode.Global != nil {
   107  			replicas = "global"
   108  		}
   109  		fmt.Fprintf(
   110  			writer,
   111  			listItemFmt,
   112  			stringid.TruncateID(service.ID),
   113  			service.Spec.Name,
   114  			replicas,
   115  			service.Spec.TaskTemplate.ContainerSpec.Image,
   116  			strings.Join(service.Spec.TaskTemplate.ContainerSpec.Args, " "))
   117  	}
   118  }
   119  
   120  func printQuiet(out io.Writer, services []swarm.Service) {
   121  	for _, service := range services {
   122  		fmt.Fprintln(out, service.ID)
   123  	}
   124  }