这个问题已经在许多其他语言中得到了回答。在具有简单映射(无嵌套(的golang中,如何找出一个映射是否是另一个映射的子集。例如:map[string]string{"a": "b", "e": "f"}
是map[string]string{"a": "b", "c": "d", "e": "f"}
的子集。我想要一个通用方法。我的代码:
package main
import (
"fmt"
"reflect"
)
func main() {
a := map[string]string{"a": "b", "c": "d", "e": "f"}
b := map[string]string{"a": "b", "e": "f"}
c := IsMapSubset(a, b)
fmt.Println(c)
}
func IsMapSubset(mapSet interface{}, mapSubset interface{}) bool {
mapSetValue := reflect.ValueOf(mapSet)
mapSubsetValue := reflect.ValueOf(mapSubset)
if mapSetValue.Kind() != reflect.Map || mapSubsetValue.Kind() != reflect.Map {
return false
}
if reflect.TypeOf(mapSetValue) != reflect.TypeOf(mapSubsetValue) {
return false
}
if len(mapSubsetValue.MapKeys()) == 0 {
return true
}
iterMapSubset := mapSubsetValue.MapRange()
for iterMapSubset.Next() {
k := iterMapSubset.Key()
v := iterMapSubset.Value()
if value := mapSetValue.MapIndex(k); value == nil || v != value { // invalid: value == nil
return false
}
}
return true
}
当我想检查集合映射中是否存在子集映射键时,MapIndex
返回类型的零值,使其无法与任何东西进行比较。
毕竟,我能把同样的工作做得更好吗?
我想要一个通用方法。
现在Go 1.18和泛型都在这里了,您可以编写这样一个泛型函数;请参阅下文和本游乐场。通常不鼓励使用反射来实现此功能。
package main
import "fmt"
func IsMapSubset[K, V comparable](m, sub map[K]V) bool {
if len(sub) > len(m) {
return false
}
for k, vsub := range sub {
if vm, found := m[k]; !found || vm != vsub {
return false
}
}
return true
}
type MyMap map[string]string
func main() {
a := map[string]string{"a": "b", "c": "d", "e": "f"}
b := map[string]string{"a": "b", "e": "f"}
c := map[string]string{"a": "b", "e": "g"}
fmt.Println(IsMapSubset(a, b))
fmt.Println(IsMapSubset(a, c))
fmt.Println(IsMapSubset(MyMap(a), c))
}
输出:
true
false
不过,关于NaN
的常见注意事项适用。
Value.MapIndex()
返回一个结构体reflect.Value
,而nil
不是结构体的有效值。无法将结构值与nil
进行比较。
Value.MapIndex()
指出:
如果在映射中找不到键,或者v表示nil映射,则返回零值。
因此,要判断是否在映射中未找到密钥,请检查返回的reflect.Value
是否为零值。为此,您可以使用Value.IsValid()
方法。
您也不能(不应该(比较reflect.Value
值。相反,使用Value.Interface()
获得它们的包装值,并对它们进行比较。
if v2 := mapSetValue.MapIndex(k); !v2.IsValid() || v.Interface() != v2.Interface() {
return false
}
测试:
a := map[string]string{"a": "b", "c": "d", "e": "f"}
b := map[string]string{"a": "b", "e": "f"}
fmt.Println(IsMapSubset(a, b))
c := map[string]string{"a": "b", "e": "X"}
fmt.Println(IsMapSubset(a, c))
输出将是(在Go Playground上尝试(:
true
false
如果有人需要,这是一个有效的解决方案:
// IsMapSubset returns true if mapSubset is a subset of mapSet otherwise false
func IsMapSubset(mapSet interface{}, mapSubset interface{}) bool {
mapSetValue := reflect.ValueOf(mapSet)
mapSubsetValue := reflect.ValueOf(mapSubset)
if fmt.Sprintf("%T", mapSet) != fmt.Sprintf("%T", mapSubset) {
return false
}
if len(mapSetValue.MapKeys()) < len(mapSubsetValue.MapKeys()) {
return false
}
if len(mapSubsetValue.MapKeys()) == 0 {
return true
}
iterMapSubset := mapSubsetValue.MapRange()
for iterMapSubset.Next() {
k := iterMapSubset.Key()
v := iterMapSubset.Value()
value := mapSetValue.MapIndex(k)
if !value.IsValid() || v.Interface() != value.Interface() {
return false
}
}
return true
}