3

对我的应用程序的每个请求都应该使用一些中间件。使用 Negroni 文档，我已经实现了它，如下所示：

func MyMiddleware(rw http.ResponseWriter, r *http.Request, next http.HandlerFunc) {
  // do some stuff before
  next(rw, r)
}

我在其他地方使用：

n.Use(negroni.HandlerFunc(MyMiddleware))

但是，中间件需要接收一个额外的参数，我不确定如何构建它。目前我只是将值设置为全局变量以使其可用，但我确信有一种更优雅的方式?

我希望能够做这样的事情：

n.Use(negroni.HandlerFunc(MyMiddleware(val)))
4

1 回答 1

5

最好的方法是将您的中间件封装为一个保存其状态的结构，而不仅仅是一个无状态函数。（您也可以将其包装为闭包，但结构更清洁 IMO）：

type MyMiddleware struct {
    someval string
}

func NewMyMiddleware(someval string) *MyMiddleware {
    return &MyMiddleware{
       someval: someval,
    }
}


func (m *MyMiddleware) ServeHTTP(w http.ResponseWriter, r *http.Request, next http.HandlerFunc) {

    // Do Something with someval
    fmt.Println(m.someval)

    next(w, req)
}

初始化它很简单：

n.Use(NewMyMiddleware("foo"))

编辑：也许关闭实际上很简单：

 someval := foo

 n.Use(negroni.HandlerFunc(func(rw http.ResponseWriter, r *http.Request, next http.HandlerFunc) {
   // Do Something with someval
   fmt.Println(someval)

   next(w, req)
}))

或者你可以有一个返回中间件函数的函数：

func NewMiddleware(someval string) negroni.HandlerFunc {
     return negroni.HandlerFunc(func(rw http.ResponseWriter, r *http.Request, next http.HandlerFunc) {
       // Do Something with someval
       fmt.Println(someval)

       next(w, req)
    })
}

接着

n.Use(NewMiddleware("foo"))
于 2015-08-26T20:04:44.310 回答