fix query

longganhua

2019-07-18 c610f5adea868e6d46ac63a2b9ffeff2c24225fc

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
package command
 
import (
    "flag"
    "fmt"
    "strings"
 
    "github.com/mitchellh/cli"
)
 
// LeaveCommand is a Command implementation that instructs
// the Serf agent to gracefully leave the cluster
type LeaveCommand struct {
    Ui cli.Ui
}
 
var _ cli.Command = &LeaveCommand{}
 
func (c *LeaveCommand) Help() string {
    helpText := `
Usage: serf leave
 
  Causes the agent to gracefully leave the Serf cluster and shutdown.
 
Options:
 
  -rpc-addr=127.0.0.1:7373  RPC address of the Serf agent.
  -rpc-auth=""              RPC auth token of the Serf agent.
`
    return strings.TrimSpace(helpText)
}
 
func (c *LeaveCommand) Run(args []string) int {
    cmdFlags := flag.NewFlagSet("leave", flag.ContinueOnError)
    cmdFlags.Usage = func() { c.Ui.Output(c.Help()) }
    rpcAddr := RPCAddrFlag(cmdFlags)
    rpcAuth := RPCAuthFlag(cmdFlags)
    if err := cmdFlags.Parse(args); err != nil {
        return 1
    }
 
    client, err := RPCClient(*rpcAddr, *rpcAuth)
    if err != nil {
        c.Ui.Error(fmt.Sprintf("Error connecting to Serf agent: %s", err))
        return 1
    }
    defer client.Close()
 
    if err := client.Leave(); err != nil {
        c.Ui.Error(fmt.Sprintf("Error leaving: %s", err))
        return 1
    }
 
    c.Ui.Output("Graceful leave complete")
    return 0
}
 
func (c *LeaveCommand) Synopsis() string {
    return "Gracefully leaves the Serf cluster and shuts down"
}