internal/cli/nodes.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
package cli
import (
"bytes"
"flag"
"fmt"
"os"
"sort"
"strings"
"text/tabwriter"
"j3s.sh/cascade/api"
)
type nodesCommand struct {
flagAPIAddr string
}
func (c nodesCommand) Usage() {
fmt.Printf(`usage: cascade nodes [flags]
list the nodes in the cascade cluster.
flags:
-api
address of the cascade http api to target (default = 127.0.0.1:8500)
`)
}
func (c *nodesCommand) 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 RunNodes(args []string) {
c := nodesCommand{}
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)
}
nodes, err := client.Agent().Members()
if err != nil {
fmt.Println(err)
os.Exit(1)
}
// Fetch the full instance list once; group by owning node.
instances, _ := client.Catalog().Instances()
servicesByNode := map[string][]string{}
for _, inst := range instances {
servicesByNode[inst.Node] = append(servicesByNode[inst.Node], inst.ServiceName)
}
for n := range servicesByNode {
sort.Strings(servicesByNode[n])
}
var b bytes.Buffer
tw := tabwriter.NewWriter(&b, 0, 2, 2, ' ', 0)
fmt.Fprintf(tw, "node\taddr\tstatus\tservices\n")
for _, n := range nodes {
services := "<none>"
if names := servicesByNode[n.Name]; len(names) > 0 {
services = strings.Join(names, ", ")
}
fmt.Fprintf(tw, "%s\t%s:%d\t%s\t%s\n", n.Name, n.Addr, n.Port, n.StatusPretty(), services)
}
if err := tw.Flush(); err != nil {
fmt.Printf("error flushing tabwriter: %s", err)
os.Exit(1)
}
fmt.Print(b.String())
}