Go 中的 Cookie 和 Session

var sessionStore sync.Map
type User struct {
    Name     string `json:"name" binding:"required"`
    Password string `json:"password" binding:"required"`
}

func generateSessionID(user User) (string, error) {
    sessionID := uuid.NewString()
    userData, _ := json.Marshal(user)
    sessionStore.Store(sessionID, userData)
    return sessionID, nil
}

func getUserFromSession(sessionID string) (*User, bool) {
    if data, ok := sessionStore.Load(sessionID); ok {
        var user User
        json.Unmarshal(data.([]byte), &user)
        return &user, true
    }
    return nil, false
}

func loginHandler(ctx *gin.Context) {
    var user User
    if ctx.ShouldBindJSON(&user) != nil {
        ctx.JSON(400, gin.H{"error": "bad request"})
        return
    }
    if sessionID, _ := ctx.Cookie("auth"); sessionID != "" {
        if cachedUser, found := getUserFromSession(sessionID); found {
            ctx.JSON(200, cachedUser)
            return
        }
    }
    sessionID, _ := generateSessionID(user)
    ctx.SetCookie("auth", sessionID, 3600, "/", "localhost", false, false)
    ctx.JSON(200, gin.H{"message": "登录成功"})
}