使用 Go 作为参数传递任意函数

有志者，事竟成！如果你在学习Golang，那么本文《使用 Go 作为参数传递任意函数》，就很适合你！文章讲解的知识点主要包括，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我正在尝试扩展我对 go 函数指针的了解，并且我有一个问题：在 go 中将函数作为参数传递可以做什么，不可以做什么。

假设我想编写一个可以包装任何现有函数的 decorator() 函数。为简单起见，我们将其限制为仅接受一个参数并仅返回一个值的函数。

如果我编写一个接受 func(interface{}) interface{} 作为参数的装饰器，只要我传入的函数也接受/返回 interface{} 类型，它就会隐式工作（请参阅 funca） .

我的问题是 - 有没有办法将 func(string) string 类型的现有函数转换为 func(interface{}) interface{} 类型，以便它也可以传递到装饰器函数中，而无需只是将其包装在一个新的匿名函数中（请参阅 funcb）？

package main

import (
    "fmt"
)

func decorate(inner func(interface{}) interface{}, args interface{}) interface {} {
    fmt.Println("Before inner")
    result := inner(args)
    fmt.Println("After inner")
    return result
}

func funcA(arg interface{}) interface{} {
    fmt.Print("Inside A, with arg: ")
    fmt.Println(arg)
    return "This is A's return value"
}

func funcB(arg string) string {
    fmt.Print("Inside B, with arg: ")
    fmt.Println(arg)
    return "This is B's return value"
}

func main() {
    
    // This one works. Output is:
    //
    //   Before inner
    //   Inside A, with arg: (This is A's argument)
    //   After inner
    //   This is A's return value
    //
    fmt.Println(decorate(funcA, "(This is A's argument)"))
    
    // This doesn't work. But can it?
    //fmt.Println(decorate(funcB, "(This is B's argument)"))
}

正确答案

这是不可能的。原因之一是传递参数的机制因函数而异，并且使用 interface{} arg 并不意味着“接受任何内容”。例如，采用结构体作为参数的函数将接收该结构体的每个成员，但采用包含该结构体的 interface{} 的函数将接收两个单词，一个包含该结构体的类型，另一个包含指向该结构体的指针。它。

因此，在不使用泛型的情况下，实现此目的的唯一方法是使用适配器函数。

使用 reflect package 处理具有任意参数和结果类型的函数。

func decorate(inner interface{}, args interface{}) interface{} {
    fmt.Println("Before inner")
    result := reflect.ValueOf(inner).Call([]reflect.Value{reflect.ValueOf(args)})
    fmt.Println("After inner")
    return result[0].Interface()
}

Run the code on the playground。

与问题中的 decorate 函数一样，此答案中的函数假定一个参数和一个结果。必须修改该函数才能处理其他函数类型。

op应该考虑问题中提出的匿名包装函数和此处使用reflect包之间的权衡。通过反射 api 调用函数比通过匿名包装器调用函数慢。反射 api 也会失去类型安全性。匿名包装函数增加了冗长性。

好了，本文到此结束，带大家了解了《使用 Go 作为参数传递任意函数》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！