泛型类型推断在函数级联调用中的应用

小伙伴们有没有觉得学习Golang很有意思？有意思就对了！今天就给大家带来《泛型类型推断在函数级联调用中的应用》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

在构建（我的第一个）泛型重型库时，我遇到了泛型类型检查实现的一些明显限制 - 更可能是我缺乏知识。

有什么想法可以让下面的东西发挥作用吗？

package main

import (
    "fmt"
    "reflect"
)

type Number interface {
    int | float32
}

type MultiDimensionSlice interface {
    int | float32 | []int | []float32 | [][]int | [][]float32
}

func dimensions[S MultiDimensionSlice](s S) int {
    dims := 0
    t := reflect.TypeOf(s)
    for t.Kind() == reflect.Slice {
        dims += 1
        t = t.Elem()
    }
    return dims
}

func indirection[T Number](v T) int {
    var slice2D [][]T
    return dimensions(slice2D)
}

func main() {
    x := [][]float32{{1}, {2}, {3}}
    fmt.Printf("x=%v, dims=%d\n", x, dimensions(x))
    fmt.Printf("indirection should return 2, got %d\n", indirection(0))
}

编译失败，并显示消息 [][]t does not implement multidimensionslice ([][]t missing in int | float32 | []int | []float32 | [][]int | [][] float32)

但在函数 indirection() 中，t 的所有允许值都将在 dimensions() 中实现。

任何帮助或指示将不胜感激！

（演示链接）

正确答案

正如评论中提到的，go 在泛型方面有一些限制。您可以通过解决方法来实现您的要求。

首先，您需要更改您定义的接口。 （也使其通用）

type number interface {
    int | float32
}

type multidimensionslice[t number] interface {
    number | []t | [][]t
}

然后我们需要更改 dimension 方法类型参数。如果 go 让我们像这样定义 dimensions 方法，那就干净多了

func  dimensions[s number](s multidimensionslice[s]) int {

但我们能做的就是：

func dimensions[s number, k multidimensionslice[s]](s k) int {
    dims := 0
    t := reflect.typeof(s)
    for t.kind() == reflect.slice {
        dims += 1
        t = t.elem()
    }
    return dims
}

然后我们需要更改调用 dimensions 方法的方式。我们需要提供一个额外的类型参数，以便可以推断类型参数 s

func indirection[T Number](v T) int {
    var slice2D [][]T
    return dimensions[T](slice2D)
}

func main() {
    x := [][]float32{{1}, {2}, {3}}
    fmt.Printf("x=%v, dims=%d\n", x, dimensions[float32](x))
    fmt.Printf("indirection should return 2, got %d\n", indirection(0))
}

Playground

今天关于《泛型类型推断在函数级联调用中的应用》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！