Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

ctlv3: support TLS endpoints for move-leader command

Browse Source 
Signed-off-by: Gyuho Lee <gyuhox@gmail.com>
This commit is contained in:
Gyuho Lee 2018-06-05 13:36:04 -07:00
parent 6d9cb7ac9a
commit 65192fddfa
2 changed files with 9 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
etcdctl/ctlv3/command/global.go
View File

@ -112,11 +112,12 @@ func (*discardValue) Type() string { return "" }
func clientConfigFromCmd(cmd *cobra.Command) *clientConfig { func clientConfigFromCmd(cmd *cobra.Command) *clientConfig {
fs := cmd.InheritedFlags() fs := cmd.InheritedFlags()
if strings.HasPrefix(cmd.Use, "watch") {
// silence "pkg/flags: unrecognized environment variable ETCDCTL_WATCH_KEY=foo" warnings // silence "pkg/flags: unrecognized environment variable ETCDCTL_WATCH_KEY=foo" warnings
// silence "pkg/flags: unrecognized environment variable ETCDCTL_WATCH_RANGE_END=bar" warnings // silence "pkg/flags: unrecognized environment variable ETCDCTL_WATCH_RANGE_END=bar" warnings
fs.AddFlag(&pflag.Flag{Name: "watch-key", Value: &discardValue{}}) fs.AddFlag(&pflag.Flag{Name: "watch-key", Value: &discardValue{}})
fs.AddFlag(&pflag.Flag{Name: "watch-range-end", Value: &discardValue{}}) fs.AddFlag(&pflag.Flag{Name: "watch-range-end", Value: &discardValue{}})
}
flags.SetPflagsFromEnv("ETCDCTL", fs) flags.SetPflagsFromEnv("ETCDCTL", fs)
debug, err := cmd.Flags().GetBool("debug") debug, err := cmd.Flags().GetBool("debug")

11
etcdctl/ctlv3/command/move_leader_command.go
View File

@ -17,7 +17,6 @@ package command
import ( import (
"fmt" "fmt"
"strconv" "strconv"
"time"
"github.com/coreos/etcd/clientv3" "github.com/coreos/etcd/clientv3"
"github.com/spf13/cobra" "github.com/spf13/cobra"
@ -53,13 +52,9 @@ func transferLeadershipCommandFunc(cmd *cobra.Command, args []string) {
var leaderCli *clientv3.Client var leaderCli *clientv3.Client
var leaderID uint64 var leaderID uint64
for _, ep := range eps { for _, ep := range eps {
cli, cerr := clientv3.New(clientv3.Config{ cfg := clientConfigFromCmd(cmd)
Endpoints: []string{ep}, cfg.endpoints = []string{ep}
DialTimeout: 3 * time.Second, cli := cfg.mustClient()
})
if cerr != nil {
ExitWithError(ExitError, cerr)
}
resp, serr := cli.Status(ctx, ep) resp, serr := cli.Status(ctx, ep)
if serr != nil { if serr != nil {
ExitWithError(ExitError, serr) ExitWithError(ExitError, serr)