go - Golang中每个函数调用的Pre和Post函数

Question

我想对每个功能执行一些常见操作。在函数开始之前和函数结束之前。

func Command1Processor() error {
   preCheck()
   // Actual command1 logic
   postCheck()
}

func Command2Processor() error {
   preCheck()
   // Actual command2 logic
   postCheck()
}

GoLang中有什么支持此功能的，以便我只编写逻辑，并且自动调用pre/post函数。

Answer 1

您可以使用中间件模式来实现此目的，该中间件模式在HTTP处理程序中非常流行。

您需要一个通用的中间件功能，例如

func withCheck(http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        preCheck()
        h.ServeHTTP(w, r)
        postCheck()
    })
}

然后，您可以通过将处理程序另外包装到 withCheck中间件中来连接处理程序:

myHandler := newMyHandler()
http.Handle("/foo", withCheck(myHandler))

这样，将在每个处理程序调用之前和之后调用 preCheck和 postCheck。

对于不带参数的命令处理器，示例如下所示:

type CommandProcessor func()

func withCheck(cp CommandProcessor) CommandProcessor {
    return CommandProcessor(func() {
        preCheck()
        cp()
        postCheck()
    })
}

func CommandProcessor1() {
    // Actual logic
}

func main() {
    cp1 := withChecker(CommandProcessor1)
}

这种方法的优点是您可以拥有多个中间件。这种模式很容易适应任何数量的参数和/或函数结果。