HikariTish

1
func main() {
2
    r := gin.Default()
3

4
    r.GET("/cookie/set", func(c *gin.Context) {
5
        c.SetCookie(
6
            "session_id",           // name
7
            "abc123",               // value
8
            3600,                   // maxAge (秒)
9
            "/",                    // path
10
            "localhost",            // domain
11
            false,                  // secure
12
            true,                   // httpOnly
13
        )
14
        c.JSON(http.StatusOK, gin.H{"message": "Cookie set"})
15
    })
16

17
    r.GET("/cookie/get", func(c *gin.Context) {
18
        sessionID, err := c.Cookie("session_id")
19
        if err != nil {
20
            c.JSON(http.StatusNotFound, gin.H{
21
                "error": "Cookie not found",
22
            })
23
            return
24
        }
25
        c.JSON(http.StatusOK, gin.H{"session_id": sessionID})
26
    })
27

28
    r.GET("/cookie/delete", func(c *gin.Context) {
29
        c.SetCookie("session_id", "", -1, "/", "localhost", false, true)
30
        c.JSON(http.StatusOK, gin.H{"message": "Cookie deleted"})
31
    })
32

33
    r.Run(":8080")
34
}

1
import (
2
    "github.com/gin-contrib/sessions"
3
    "github.com/gin-contrib/sessions/cookie"
4
)
5

6
func main() {
7
    r := gin.Default()
8

9
    // 创建 cookie store
10
    store := cookie.NewStore([]byte("secret-key"))
11
    r.Use(sessions.Sessions("mysession", store))
12

13
    r.POST("/login", func(c *gin.Context) {
14
        session := sessions.Default(c)
15

16
        var loginForm struct {
17
            Username string `json:"username" binding:"required"`
18
            Password string `json:"password" binding:"required"`
19
        }
20

21
        if err := c.ShouldBindJSON(&loginForm); err != nil {
22
            c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
23
            return
24
        }
25

26
        // 验证用户（示例）
27
        if loginForm.Username == "admin" && loginForm.Password == "password" {
28
            session.Set("user_id", 1)
29
            session.Set("username", loginForm.Username)
30
            session.Save()
31

32
            c.JSON(http.StatusOK, gin.H{"message": "Login successful"})
33
        } else {
34
            c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid credentials"})
35
        }
36
    })
37

38
    r.GET("/profile", func(c *gin.Context) {
39
        session := sessions.Default(c)
40
        userID := session.Get("user_id")
41

42
        if userID == nil {
43
            c.JSON(http.StatusUnauthorized, gin.H{"error": "Not logged in"})
44
            return
45
        }
46

47
        username := session.Get("username")
48
        c.JSON(http.StatusOK, gin.H{
49
            "user_id":  userID,
50
            "username": username,
51
        })
52
    })
53

54
    r.POST("/logout", func(c *gin.Context) {
55
        session := sessions.Default(c)
56
        session.Clear()
57
        session.Save()
58
        c.JSON(http.StatusOK, gin.H{"message": "Logged out"})
59
    })
60

61
    r.Run(":8080")
62
}