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

Value.MapIndex() 返回一个 reflect.Value，它是一个结构体，而 nil 不是结构体的有效值。您无法将结构值与 nil 进行比较。

Value.MapIndex() 声明：

<块引用>

如果在映射中找不到键或者如果 v 代表一个 nil 映射，它返回零值。

因此要判断是否在映射中找不到键，请检查返回的 reflect.Value 是否为零值。为此，您可以使用 Value.IsZero() 方法。

您也不能（不应该）比较 reflect.Value 值。而是使用 Value.Interface() 获取它们的包装值，然后比较它们。

if v2 := mapSetValue.MapIndex(k); v2.IsZero() || 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

Answer 2

<块引用>

我想要一个通用方法。

如果你能等到 Go 1.18（将于 2022 年 2 月发布），该版本计划将泛型引入该语言，你将能够编写这样一个泛型函数；见下文和本。这样就不需要诉诸反思了，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
}

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))
}

输出：

true
false

Answer 3

如果有人需要，这是可行的解决方案：

// 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
}

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

3 个答案: