在Go语言Gorilla框架中高效管理会话变量

package main

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

// 声明认证密钥和加密密钥
// 强烈建议在生产环境中使用更长、更随机的密钥，并从环境变量或配置文件中加载
var authKey = []byte("super-secret-authentication-key-that-is-at-least-32-bytes-long")
var encKey = []byte("super-secret-encryption-key-that-is-at-least-32-bytes-long")

// 初始化一个全局的CookieStore实例
var store = sessions.NewCookieStore(authKey, encKey)

func init() {
    // 可以在这里配置store的全局选项，例如默认的会话名称
    // store.Options = &sessions.Options{
    //  Path:     "/",
    //  MaxAge:   86400 * 7, // 7天
    //  HttpOnly: true,
    //  Secure:   true, // 生产环境强烈建议设置为true
    // }
}

func main() {
    http.HandleFunc("/", HomeHandler)
    http.ListenAndServe(":8080", nil)
}

package main

import (
    "fmt"
    "github.com/gorilla/sessions"
    "net/http"
)

// store 变量已在上面定义和初始化

func HomeHandler(w http.ResponseWriter, r *http.Request) {
    // 1. 获取会话
    // "my-session-name" 是会话的名称，浏览器会用这个名称作为cookie的键
    session, err := store.Get(r, "my-session-name")
    if err != nil {
        // 实际应用中应进行更详细的错误处理，例如记录日志或返回错误页面
        http.Error(w, "Failed to get session: "+err.Error(), http.StatusInternalServerError)
        return
    }

    // 2. 配置会话选项 (仅当会话是新创建时)
    // session.IsNew 属性指示会话是否是本次请求中新创建的
    if session.IsNew {
        // 设置Cookie的Domain，Path，MaxAge等选项
        // MaxAge=0 意味着会话Cookie是浏览器会话Cookie，关闭浏览器即失效
        // MaxAge>0 意味着持久化Cookie
        session.Options.Domain = "" // 默认为当前请求的域名，或指定特定域名
        session.Options.Path = "/"
        session.Options.MaxAge = 3600 // 会话有效期1小时
        session.Options.HttpOnly = true // 防止客户端脚本访问Cookie
        session.Options.Secure = true   // 仅在HTTPS连接下发送Cookie (生产环境强烈建议开启)
        fmt.Println("New session created.")
    }

    // 3. 读取和设置会话值
    // 会话值存储在 session.Values 映射中
    // 读取会话值
    if username, ok := session.Values["username"]; ok {
        fmt.Fprintf(w, "Welcome back, %s! ", username)
    } else {
        fmt.Fprint(w, "Hello, guest! ")
    }

    // 设置或更新会话值
    session.Values["username"] = "GorillaUser"
    session.Values["login_time"] = "2023-10-27 10:00:00"

    // 4. 保存会话
    // 这一步至关重要，它将修改后的会话数据写入HTTP响应头，发送给客户端
    if err := session.Save(r, w); err != nil {
        http.Error(w, "Failed to save session: "+err.Error(), http.StatusInternalServerError)
        return
    }

    fmt.Fprint(w, "Session updated. Refresh to see changes.")
}