mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-17 16:56:38 +08:00
46 lines
1.0 KiB
Go
46 lines
1.0 KiB
Go
package utils
|
||
|
||
import (
|
||
"chatplus/core/types"
|
||
"chatplus/store/model"
|
||
"errors"
|
||
|
||
"github.com/gin-contrib/sessions"
|
||
"github.com/gin-gonic/gin"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
func SetLoginUser(c *gin.Context, user model.User) error {
|
||
session := sessions.Default(c)
|
||
session.Set(types.SessionUser, user.Id)
|
||
// TODO: 后期用户数量增加，考虑将用户数据存储到 leveldb，避免每次查询数据库
|
||
return session.Save()
|
||
}
|
||
|
||
func SetLoginAdmin(c *gin.Context, admin types.Manager) error {
|
||
session := sessions.Default(c)
|
||
session.Set(types.SessionAdmin, admin)
|
||
return session.Save()
|
||
}
|
||
|
||
func GetLoginUser(c *gin.Context, db *gorm.DB) (model.User, error) {
|
||
value, exists := c.Get(types.LoginUserCache)
|
||
if exists {
|
||
return value.(model.User), nil
|
||
}
|
||
|
||
session := sessions.Default(c)
|
||
userId := session.Get(types.SessionUser)
|
||
if userId == nil {
|
||
return model.User{}, errors.New("user not login")
|
||
}
|
||
|
||
var user model.User
|
||
res := db.First(&user, userId)
|
||
// 更新缓存
|
||
if res.Error == nil {
|
||
c.Set(types.LoginUserCache, user)
|
||
}
|
||
return user, res.Error
|
||
}
|