在 Go 中，函数参数可以有两种不同类型吗？

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《在 Go 中，函数参数可以有两种不同类型吗？》，聊聊，希望可以帮助到正在努力赚钱的你。

你能用go写一个参数可以是两种不同类型的函数吗？例如，如果我编写一个简单的函数，它接受 int 的数组/切片并简单地返回第一个值：

func first(array []int) int {
  return array[0]
}

有没有办法输入这个，这样我们也可以传入 []string 等？例如，在 typescript 中，我们可以这样做，而不必将数组键入为 any：

const first = (array: (number | string)[]): number | string => {
  return array[0];
};

我见过解释在这种情况下使用 interface{} 的答案...也许这是唯一的方法，但它似乎接近 ts 中的 any ，感觉可能有更好的方法这样做。

正确答案

（我已经有几年没有使用 go 了，早在他们引入泛型之前 - 所以我的答案基于 their documentation for generics）

typescript 的“泛型”（用引号引起来）与 go 的泛型无法真正相比。 typescript 的全部意义在于能够为围绕类型擦除（即 javascript）构建的运行时系统描述一个接口（抽象意义上的），而 go 则是......_{老实说，我不知道。我只是不知道 go 的泛型是如何实现的，也不知道它们的运行时特性：Articles on Go's generics、even from the language authors blog 或 their own documentation site 没有提及像类型擦除、具体化泛型，（模板）实例化或单态，所以如果有人有更好的理解请编辑这篇文章，或在评论中告诉我！}

无论如何，好消息是，从 go 1.18 beta 开始，它对泛型的支持包括对泛型约束的支持，而且还支持联合类型作为泛型类型参数的约束（尽管我尚未找到任何有关支持其他 adt（例如产品类型和交叉点类型）的信息。

_{（请注意，至少目前，go 不支持联合类型作为具体类型，但实际上这不应该成为问题）} p>

在你的情况下，如果你想要一个返回切片的第一个元素的函数，该元素可以是 []int 或 []string （或者如果切片为空则返回一些默认值），那么你可以这样做：

func first[t int | string](arr []t, ifempty t) t {
    for _, v := range arr {
        return v
    }

    return ifempty
}

虽然乍一看您可能认为这将允许 arr 切片同时包含 int 和 string 值，但这是不允许的（见下文）。请记住，泛型参数参数是由调用者提供的，并且必须是有效的具体类型，因此 first 只能实例化为 first[int] 或 first[string]，这又意味着arr 可以是 []int 或 []string。

无论如何，下面的完整示例在 the Go playground 上的 go dev 分支 中编译并运行：

package main

import "fmt"

func first[t int | string](arr []t, ifempty t) t {
    for _, v := range arr {
        return v
    }

    return ifempty
}

func main() {

    // first[int]:
    arrayofints := []int{2, 3, 5, 7, 11, 13}

    firstint := first(arrayofints, -1)
    fmt.println(firstint) // 2

    // first[string]:
    arrayofstrings := []string{"life", "is short", "and love is always over", "in the morning"}
    
    firststring := first(arrayofstrings, "empty")
    fmt.println(firststring) // "life"

}

您还可以提取约束 t int | string 并将其移动到一个接口，然后使用它作为约束，我个人认为这样更容易阅读，特别是当您可能需要在多个地方重复相同的约束时：

type intorstring interface {
    int | string
}

func first[t intorstring](arr []t, ifempty t) t {
    for _, v := range arr {
        return v
    }

    return ifempty
}

你不能做的事情...

请注意，go（至少目前）不允许使用 type 将联合描述为变量的类型本身（也不能用作切片的元素类型）；您只能使用联合作为约束，否则您将收到“接口包含类型约束”错误。其中 means you can't describe an array 可以同时包含 int 和 string 值，然后将该接口用于具体的数组类型：

package main

import "fmt"

type IntOrString interface {
    int | string
}

func First[T IntOrString](arr []T, ifEmpty T) T {
    for _, v := range arr {
        return v
    }

    return ifEmpty
}

func main() {

    arrayOfIntsOrStrings := []IntOrString{2, "foo", 3, "bar", 5, "baz", 7, 11, 13} // ERROR: interface contains type constraints

    firstValue := First(arrayOfIntsOrStrings, -1)
    fmt.Println(firstValue)
}

终于介绍完啦！小伙伴们，这篇关于《在 Go 中，函数参数可以有两种不同类型吗？》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！