add flag of ttl for lock cmd

etcd-io · Aug 5, 2017 · 989a955 · 989a955
1 parent 366f538
commit 989a955
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/etcdctl/ctlv3/command/lock_command.go b/etcdctl/ctlv3/command/lock_command.go
@@ -28,13 +28,19 @@ 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{
 		Use:   "lock <lockname> [exec-command arg1 arg2 ...]",
 		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
 	}