gorilla mux 中间件：包装处理程序

Question

我需要将 GetAssetsCompute 函数包装在中间件中

r.Handle("/api/v1/assets/ComputeBlade", GetAssetsCompute(assetService)).Methods("GET")

func GetAssetsCompute(assetService ServiceType) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        //  stuff here
    })
}

但是因为中间件将 HTTP 处理程序作为参数，而我的函数不是处理程序，所以我不能。

我正在考虑做这样的事情。

func GetAssetsCompute(assetService ServiceType) http.Handler {
        return MyMiddleware(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            //  stuff here
        }))
    }


func MyMiddleware(next http.Handler) http.Handler {

}

这是正确的吗？或者有更好的方法来做到这一点。同样在中间件内部，我需要访问 URL 端点，进行一些处理并存储这个处理后的值，然后再次在主处理程序中访问它。我该怎么做？

编辑：我只想将此中间件应用于我拥有的端点的一个子集（> 1）。不是全部

我还需要处理程序中 GetAssetsCompute(assetService ServiceType) 函数中使用的 assetService 变量。所以，我也需要这个闭包。

Answer 1

您似乎正在尝试做两件事。 1 - 仅将中间件应用于您的某些请求处理程序。 2 - 将数据从中间件传递到请求处理程序。

对于第一个，我可以想到三个选项。第一个是你现在正在做的，有一个中间件函数，当你将它们传递给 r.Handle 时，你将它们包装在其中。伪代码：

r.Handle("/path1", Mware(Handler1())).Methods("GET")
r.Handle("/path2", Mware(Handler2())).Methods("GET")
r.Handle("/path3-nomiddleware", Handler3()).Methods("GET")

您可以做的第二件事是将代码添加到您的中间件以根据 URI 路径进行过滤，然后使用 r.Use 注册您的中间件。伪代码：

const mwarePaths []string = ...
func Mware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
       if r.RequestURI is in mwarePaths {
          // do the middleware
       }
    }
}

r.Use(Mware)

第三，您可以将代码放在您在处理程序中直接调用的函数中，而不是像中间件那样注册它。伪代码：

func myUtil(w http.ResponseWriter, r *http.Request){ ... }

func GetAssetsCompute(assetService ServiceType) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
       myUtil(w, r)

       //  stuff here
    })
}

对于第二件事——将数据从中间件传递到请求处理程序——这里有一些想法。

首先，如果您使用上面的常规函数、无中间件设置，这个问题就会消失，因为您在处理程序中需要的任何东西都可以只是函数中的 return 值。

如果你确实使用中间件，你可以使用 context 库（也来自 gorilla）将变量绑定到 http.Request 实例以传递给你的处理程序：http://www.gorillatoolkit.org/pkg/context 。使用它看起来像这样：

import "github.com/gorilla/context"

func middleware(...) {
   return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
       context.Set(r, "myKey", "bar")
    }
}

func handler(w http.ResponseWriter, r *http.Request) {
   val, ok := context.GetOk(r, "myKey") // returns "bar", true
}

您选择使用哪些选项取决于您（您知道自己的需求）。但是，正如评论中所提到的，一个好的经验法则是处理与您的请求处理程序所做的无关的问题的代码可以是中间件。处理与您的请求处理程序正在做的事情直接相关的问题的代码可以直接放在处理程序中。

gorilla mux 中间件：包装处理程序

gorilla mux middleware: wrapping a handler

go

gorilla

mux