卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章34084本站已运行391

进一步约束Golang中的类型参数(使用Contains方法实现泛型List)

进一步约束golang中的类型参数(使用contains方法实现泛型list)

问题内容

假设我想编写一个通用的 list 类型,其中包含一些有用的方法,例如:

type list[t any] []t

func (l *list[t]) len() int
func (l *list[t]) get(pos int) (t t, err error)
func (l *list[t]) set(pos int, t t) error
func (l *list[t]) append(t ...t)
func (l *list[t]) insert(t t, pos int) error
func (l *list[t]) remove(pos int) (t t, err error)
// etc...

但是,还有其他有用的方法可能需要进一步限制列表的元素类型 t。例如,我们无法在此 list 类型上实现 contains 方法:

func (l *list[t]) contains(t t) bool {
    for _, s := range *l {
        if s == t { // compiler error: invalid operation: s == t (incomparable types in type set)
            return true
        }
    }
    return false
}

如果我们声明 list 为,我们只能实现 contains

type List[T comparable] []T

但是这使得不可能创建不可比较类型的 list

有没有办法可以两全其美?即有一个可用于不可比较类型 tlist[t] ,但在 t 具有可比性的情况下允许它有一个 contains 方法?

我想到了:

  • 使用不同类型(例如 uncomparablelist/listlist/comparablelist
  • 使 包含 为函数而不是方法

但我不太喜欢其中任何一个。


正确答案


go 没有专门化,所以我认为你不能让它完全像这样工作(尽管不是泛型专家)。

我认为解决此问题的一种合理的 go 方式是传递一个显式比较器:

func (l *list[t]) contains(t t, cmp func(t, t) bool) bool {
    for _, s := range *l {
        if cmp(s, t) {
            return true
        }
    }
    return false
}

然后你就可以了

func main() {
    list := list[int]([]int{1,2,3})
    fmt.println(list.contains(2, func(a, b int) bool { return a == b })) // true
}

对于类似的类型,您可以提供默认值:

func eq[t comparable](a, b t) bool {
    return a == b
}

所以上面就变成了

func main() {
    list := List[int]([]int{1,2,3})
    fmt.Println(list.Contains(2, Eq[int]) // true
}

您还可以在 list 类型中嵌入一个比较器,并为其指定默认值 func(a, b t) bool { return false } 并公开一个可以将自定义比较器传递到其中的构造函数。但这可能对您来说太隐晦了。

卓越飞翔博客
上一篇: 如何构建Catboost C评估库API?
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏