From 989a95566ed5e6a248f779866833f40ee20f640d Mon Sep 17 00:00:00 2001 From: Zhou Date: Sat, 5 Aug 2017 13:06:46 +0800 Subject: [PATCH] add flag of ttl for lock cmd --- etcdctl/ctlv3/command/lock_command.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/etcdctl/ctlv3/command/lock_command.go b/etcdctl/ctlv3/command/lock_command.go index e130493f81f4..963563a3da13 100644 --- a/etcdctl/ctlv3/command/lock_command.go +++ b/etcdctl/ctlv3/command/lock_command.go @@ -28,6 +28,11 @@ import ( "golang.org/x/net/context" ) +var ( + sessionTTL int + defaultSessionTTL = 10 +) + // NewLockCommand returns the cobra command for "lock". func NewLockCommand() *cobra.Command { c := &cobra.Command{ @@ -35,6 +40,7 @@ func NewLockCommand() *cobra.Command { Short: "Acquires a named lock", Run: lockCommandFunc, } + c.Flags().IntVarP(&sessionTTL, "session-ttl", "t", defaultSessionTTL, "timeout for session") return c } @@ -49,7 +55,7 @@ func lockCommandFunc(cmd *cobra.Command, args []string) { } func lockUntilSignal(c *clientv3.Client, lockname string, cmdArgs []string) error { - s, err := concurrency.NewSession(c) + s, err := concurrency.NewSession(c, concurrency.WithTTL(sessionTTL)) if err != nil { return err }