Negroni: передача контекста от промежуточного программного обеспечения обработчикам

Я пытаюсь добавить сеанс Gorilla в контекст запроса в обработчике промежуточного программного обеспечения Negroni, чтобы я мог получить к нему доступ в своих обработчиках Gorilla 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.
}

Надеюсь, это имеет смысл. Может ли кто-нибудь указать мне на то, что я делаю не так?


go negroni
person 17xande    schedule 26.01.2017    source источник
comment
Вы проверили, возвращает ли store.Get *sessions.Session?   -  person jmaloney    schedule 26.01.2017
comment
Спасибо @jmaloney, ваш комментарий помог мне найти ответ, я застрял.   -  person 17xande    schedule 27.01.2017
comment
Рад, что смог помочь   -  person jmaloney    schedule 27.01.2017

Ответы (1)


arrow_upward
3
arrow_downward

Оператор Redirect получает исходный запрос без нового контекста, содержащего сеанс. Здесь также необходимо использовать функцию 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 за то, что направили меня по правильному пути.

person 17xande    schedule 27.01.2017
comment
Работает, но для WithValue не рекомендуется использовать строковые ключи. Согласно документации: https://godoc.org/net/http#Request.WithContext - person Jason Stewart; 16.08.2019
comment
Предоставленный ключ должен быть сопоставимым и не должен быть строкового типа или любого другого встроенного типа, чтобы избежать конфликтов между пакетами, использующими контекст. Пользователи WithValue должны определять свои собственные типы ключей. Чтобы избежать выделения при назначении интерфейсу {}, ключи контекста часто имеют конкретный тип struct {}. В качестве альтернативы статическим типом экспортируемых переменных ключа контекста должен быть указатель или интерфейс.) - person Jason Stewart; 16.08.2019