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

【问题标题】：gorilla mux middleware: wrapping a handlergorilla mux 中间件：包装处理程序
【发布时间】：2020-08-26 09:31:06
【问题描述】：

我需要将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 变量。所以，我也需要这个闭包。

【问题讨论】：

当你说访问和存储“端点”是什么意思？如果您只是指请求路径，您可以使用 r.RequestURI 在中间件和处理程序中获取它。
只是问一下，如果这仅适用于一个端点，为什么要将其实现为中间件？
这不是唯一一个。还有其他端点。但除了这些（我想使用中间件的那些）之外，还有一些我不想使用的端点。所以如果有m端点，那么其中n使用这个中间件，其余m-n不要。
它变成了中间件预期功能的功能。如果它高度依赖于您的主处理程序正在做什么，那么将代码添加到主处理程序中可能比尝试构建中间件更有意义。中间件通常用于日志记录、身份验证、指标捕获等功能。这里它在一定程度上独立于主处理程序。
同意 praveent。我认为您有一些选择，具体取决于中间件/代码的确切功能。让我试着和他们中的一些人一起回答。

标签： go gorilla mux

【解决方案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
    })
}

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

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

如果您确实使用中间件，您可以使用 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
}

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

【讨论】：