如何修改Go语言transformNumbers函数以适配泛型函数doubleG与tripleG

问题背景 #

现有代码中，transformNumbers函数接受非泛型的transformFn（func(int) int）作为参数，能正常适配double、triple这类非泛型函数，但无法直接适配doubleG、tripleG这类泛型函数。尝试将参数改为泛型函数类型后执行报错，核心需求是让transformNumbers同时兼容非泛型和泛型转换函数。

原代码：

type transformFn func(int) int

func transformNumbers(numbers *[]int, transform transformFn) []int {
    dNumbers := []int{}
    for _, value := range *numbers {
        dNumbers = append(dNumbers, transform(value))
    }
    return dNumbers
}

func double(value int) int {
    return 2 * value
}

func triple(value int) int {
    return 3 * value
}

func doubleG[T int | float64 | float32](value T) T {
    return 2 * value
}

func tripleG[T int | float64 | float32](value T) T {
    return 3 * value
}

错误修改尝试：

func transformNumbers(numbers *[]int, transform func[T int|float64|float32](T)T) []int {
    dNumbers := []int{}
    for _, value := range *numbers {
        dNumbers = append(dNumbers, transform(value)) // 此处报错：无法推导T的类型
    }
    return dNumbers
}

错误原因 #

Go中的泛型函数本质是函数模板，并非可直接调用的具体函数。在传递或调用泛型函数时，必须先通过指定类型参数完成实例化，生成对应类型的具体函数。上述修改中，transform参数是泛型函数类型，但函数内部调用时Go无法自动推导T的具体类型，因此触发编译错误。

解决方案 #

有两种可行方案，按需选择：

方案1：让transformNumbers本身成为泛型函数 #

如果需要transformNumbers支持多种数值类型的切片，可将其改造为泛型函数，这样能自然兼容同类型的泛型/非泛型转换函数：

// 定义泛型版本的transformNumbers，支持int/float64/float32类型切片
func transformNumbers[T int | float64 | float32](numbers []T, transform func(T) T) []T {
    dNumbers := []T{}
    for _, value := range numbers {
        dNumbers = append(dNumbers, transform(value))
    }
    return dNumbers
}

调用示例：

func main() {
    nums := []int{1, 2, 3}
    
    // 适配非泛型函数
    doubled := transformNumbers(nums, double)
    tripled := transformNumbers(nums, triple)
    
    // 适配泛型函数（Go可自动推导类型，也可显式指定）
    doubledG := transformNumbers(nums, doubleG) // 自动推导T=int
    tripledG := transformNumbers(nums, tripleG[int]) // 显式指定类型
}

方案2：保留原transformNumbers，实例化泛型函数后传递 #

如果只需要transformNumbers处理[]int类型，无需修改原函数，只需在传递泛型函数时显式实例化其int版本即可：

func main() {
    nums := []int{1, 2, 3}
    
    // 调用非泛型函数（原逻辑不变）
    doubled := transformNumbers(&nums, double)
    
    // 调用泛型函数的int实例：doubleG[int]是具体的func(int)int类型，符合参数要求
    doubledG := transformNumbers(&nums, doubleG[int])
    tripledG := transformNumbers(&nums, tripleG[int])
}

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