rename.go 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. package mxj
  2. import (
  3. "errors"
  4. "strings"
  5. )
  6. // RenameKey renames a key in a Map.
  7. // It works only for nested maps. It doesn't work for cases when it buried in a list.
  8. func (mv Map) RenameKey(path string, newName string) error {
  9. if !mv.Exists(path) {
  10. return errors.New("RenameKey: the path not found: " + path)
  11. }
  12. if mv.Exists(parentPath(path) + "." + newName) {
  13. return errors.New("RenameKey: the key already exists: " + newName)
  14. }
  15. m := map[string]interface{}(mv)
  16. return renameKey(m, path, newName)
  17. }
  18. func renameKey(m interface{}, path string, newName string) error {
  19. val, err := prevValueByPath(m, path)
  20. if err != nil {
  21. return err
  22. }
  23. oldName := lastKey(path)
  24. val[newName] = val[oldName]
  25. delete(val, oldName)
  26. return nil
  27. }
  28. // returns a value which contains a last key in the path
  29. // For example: prevValueByPath("a.b.c", {a{b{c: 3}}}) returns {c: 3}
  30. func prevValueByPath(m interface{}, path string) (map[string]interface{}, error) {
  31. keys := strings.Split(path, ".")
  32. switch mValue := m.(type) {
  33. case map[string]interface{}:
  34. for key, value := range mValue {
  35. if key == keys[0] {
  36. if len(keys) == 1 {
  37. return mValue, nil
  38. } else {
  39. // keep looking for the full path to the key
  40. return prevValueByPath(value, strings.Join(keys[1:], "."))
  41. }
  42. }
  43. }
  44. }
  45. return nil, errors.New("prevValueByPath: didn't find the path – " + path)
  46. }