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

哈喽！大家好，很高兴又见面了，我是golang学习网的一名作者，今天由我给大家带来一篇《Gorilla mux 中间件：包装处理程序》，本文主要会讲到等等知识点，希望大家一起学习进步，也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧！

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

解决方案

您似乎正在尝试做两件事。 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
}

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

今天关于《Gorilla mux 中间件：包装处理程序》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！