调用在golang中作为接口变量接收的函数

Question

我有一个类似于以下的代码

package main

import "fmt"

func PrintThis(arg string) {
    fmt.Printf("I'm printing %s", arg)
}

func PrintThisAndThat(arg1, arg2 string) {
    fmt.Printf("Now printing %s and %s", arg1, arg2)
}

func Invoke(fn interface{}, args ...string) {
    //fn(args...)
}

func main() {
    Invoke(PrintThis, "foo")
    Invoke(PrintThisAndThat, "foo", "bar")
}

这不是实际的生产代码,但这是一个简化版本.

问题: - 如果我取消注释该行,//fn(args...)我会收到编译错误prog.go:14: cannot call non-function fn (type interface {})

如何执行Invoke()函数作为参数接收的函数？

实现这一目标的正确方法是什么？

Answer 1

您可以使用Call或的CallSlice方法reflect.Value将其称为函数.与所有reflect.Value方法一样,这种恐慌是fn错误的类型.

func Invoke(fn interface{}, args ...string) {
    v := reflect.ValueOf(fn)
    rargs := make([]reflect.Value, len(args))
    for i, a := range args {
        rargs[i] = reflect.ValueOf(a)
    }
    v.Call(rargs)
}

http://play.golang.org/p/xGmNLDcLL_

Answer 2

您可以使用类似http://play.golang.org/p/opotbIGdrA的类型开关

package main

import "fmt"

func PrintThis(arg string) {
    fmt.Printf("I'm printing %s", arg)
}

func PrintThisAndThat(arg1, arg2 string) {
    fmt.Printf("Now printing %s and %s", arg1, arg2)
}

func Invoke(fn interface{}, args ...string) {
    switch m := fn.(type) {
    case func(string):
        m(args[0])
    case func(string, string):
        m(args[0], args[1])
    default:

    }
}

func main() {
    Invoke(PrintThis, "foo")
    Invoke(PrintThisAndThat, "foo", "bar")
}

但是您有点需要知道要传递什么功能才能使其正常工作。

顺便说一句，您可以PrintThis使用...string而不是多个字符串参数来变成可变参数函数。