检查一个地图是否是另一个地图的子集

Question

这个问题已经用许多其他语言得到了回答。在带有简单映射（无嵌套）的 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返回类型的零值，并且无法将其与任何内容进行比较。

毕竟，同样的工作我能做得更好吗？

Answer 1

我想要一个通用方法。

现在Go 1.18和泛型已经到来，你可以编写这样的泛型函数；请参阅下文和此Playground中。通常不鼓励使用反射来实现此功能。

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仍然适用。

Answer 2

Value.MapIndex()返回 a ，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