open-consul/command/info.go
2014-04-29 11:09:14 -07:00

97 lines
2 KiB
Go

package command
import (
"flag"
"fmt"
"github.com/mitchellh/cli"
"sort"
"strings"
)
// InfoCommand is a Command implementation that queries a running
// Consul agent for various debugging statistics for operators
type InfoCommand struct {
Ui cli.Ui
}
func (i *InfoCommand) Help() string {
helpText := `
Usage: consul info [options]
Provides debugging information for operators
Options:
-rpc-addr=127.0.0.1:8400 RPC address of the Consul agent.
`
return strings.TrimSpace(helpText)
}
func (i *InfoCommand) Run(args []string) int {
cmdFlags := flag.NewFlagSet("info", flag.ContinueOnError)
cmdFlags.Usage = func() { i.Ui.Output(i.Help()) }
rpcAddr := RPCAddrFlag(cmdFlags)
if err := cmdFlags.Parse(args); err != nil {
return 1
}
client, err := RPCClient(*rpcAddr)
if err != nil {
i.Ui.Error(fmt.Sprintf("Error connecting to Consul agent: %s", err))
return 1
}
defer client.Close()
stats, err := client.Stats()
if err != nil {
i.Ui.Error(fmt.Sprintf("Error querying agent: %s", err))
return 1
}
// Check for specific warnings
didWarn := false
runtime, ok := stats["runtime"]
if ok {
if maxprocs := runtime["max_procs"]; maxprocs == "1" {
i.Ui.Output("WARNING: It is highly recommended to set GOMAXPROCS higher than 1")
didWarn = true
}
}
// Add a blank line if there are any warnings
if didWarn {
i.Ui.Output("")
}
// Get the keys in sorted order
keys := make([]string, 0, len(stats))
for key := range stats {
keys = append(keys, key)
}
sort.Strings(keys)
// Iterate over each top-level key
for _, key := range keys {
i.Ui.Output(key + ":")
// Sort the sub-keys
subvals := stats[key]
subkeys := make([]string, 0, len(subvals))
for k := range subvals {
subkeys = append(subkeys, k)
}
sort.Strings(subkeys)
// Iterate over the subkeys
for _, subkey := range subkeys {
val := subvals[subkey]
i.Ui.Output(fmt.Sprintf("\t%s = %s", subkey, val))
}
}
return 0
}
func (i *InfoCommand) Synopsis() string {
return "Provides debugging information for operators"
}