进一步约束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 。

立即学习“go语言免费学习笔记（深入）”；

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

我想到了：

• 使用不同类型（例如 uncomparablelist/list 或 list/comparablelist）
• 使 包含 为函数而不是方法

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

蝉妈妈AI

电商人专属的AI营销助手

下载

正确答案

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 } 并公开一个可以将自定义比较器传递到其中的构造函数。但这可能对您来说太隐晦了。