mkdir_command.go 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  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. "time"
  18. "github.com/urfave/cli"
  19. "go.etcd.io/etcd/client"
  20. )
  21. // NewMakeDirCommand returns the CLI command for "mkdir".
  22. func NewMakeDirCommand() cli.Command {
  23. return cli.Command{
  24. Name: "mkdir",
  25. Usage: "make a new directory",
  26. ArgsUsage: "<key>",
  27. Flags: []cli.Flag{
  28. cli.IntFlag{Name: "ttl", Value: 0, Usage: "key time-to-live in seconds"},
  29. },
  30. Action: func(c *cli.Context) error {
  31. mkdirCommandFunc(c, mustNewKeyAPI(c), client.PrevNoExist)
  32. return nil
  33. },
  34. }
  35. }
  36. // mkdirCommandFunc executes the "mkdir" command.
  37. func mkdirCommandFunc(c *cli.Context, ki client.KeysAPI, prevExist client.PrevExistType) {
  38. if len(c.Args()) == 0 {
  39. handleError(c, ExitBadArgs, errors.New("key required"))
  40. }
  41. key := c.Args()[0]
  42. ttl := c.Int("ttl")
  43. ctx, cancel := contextWithTotalTimeout(c)
  44. resp, err := ki.Set(ctx, key, "", &client.SetOptions{TTL: time.Duration(ttl) * time.Second, Dir: true, PrevExist: prevExist})
  45. cancel()
  46. if err != nil {
  47. handleError(c, ExitServerError, err)
  48. }
  49. if c.GlobalString("output") != "simple" {
  50. printResponseKey(resp, c.GlobalString("output"))
  51. }
  52. }