根据map [string] somestruct

时间:2019-07-13 08:25:04

标签: go closures map-function

假设我有很多带有接收器的函数或方法,每个函数或方法都有不同类型的参数。我想使用表驱动的方法来调度函数或方法调用。所以我将构建一个这样的表:

type command struct {
   name string
   handler func(parameter ...interface{}) // I don't know whether to use `...interface{}` is correct
}

table := map[string]command { ... }

func (c command)foo(f1 int, f2 string) {}
func (c command)bar(b1 bool, b2 int, b3 string) {}
// methods and so on

无论我收到什么命令,我将始终使用table[some-command-name](parameter1, parameter2, ...)之类的一条语句,而不要使用难看的switch-case结构。

这可能吗?不同的方法有不同数量的参数是一个问题吗?

1 个答案:

答案 0 :(得分:1)

有两种可能的方法。一种是通过反射,即reflect包,尤其是Call方法。第二个是通过函数值和闭包。我不推荐第一种解决方案,通常不建议使用反射,因为它很复杂,容易出错且价格昂贵。

通过反射解决方案(https://play.golang.org/p/3b5I77QMsFI):

type command struct {
    name    string
    handler reflect.Value
    args    []reflect.Value
}

var table = map[string]command{
    "Foo": {
        name:    "Foo",
        handler: reflect.ValueOf(foo),
        args: []reflect.Value{
            reflect.ValueOf(1),
            reflect.ValueOf("hello"),
        },
    },
    "Bar": {
        name:    "Bar",
        handler: reflect.ValueOf(bar),
        args: []reflect.Value{
            reflect.ValueOf(true),
            reflect.ValueOf(5),
            reflect.ValueOf("hello"),
        },
    },
}

func foo(f1 int, f2 string) {
    fmt.Println("Foo:", f1, f2)
}
func bar(b1 bool, b2 int, b3 string) {
    fmt.Println("Bar:", b1, b2, b3)
}

func main() {
    for name, command := range table {
        fmt.Println("Running", name)
        command.handler.Call(command.args)
    }
}

通过函数闭包(https://play.golang.org/p/8fM86lxalq1)的解决方案:

type MyFuncType func()

type command struct {
    name    string
    handler MyFuncType
}

var table = map[string]command{
    "Foo": {
        name:    "Foo",
        handler: fooClosure(1, "hello"),
    },
    "Bar": {
        name:    "Bar",
        handler: barClosure(true, 5, "hello"),
    },
}

func foo(f1 int, f2 string) {
    fmt.Println("Foo:", f1, f2)
}

func fooClosure(f1 int, f2 string) MyFuncType {
    return func() {
        foo(f1, f2)
    }
}

func bar(b1 bool, b2 int, b3 string) {
    fmt.Println("Bar:", b1, b2, b3)
}

func barClosure(b1 bool, b2 int, b3 string) MyFuncType {
    return func() {
        bar(b1, b2, b3)
    }
}

func main() {
    for name, command := range table {
        fmt.Println("Running", name)
        command.handler()
    }
}