dict.go 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
  1. package sprig
  2. import (
  3. "github.com/imdario/mergo"
  4. "github.com/mitchellh/copystructure"
  5. )
  6. func set(d map[string]interface{}, key string, value interface{}) map[string]interface{} {
  7. d[key] = value
  8. return d
  9. }
  10. func unset(d map[string]interface{}, key string) map[string]interface{} {
  11. delete(d, key)
  12. return d
  13. }
  14. func hasKey(d map[string]interface{}, key string) bool {
  15. _, ok := d[key]
  16. return ok
  17. }
  18. func pluck(key string, d ...map[string]interface{}) []interface{} {
  19. res := []interface{}{}
  20. for _, dict := range d {
  21. if val, ok := dict[key]; ok {
  22. res = append(res, val)
  23. }
  24. }
  25. return res
  26. }
  27. func keys(dicts ...map[string]interface{}) []string {
  28. k := []string{}
  29. for _, dict := range dicts {
  30. for key := range dict {
  31. k = append(k, key)
  32. }
  33. }
  34. return k
  35. }
  36. func pick(dict map[string]interface{}, keys ...string) map[string]interface{} {
  37. res := map[string]interface{}{}
  38. for _, k := range keys {
  39. if v, ok := dict[k]; ok {
  40. res[k] = v
  41. }
  42. }
  43. return res
  44. }
  45. func omit(dict map[string]interface{}, keys ...string) map[string]interface{} {
  46. res := map[string]interface{}{}
  47. omit := make(map[string]bool, len(keys))
  48. for _, k := range keys {
  49. omit[k] = true
  50. }
  51. for k, v := range dict {
  52. if _, ok := omit[k]; !ok {
  53. res[k] = v
  54. }
  55. }
  56. return res
  57. }
  58. func dict(v ...interface{}) map[string]interface{} {
  59. dict := map[string]interface{}{}
  60. lenv := len(v)
  61. for i := 0; i < lenv; i += 2 {
  62. key := strval(v[i])
  63. if i+1 >= lenv {
  64. dict[key] = ""
  65. continue
  66. }
  67. dict[key] = v[i+1]
  68. }
  69. return dict
  70. }
  71. func merge(dst map[string]interface{}, srcs ...map[string]interface{}) interface{} {
  72. for _, src := range srcs {
  73. if err := mergo.Merge(&dst, src); err != nil {
  74. // Swallow errors inside of a template.
  75. return ""
  76. }
  77. }
  78. return dst
  79. }
  80. func mergeOverwrite(dst map[string]interface{}, srcs ...map[string]interface{}) interface{} {
  81. for _, src := range srcs {
  82. if err := mergo.MergeWithOverwrite(&dst, src); err != nil {
  83. // Swallow errors inside of a template.
  84. return ""
  85. }
  86. }
  87. return dst
  88. }
  89. func values(dict map[string]interface{}) []interface{} {
  90. values := []interface{}{}
  91. for _, value := range dict {
  92. values = append(values, value)
  93. }
  94. return values
  95. }
  96. func deepCopy(i interface{}) interface{} {
  97. c, err := copystructure.Copy(i)
  98. if err != nil {
  99. panic("deepCopy error: " + err.Error())
  100. }
  101. return c
  102. }