Explorar el Código

etcdctlv3: handle a case of 0 arguments in put command

Hitoshi Mitake hace 10 años
padre
commit
674d54ad9b
Se han modificado 1 ficheros con 4 adiciones y 0 borrados
  1. 4 0
      etcdctlv3/command/put_command.go

+ 4 - 0
etcdctlv3/command/put_command.go

@@ -55,6 +55,10 @@ will store the content of the file to <key>.
 
 // putCommandFunc executes the "put" command.
 func putCommandFunc(cmd *cobra.Command, args []string) {
+	if len(args) == 0 {
+		ExitWithError(ExitBadArgs, fmt.Errorf("put command needs 1 argument and input from stdin or 2 arguments."))
+	}
+
 	key := []byte(args[0])
 	value, err := argOrStdin(args, os.Stdin, 1)
 	if err != nil {