/
cluster.go
94 lines (83 loc) · 2.02 KB
/
cluster.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
94
package cmd
import (
json "github.com/dutchcoders/elastico/json"
"github.com/codegangsta/cli"
)
var (
_ = registerTemplate("cluster:state", `=== Cluster state
cluster name {{ .cluster_name }}
master node {{ .master_node }}
= Nodes
{{range $name, $node := .nodes -}}
{{ $name }} {{ $node.name }} {{ $node.transport_address }}
{{end}}
`)
_ = registerTemplate("cluster:health", `=== Cluster health
name {{ .cluster_name }}
{{- if eq .status "green" }}
status {{ .status | green }}
{{ end -}}
{{- if eq .status "yellow" }}
status {{ .status | yellow }}
{{ end -}}
{{- if eq .status "red" }}
status {{ .status | red }}
{{ end -}}
# nodes {{ .number_of_nodes }}
# data nodes {{ .number_of_data_nodes }}
# active shards {{ .active_shards }}
# active primary shards {{ .active_primary_shards }}
# initializing shards {{ .initializing_shards }}
# relocating shards {{ .relocating_shards }}
# unassigned shards {{ .unassigned_shards }}
`)
)
var clusterCmds = []cli.Command{
cli.Command{
Name: "cluster:state",
Usage: "Retrieve cluster state",
Description: ``,
Action: run(runClusterState),
Flags: []cli.Flag{},
},
cli.Command{
Name: "cluster:health",
Usage: "Retrieve cluster health",
Description: ``,
Action: run(runClusterHealth),
Flags: []cli.Flag{
cli.StringFlag{
Name: "type",
Value: "fs",
},
cli.BoolFlag{
Name: "compress",
},
cli.StringFlag{
Name: "location",
},
},
},
}
func runClusterState(c *cli.Context) (json.M, error) {
req, err := e.NewRequest("GET", "/_cluster/state", nil)
if err != nil {
return nil, err
}
var resp json.M
if err := e.Do(req, &resp); err != nil {
return nil, err
}
return resp, nil
}
func runClusterHealth(c *cli.Context) (json.M, error) {
req, err := e.NewRequest("GET", "/_cluster/health", nil)
if err != nil {
return nil, err
}
var resp json.M
if err := e.Do(req, &resp); err != nil {
return nil, err
}
return resp, nil
}