Negroni：将上下文从中间件传递到处理程序

Question

我正在尝试将 Gorilla 会话添加到 Negroni 中间件处理程序中的 Request Context，以便我可以访问它我的大猩猩 Mux 处理程序。这是我的代码的精简版：

// Session Middleware function
func sessMid(w http.ResponseWriter, r *http.Request, next 
http.HandlerFunc) {
  ctx := r.Context()
  s, _ := store.Get(r, "user") // store is a CookieStore
  ctx = context.WithValue(ctx, "example", s)

  if !loggedIn() {
    http.Redirect(w, r, "/login", http.StatusFound)
  }

  next(w, r.WithContext(ctx))
}

// Page handler
func pgHandler(w http.ResponseWriter, r *http.Request) {
  ctx := r.Context()
  s, ok := ctx.Value("example").(*sessions.Session)
  // ok returns false here, meaning that the session was not returned successfully.
}

希望这是有道理的。谁能指出我做错了什么？

Answer 1

重定向语句正在接收没有包含会话的新上下文的原始请求。这里也需要用到WithContext(ctx)函数：

// Session Middleware function
func sessMid(w http.ResponseWriter, r *http.Request, next 
http.HandlerFunc) {
  ctx := r.Context()
  s, _ := store.Get(r, "user") // store is a CookieStore
  ctx = context.WithValue(ctx, "example", s)

  if !loggedIn() {
    // Make sure to add the context to the request sent in the Redirect
    http.Redirect(w, r.WithContext(ctx), "/login", http.StatusFound)
  }

  next(w, r.WithContext(ctx))
}

感谢 @jmaloney 让我走上了正确的道路。

Negroni：将上下文从中间件传递到处理程序

Negroni: passing context from middleware to handlers

go

negroni