mk_command.go 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. // Copyright 2015 The etcd Authors
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package command
  15. import (
  16. "errors"
  17. "os"
  18. "time"
  19. "github.com/urfave/cli"
  20. "go.etcd.io/etcd/client"
  21. )
  22. // NewMakeCommand returns the CLI command for "mk".
  23. func NewMakeCommand() cli.Command {
  24. return cli.Command{
  25. Name: "mk",
  26. Usage: "make a new key with a given value",
  27. ArgsUsage: "<key> <value>",
  28. Flags: []cli.Flag{
  29. cli.BoolFlag{Name: "in-order", Usage: "create in-order key under directory <key>"},
  30. cli.IntFlag{Name: "ttl", Value: 0, Usage: "key time-to-live in seconds"},
  31. },
  32. Action: func(c *cli.Context) error {
  33. mkCommandFunc(c, mustNewKeyAPI(c))
  34. return nil
  35. },
  36. }
  37. }
  38. // mkCommandFunc executes the "mk" command.
  39. func mkCommandFunc(c *cli.Context, ki client.KeysAPI) {
  40. if len(c.Args()) == 0 {
  41. handleError(c, ExitBadArgs, errors.New("key required"))
  42. }
  43. key := c.Args()[0]
  44. value, err := argOrStdin(c.Args(), os.Stdin, 1)
  45. if err != nil {
  46. handleError(c, ExitBadArgs, errors.New("value required"))
  47. }
  48. ttl := c.Int("ttl")
  49. inorder := c.Bool("in-order")
  50. var resp *client.Response
  51. ctx, cancel := contextWithTotalTimeout(c)
  52. if !inorder {
  53. // Since PrevNoExist means that the Node must not exist previously,
  54. // this Set method always creates a new key. Therefore, mk command
  55. // succeeds only if the key did not previously exist, and the command
  56. // prevents one from overwriting values accidentally.
  57. resp, err = ki.Set(ctx, key, value, &client.SetOptions{TTL: time.Duration(ttl) * time.Second, PrevExist: client.PrevNoExist})
  58. } else {
  59. // If in-order flag is specified then create an inorder key under
  60. // the directory identified by the key argument.
  61. resp, err = ki.CreateInOrder(ctx, key, value, &client.CreateInOrderOptions{TTL: time.Duration(ttl) * time.Second})
  62. }
  63. cancel()
  64. if err != nil {
  65. handleError(c, ExitServerError, err)
  66. }
  67. printResponseKey(resp, c.GlobalString("output"))
  68. }