mirror of
https://github.com/k3s-io/k3s.git
synced 2024-06-07 19:41:36 +00:00
78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
|
package objx
|
||
|
|
||
|
// Exclude returns a new Map with the keys in the specified []string
|
||
|
// excluded.
|
||
|
func (m Map) Exclude(exclude []string) Map {
|
||
|
excluded := make(Map)
|
||
|
for k, v := range m {
|
||
|
if !contains(exclude, k) {
|
||
|
excluded[k] = v
|
||
|
}
|
||
|
}
|
||
|
return excluded
|
||
|
}
|
||
|
|
||
|
// Copy creates a shallow copy of the Obj.
|
||
|
func (m Map) Copy() Map {
|
||
|
copied := Map{}
|
||
|
for k, v := range m {
|
||
|
copied[k] = v
|
||
|
}
|
||
|
return copied
|
||
|
}
|
||
|
|
||
|
// Merge blends the specified map with a copy of this map and returns the result.
|
||
|
//
|
||
|
// Keys that appear in both will be selected from the specified map.
|
||
|
// This method requires that the wrapped object be a map[string]interface{}
|
||
|
func (m Map) Merge(merge Map) Map {
|
||
|
return m.Copy().MergeHere(merge)
|
||
|
}
|
||
|
|
||
|
// MergeHere blends the specified map with this map and returns the current map.
|
||
|
//
|
||
|
// Keys that appear in both will be selected from the specified map. The original map
|
||
|
// will be modified. This method requires that
|
||
|
// the wrapped object be a map[string]interface{}
|
||
|
func (m Map) MergeHere(merge Map) Map {
|
||
|
for k, v := range merge {
|
||
|
m[k] = v
|
||
|
}
|
||
|
return m
|
||
|
}
|
||
|
|
||
|
// Transform builds a new Obj giving the transformer a chance
|
||
|
// to change the keys and values as it goes. This method requires that
|
||
|
// the wrapped object be a map[string]interface{}
|
||
|
func (m Map) Transform(transformer func(key string, value interface{}) (string, interface{})) Map {
|
||
|
newMap := Map{}
|
||
|
for k, v := range m {
|
||
|
modifiedKey, modifiedVal := transformer(k, v)
|
||
|
newMap[modifiedKey] = modifiedVal
|
||
|
}
|
||
|
return newMap
|
||
|
}
|
||
|
|
||
|
// TransformKeys builds a new map using the specified key mapping.
|
||
|
//
|
||
|
// Unspecified keys will be unaltered.
|
||
|
// This method requires that the wrapped object be a map[string]interface{}
|
||
|
func (m Map) TransformKeys(mapping map[string]string) Map {
|
||
|
return m.Transform(func(key string, value interface{}) (string, interface{}) {
|
||
|
if newKey, ok := mapping[key]; ok {
|
||
|
return newKey, value
|
||
|
}
|
||
|
return key, value
|
||
|
})
|
||
|
}
|
||
|
|
||
|
// Checks if a string slice contains a string
|
||
|
func contains(s []string, e string) bool {
|
||
|
for _, a := range s {
|
||
|
if a == e {
|
||
|
return true
|
||
|
}
|
||
|
}
|
||
|
return false
|
||
|
}
|