Gorilla session.AddFlash 不添加 Flash 消息

Question

我有一个注册页面，其中包含两个处理程序，一个用于显示表单，一个用于处理表单提交。

我正在尝试使用 session.AddFlash 方法保存错误，然后执行 302 重定向回注册表并显示错误。

我设置了一个会话存储：

package web

import (
    "github.com/gorilla/sessions"
)

var sessionStore = sessions.NewCookieStore([]byte(sessionSecret))

然后我的处理程序看起来像这样：

package web

import (
    "html/template"
    "log"
    "net/http"
)

func registerForm(w http.ResponseWriter, r *http.Request) {
    session, err := sessionStore.Get(r, "mysession")
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    data := map[string]interface{}{}
    log.Print("Flashes: ")
    log.Print(session.Flashes())
    if flashes := session.Flashes(); len(flashes) > 0 {
        data["error"] = flashes[0]
    }

    tmpl, _ := template.ParseFiles("web/templates/register.html.tmpl")
    tmpl.Execute(w, data)
}

func register(w http.ResponseWriter, r *http.Request) {
    session, err := sessionStore.Get(r, "mysession")
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    r.ParseForm()
    username := r.Form["username"][0]
    password := r.Form["password"][0]

    if UserExists(username) {
        log.Print("Username already taken")
        session.AddFlash("Username already taken")
        http.Redirect(w, r, "/web/register", http.StatusFound)
        return
    }

    _, err = CreateUser(username, password)
    log.Print(err)

    if err != nil {
        session.AddFlash(err.Error())
        http.Redirect(w, r, "/web/register", http.StatusFound)
        return
    }

    http.Redirect(w, r, "/web/login", http.StatusFound)
}

通过添加日志，我可以看到 UserExists 返回 true，因此应该添加一条 flash 消息，但是在重定向到表单处理程序后，会话中没有保存 flash 消息。

Answer 1

我认为您必须在重定向之前保存会话。

session.Save(r, w)

http://www.gorillatoolkit.org/pkg/sessions#Session.Save