使用泛型嵌套：在函数中复用类型参数

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《使用泛型嵌套：在函数中复用类型参数》，介绍一下，希望对大家的知识积累有所帮助，助力实战开发！

给出以下定义：

type collector[t, c any] interface {
    collect(t, c) c
}

type mapcollector[k comparable, v any] struct {
    getkey func(result v) k
}

func (m *mapcollector[k, v]) collect(value v, c *sync.map) *sync.map {
    c.store(m.getkey(value), value)
    return c
}

我正在尝试为 mapcollector 编写模板化 make 函数。 这不起作用

func makeusermapcollector[k userid, v user]() *mapcollector[k, v] {
    c := mapcollector[k, v]{
        getkey: func(user v) k {
            return user.getid()
        }
    }
    return &c
}

相反，我必须重复

func MakeUserMapCollector() *mapCollector[UserId, User] {
    c := mapCollector[UserId, User]{
        GetKey: func(user User) UserId {
            return user.getId()
        }
    }
}

有没有办法重用 makeusermapcollector 类型参数来声明函数中的类型？

正确答案

您可以创建一个通用构造函数，它接受“集合”函数，然后将其传递给您的 mapcollector 结构。

看起来像这样：

func makeusermapcollector[k comparable, v any](collect func(result v) k) *mapcollector[k, v] {
    return &mapcollector[k, v]{
        getkey: collect,
    }
}

然后您可以使用您的类型作为类型参数和“集合”函数来调用构造函数：

mc := makeusermapcollector[userid, user](func(result user) userid { return result.getid() })

完整示例 (Go Playground)：

package main

import (
    "fmt"
)

type mapCollector[K comparable, V any] struct {
    GetKey func(result V) K
}

func (m mapCollector[K, V]) Collect(result V) K {
    return m.GetKey(result)
}

type UserId string

type User struct {
    Id UserId
}

func (u User) GetId() UserId {
    return u.Id
}

func MakeUserMapCollector[K comparable, V any](collect func(result V) K) *mapCollector[K, V] {
    return &mapCollector[K, V]{
        GetKey: collect,
    }
}

func main() {
    u := User{
        Id: "12",
    }

    mc := MakeUserMapCollector[UserId, User](func(result User) UserId { return result.GetId() })
    fmt.Println(mc.Collect(u))
}

好了，本文到此结束，带大家了解了《使用泛型嵌套：在函数中复用类型参数》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！