json - Map 中的 Golang 函数调用

Question

我是 GO Programming 的新手。这是场景:-

存在一个如下所示的 JSON 文件:-

{
        "template": "linuxbase1",
        "checkname": ["check_disk"],
        "checkmethod": ["check_disk"]
}

我正在将这些数据解码为一个结构:-

package func1

import (
        "io/ioutil"
        "os"
        "encoding/json"
        "fmt"
)

type pluginfunc func() string
type Plugindata struct {
        Template string `json:"template"`
        Checkname []string `json:"checkname"`
        Checkmethod []pluginfunc `json:"checkmethod"`
}

var (
        Templatepath = "json_sample1.json"
        Templateitems Plugindata
)

func Gettemplatedata() {
        tdata, err := ioutil.ReadFile(Templatepath)
        if err != nil {
                fmt.Printf("Unable to read file %s. Error - %v\n",Templatepath, err.Error())
                os.Exit(3)
        }
        json.Unmarshal(tdata, &Templateitems)
}

“check_disk”函数在这里:-

package func1

func check_disk() string {
        return "Called check_disk"
}

这是带有 main() 的程序:-

package main

import (
        "fmt"
        "checksexpt/func1"
)

func main() {
        func1.Gettemplatedata()
        fmt.Printf("Templateitems in Main() => %v\n",func1.Templateitems)
        for index,funcname := range func1.Templateitems.Checkmethod {
                fmt.Printf("%d = %s\n",index,funcname())
        }

}

正如预期的那样，当我运行 main();我看到了错误:-

Templateitems in Main() => {linuxbase1 [check_cpu check_disk] [<nil> <nil>]}
panic: runtime error: invalid memory address or nil pointer dereference
[signal 0xb code=0x1 addr=0x0 pc=0x40115e]

goroutine 1 [running]:
panic(0x50e980, 0xc82000a100)
        /opt/go/src/runtime/panic.go:481 +0x3e6

因此，我试图从 JSON 文件中获取一个字符串并将其视为函数调用。那显然失败了!但是，这里的主要限制是必须从 JSON 文件中选择函数名称。我怎样才能做到这一点 ？我知道我可以按如下方式创建静态 map :-

type checkfunc func() string
var (
        Templateitems = map[string]map[string]checkfunc {
                "linuxbase1": {
                        "check_disk": check_disk,
                },
        }
)

因此，像 Templateitems["linuxbase1"]["check_disk"]() 这样的调用就可以正常工作。但是，我不想创建任何此类静态 map ，因为该 map 中的元素需要不断增长。对此有什么想法吗？

Answer 1

没有直接的方法可以直接从 JSON 值解析函数。此外，您不能使用字符串值来引用变量。因此字符串 check_cpu 将无法直接引用具有相同名称的函数。

您可以做的是按原样解析 json 字符串，并为函数创建一个全局映射。这样，您就可以像这样调用您的函数:

var funcMap = map[string]pluginfunc{
    "check_disk": check_disk,
    "check_cpu": check_cpu
}

在你的 main 循环中:

for index, funcname := range func1.Templateitems.Checkmethod {
        fmt.Printf("%d = %s\n", index, funcMap[funcname]())
}

但是，如果您确实需要将值放入您的结构中，您可以尝试从 json.Unmarshaler 接口(interface)实现 UnmarshalJSON。一个简单的例子是:

type pf map[string]pluginfunc

type Plugindata struct {
        Template string `json:"template"`
        Checkname []string `json:"checkname"`
        Checkmethod pf `json:"checkmethod"`
}

func (p *pf) UnmarshalJSON(data []byte) error {
    d := []string{}
    if err := json.Unmarshal(data, &d); err != nil {
        return err
    }
    *p = make(pf)
    for _, s := range d {
        (*p)[s] = funcMap[s]
    }
    return nil
}

var funcMap = pf{
    "check_disk": check_disk,
    "check_cpu": check_cpu
}

func main() {
    json.Unmarshal(tdata, &Templateitems)
    for k, f := range Templateitems.Checkmethod {
        fmt.Printf("%s -- %s\n", k, f())
    }
}

Working code

请注意，这种方式不像第一种方法那样可读性或简单，它仍然依赖于函数映射。

您可以阅读有关 json.Unmarshaler 的更多信息 here .