46 lines
946 B
Go
46 lines
946 B
Go
package middleware
|
||
|
||
import (
|
||
"log"
|
||
|
||
"github.com/gin-contrib/sessions"
|
||
"github.com/gin-contrib/sessions/cookie"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
var sessionID string
|
||
|
||
func CheckSession() gin.HandlerFunc {
|
||
return func(ctx *gin.Context) {
|
||
session := sessions.Default(ctx)
|
||
|
||
//ctx.Next()
|
||
//return
|
||
|
||
token := session.Get(sessionID)
|
||
log.Println("session:", sessionID, token)
|
||
// 未登录时重定向到登录页
|
||
if token == nil {
|
||
ctx.Redirect(301, "/sign-in")
|
||
return
|
||
}
|
||
|
||
// set simple var
|
||
ctx.Set("SESSION", token)
|
||
// validate session：1.check cache 2.check database // TODO
|
||
|
||
ctx.Next()
|
||
}
|
||
}
|
||
func EnableCookieSession(name, key string) gin.HandlerFunc {
|
||
store := cookie.NewStore([]byte(name))
|
||
store.Options(sessions.Options{
|
||
Path: "/",
|
||
MaxAge: 86400, // 24小时
|
||
HttpOnly: true,
|
||
Secure: false, // 在使用HTTPS的生产环境中设置为true
|
||
})
|
||
sessionID = name
|
||
return sessions.Sessions(key, store)
|
||
}
|