Cookie与Session

1.Cookie

1.1 Cookie的使用

c.SetCookie("abc", "123", 60, "/","localhost", false, true)   // 设置cookie
c.Cookie("abc")    // 获取cookie

1.2 模拟登陆

• 第一步：访问home页失败，由于没有访问login页面，没有设置cookie
  • http://localhost:8000/home
• 第二步：然后访问login页面设置cookie
  • http://localhost:8000/login
• 第三步：访问完login页面，再访问home页面就可以成功了

package main

import (
	"github.com/gin-gonic/gin"
	"net/http"
)

func AuthMiddleWare() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 获取客户端cookie并校验
		if cookie, err := c.Cookie("abc"); err == nil {
			if cookie == "123" {
				c.Next()
				return
			}
		}
		// 返回错误
		c.JSON(http.StatusUnauthorized, gin.H{"error": "身份验证失败"})
		// 若验证不通过，不再调用后续的函数处理
		c.Abort()
		return
	}
}

func main() {
	// 1.创建路由
	r := gin.Default()
	// 模仿登录成功，设置cookie
	r.GET("/login", func(c *gin.Context) {
		c.SetCookie("abc", "123", 60, "/",
			"localhost", false, true)
		c.String(200, "Login success!")
	})
	// 使用局部中间件，验证cookie
	r.GET("/home", AuthMiddleWare(), func(c *gin.Context) {
		c.JSON(200, gin.H{"data": "home"})
	})
	r.Run(":8000")
}

02.Session

2.1 session使用语法

// 第一：初始化一个cookie存储对象
// something-very-secret应该是一个你自己的密匙，只要不被别人知道就行
var store = sessions.NewCookieStore([]byte("something-very-secret"))

// 第二步：保存更改
session.Values["foo"] = "bar"
session.Save(r, w)

// 第三步：获取session值
session.Values["foo"]

2.2 使用演示

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

// 第一：初始化一个cookie存储对象
// something-very-secret应该是一个你自己的密匙，只要不被别人知道就行
var store = sessions.NewCookieStore([]byte("something-very-secret"))

func main() {
	http.HandleFunc("/save", SaveSession)
	http.HandleFunc("/get", GetSession)
	err := http.ListenAndServe(":8080", nil)
	if err != nil {
		fmt.Println("HTTP server failed,err:", err)
		return
	}
}

func SaveSession(w http.ResponseWriter, r *http.Request) {
	//　获取一个session对象，session-name是session的名字
	session, err := store.Get(r, "session-name")
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	// 在session中存储值
	session.Values["foo"] = "bar"
	session.Values[42] = 43
	// 第二步：保存更改
	session.Save(r, w)
}

func GetSession(w http.ResponseWriter, r *http.Request) {
	session, err := store.Get(r, "session-name")
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	// 第三步：获取session值
	foo := session.Values["foo"]
	fmt.Println(foo)
}