定义结构作为键的映射的自定义相等性 - 如何为映射中的结构键制定自定义相等规则？

小伙伴们有没有觉得学习Golang很有意思？有意思就对了！今天就给大家带来《定义结构作为键的映射的自定义相等性 - 如何为映射中的结构键制定自定义相等规则？》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

给定一个使用结构体作为键的映射，其中结构体的值是指向另一个结构体的指针：

type dog struct {
    name string
}

type cat struct {
    name string
}

type mapkey struct {
    dog *dog
    cat *cat
}

mymap := make(map[mapkey]int)

如何使用 cmp 包使以下地图相等，因为 mapkey 具有相同的值（reflect.deepequals 或 cmp.equals），因此它们被视为相等？

keyOne := MapKey{
    &Dog{Name: "bob"},
    &Cat{Name: "clive"},
}
keyTwo := MapKey{
    &Dog{Name: "bob"},
    &Cat{Name: "clive"},
}

got := map[MapKey]int{
    keyOne: 1,
}
want := map[MapKey]int{
    keyTwo: 1,
}

在 cmp 文档中，它说我可以使用 cmpopts.sortmaps (https://pkg.go.dev/github.com/google/go-cmp/cmp#equal)，但是我不明白这是怎么回事与我的场景相关。

我尝试在 mapkey 结构上定义自定义 equals 函数，但它从未被调用。

通过演示来重现此内容： https://go.dev/play/p/qmxaya3s26m

正确答案

使用 map[mapkey]int 类型的参数调用 cmp.equal，而不是 mapkey

因此必须在类型 map[mapkey]int 上定义自定义 equal 函数。

但是要定义这个函数，我们需要从 map[mapkey]int 定义一个新类型。

这是完整工作示例的演示：https://go.dev/play/p/deteHANWQ_3

type MapKeyInt map[MapKey]int

func (m MapKeyInt) Equal(other MapKeyInt) bool {
    if len(m) != len(other) {
        return false
    }

    keys, keysOther := make([]MapKey, 0), make([]MapKey, 0)
    values, valuesOther := make([]int, 0), make([]int, 0)

    for k, v := range m {
        keys = append(keys, k)
        values = append(values, v)
    }
    for k, v := range other {
        keysOther = append(keysOther, k)
        valuesOther = append(valuesOther, v)
    }

    for i := 0; i < len(m); i++ {
        if (keys[i].dog.Name != keysOther[i].dog.Name) || (keys[i].cat.Name != keysOther[i].cat.Name) {
            return false
        }
        if values[i] != valuesOther[i] {
            return false
        }
    }

    return true
}

以上就是《定义结构作为键的映射的自定义相等性 - 如何为映射中的结构键制定自定义相等规则？》的详细内容，更多关于的资料请关注golang学习网公众号！