Go 中的 Cookie 和 Session

原创

浩瀚星河

修改于 2025-03-25 03:05:06

740

文章被收录于专栏：golanggolang

今天聊聊 Cookie 和 Session 的登录认证怎么搞，以及能用在哪。

1. 准备

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

sync.Map 存 Session，线程安全。
User 定义用户名和密码，必填。

2. 生成 Session ID

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

uuid 生成随机 ID，用户信息转 JSON 存起来。

3. 查 Session

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
}

用 Session ID 查数据，反序列化成 User。

4. 登录逻辑

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": "登录成功"})
}