internal/cli/services.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package cli
import (
"bytes"
"flag"
"fmt"
"net"
"os"
"sort"
"strconv"
"strings"
"text/tabwriter"
"git.j3s.sh/cascade/api"
)
type servicesCommand struct {
flagAPIAddr string
}
func (c servicesCommand) Usage() {
fmt.Printf(`usage: cascade services [flags]
list the services registered to a cascade agent.
flags:
-api
address of the cascade http api to target (default = 127.0.0.1:8500)
`)
}
func (c *servicesCommand) Init(args []string) {
flags := flag.NewFlagSet("", flag.ContinueOnError)
flags.Usage = c.Usage
flags.StringVar(&c.flagAPIAddr, "api", "", "")
if err := flags.Parse(args); err != nil {
fmt.Println(err)
os.Exit(1)
}
}
func RunServices(args []string) {
c := servicesCommand{}
c.Init(args)
cfg := api.DefaultConfig()
if c.flagAPIAddr != "" {
cfg.Address = c.flagAPIAddr
}
client, err := api.NewClient(cfg)
if err != nil {
fmt.Println(err)
os.Exit(1)
}
services, err := client.Agent().Services()
if err != nil {
fmt.Println(err)
os.Exit(1)
}
if len(services) == 0 {
fmt.Printf("no services registered on %s\n", cfg.Address)
return
}
ids := make([]string, 0, len(services))
for id := range services {
ids = append(ids, id)
}
sort.Strings(ids)
nodeName, err := client.Agent().NodeName()
if err != nil {
fmt.Println(err)
os.Exit(1)
}
var b bytes.Buffer
tw := tabwriter.NewWriter(&b, 0, 2, 2, ' ', 0)
fmt.Fprintf(tw, "node\tid\tname\taddr\ttags\n")
for _, id := range ids {
s := services[id]
addr := s.Address
if s.Port != 0 {
addr = net.JoinHostPort(addr, strconv.Itoa(s.Port))
}
fmt.Fprintf(tw, "%s\t%s\t%s\t%s\t%s\n", nodeName, s.ID, s.Service, addr, strings.Join(s.Tags, ","))
}
if err := tw.Flush(); err != nil {
fmt.Printf("error flushing tabwriter: %s", err)
os.Exit(1)
}
fmt.Print(b.String())
}