如何在 Golang 函数中使用泛型

泛型函数允许我们定义可重用的代码,用于各种类型的数据,简化了类似功能函数的实现并减少重复代码。go 1.18 引入了泛型功能,允许我们使用类型参数指定泛型函数。我们可以使用类型约束来限制泛型函数接受的类型。通过使用泛型,我们可以合并对不同类型进行排序的多个函数,创建一个适用于任何可排序类型的通用函数。

如何在 Golang 函数中使用泛型

在 Go 函数中使用泛型

什么是泛型?

泛型函数允许我们为各种类型的数据定义可重用的代码。这可以简化实现类似功能的函数的代码,同时减少重复代码。

在 Go 中使用泛型

Go 1.18 引入了泛型功能。要定义泛型函数,我们使用方括号 将类型参数指定为函数的参数。例如,以下函数可以用于比较任何可比较类型的两个值:

func Min[T comparable](a, b T) T {
    if a < b {
        return a
    }
    return b
}

类型约束

我们还可以使用类型约束来限制泛型函数接受的类型。例如,以下函数只能比较实现 fmt.Stringer 接口的类型:

func MinStringer[T fmt.Stringer](a, b T) T {
    if a.String() < b.String() {
        return a
    }
    return b
}

实战案例

让我们看一个使用泛型的实际示例。以下是使用 sort 包对切片进行基本排序的两个泛型函数:

// SortInts 使用快速排序算法对 []int 进行排序
func SortInts(a []int) {
    sort.Slice(a, func(i, j int) bool { return a[i] < a[j] })
}

// SortStrings 使用快速排序算法对 []string 进行排序
func SortStrings(a []string) {
    sort.Slice(a, func(i, j int) bool { return a[i] < a[j] })
}

使用泛型,我们可以将这两个函数合并为一个函数,适用于任何可排序的类型:

func Sort[T sort.Interface](a []T) {
    sort.Slice(a, func(i, j int) bool { return a[i].Less(a[j]) })
}

现在,我们可以对任何实现了 sort.Interface 接口的类型(例如 []int 和 []string)使用 Sort 函数。

以上就是如何在 Golang 函数中使用泛型的详细内容,更多请关注其它相关文章!