Go泛型函数级联调用时的类型推断问题

Go的泛型类型约束采用精确静态匹配机制，编译器仅检查当前类型是否属于约束显式声明的类型集合。你的MultiDimensionSlice枚举了int、float32、[]int等具体类型，但[][]T是泛型实例化前的抽象类型，不在枚举列表中——哪怕逻辑上T只能是int或float32，编译器不会做“实例化后必然匹配”的推导，因此抛出类型不匹配错误。

有两种核心方向可以解决这个问题：

• 重构约束为递归泛型：让约束支持任意维度的切片，而非枚举具体类型，适配泛型实例化后的类型，保留编译时类型检查。
• 放宽约束至any：由于dimensions函数本身依赖反射判断维度，直接接受任意类型即可绕开静态类型检查，同时保留原有功能。

方案1：递归泛型约束 #

package main

import (
    "fmt"
    "reflect"
)

type Number interface {
    int | float32
}

// 递归定义支持任意维度的切片约束
type MultiDimensionSlice[N Number] interface {
    N | []MultiDimensionSlice[N]
}

func dimensions[N Number, S MultiDimensionSlice[N]](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[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))
}

方案2：放宽约束至any（更简洁） #

package main

import (
    "fmt"
    "reflect"
)

type Number interface {
    int | float32
}

// 直接接受任意类型，内部通过反射处理维度判断
func dimensions(s any) int {
    dims := 0
    t := reflect.TypeOf(s)
    for t != nil && 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))
}

内容的提问来源于stack exchange，提问作者Jan