登录
首页 >  Golang >  Go问答

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

来源:stackoverflow

时间:2024-03-05 14:21:25 250浏览 收藏

小伙伴们有没有觉得学习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学习网公众号!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>