如何用地图替换多个“if-else-if”

最近发现不少小伙伴都对Golang很感兴趣，所以今天继续给大家介绍Golang相关的知识，本文《如何用地图替换多个“if-else-if”》主要内容涉及到等等知识点，希望能帮到你！当然如果阅读本文时存在不同想法，可以在评论中表达，但是请勿使用过激的措辞~

我对多层 if-else-if 条件感到担忧，因此我想通过使用地图来简短说明。

请参阅 if-else-if 中的以下代码，我想用地图替换它。

function, args := APIstub.GetFunctionAndParameters()

if function == "queryProduce" {
    return s.queryProduce(APIstub, args)
} else if function == "initLedger" {
    return s.initLedger(APIstub)
} else if function == "createProduce" {
    return s.createProduce(APIstub, args)
} else if function == "queryAllProduces" {
    return s.queryAllProduces(APIstub)
} else if function == "changeProduceStatus" {
    return s.changeProduceStatus(APIstub, args)
}

return shim.Error("Invalid Smart Contract function name.")
}

解决方案

对于你有一个 switch 会很好：

switch function {
case "queryproduce":
    return s.queryproduce(apistub, args)
case "initledger":
    return s.initledger(apistub)
case "createproduce":
    return s.createproduce(apistub, args)
case "queryallproduces":
    return s.queryallproduces(apistub)
case "changeproducestatus":
    return s.changeproducestatus(apistub, args)
}

使用映射会很不方便，因为并非所有方法都具有相同的签名，但您可以使用多个映射。

另一种解决方案可能是使用反射来调用方法，但同样，处理不同的参数会很不方便。反射也比较慢，更不用说您必须注意不允许调用不打算公开的方法。例如，请参阅 Call functions with special prefix/suffix。

可以将您所拥有的内容表达为地图。这里的基本设置是，无论你走哪条路，你都会得到一些可以不带参数调用的函数，并且它总是返回相同的类型（error）。我可能会明确传递 args 。

其高级结构是拥有函数名称到函数的映射，然后调用所选函数。

funcmap := map[string]func([]string) error{...}
funcname, args := apistub.getfunctionandparameters()
f := funcmap[funcname]
if f == nil {
    f = func(_ []string) error {
        return shim.error("invalid smart contract function name.")
    }
}
return f(args)

地图语法变得有点冗长

funcMap := map[string]func([]string) error{
    "queryProduce": func(args []string) error {
        return s.queryProduce(APIstub, args)
    },
    "initLedger": func(_ []string) error {
        return s.initLedger(APIstub)
    },
}

如果您要在多个地方调用它，或者您想要一个单独的验证步骤，如果使用的话将定义一些名称，或者如果函数的实际列表是动态的（您可以添加或在运行时从地图中删除内容）。不一致的方法签名确实会带来复杂性，并且使所有内容保持一致会有所帮助（使 initledger 等函数采用参数列表，即使它未使用）。

在普通代码中，我希望 @icza 的答案中的 switch 形式更加惯用。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《如何用地图替换多个“if-else-if”》文章吧，也可关注golang学习网公众号了解相关技术文章。