如何在chi路由器上下文中存储来自登录的会话数据

package main

import (
    "net/http"

    "github.com/go-chi/chi"
    "github.com/gorilla/sessions"
)

func main() {
    r := chi.NewRouter()

    // 初始化session存储
    store := sessions.NewCookieStore([]byte("secret-key"))

    r.Use(sessions.Middleware(store))

    r.Get("/login", func(w http.ResponseWriter, r *http.Request) {
        // 获取session
        session, _ := store.Get(r, "session-name")

        // 存储会话数据
        session.Values["username"] = "user123"
        session.Values["role"] = "admin"

        // 保存会话数据
        session.Save(r, w)

        w.Write([]byte("Logged in successfully"))
    })

    r.Get("/profile", func(w http.ResponseWriter, r *http.Request) {
        // 获取session
        session, _ := store.Get(r, "session-name")

        // 检查会话数据
        if username, ok := session.Values["username"].(string); ok {
            w.Write([]byte("Username: " + username))
        } else {
            w.Write([]byte("Not logged in"))
        }
    })

    http.ListenAndServe(":8080", r)
}