mirror of
https://gitee.com/blackfox/geekai.git
synced 2024-12-05 21:58:23 +08:00
31 lines
558 B
Go
31 lines
558 B
Go
package utils
|
|
|
|
import (
|
|
"chatplus/core/types"
|
|
"chatplus/store/model"
|
|
"errors"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func GetLoginUser(c *gin.Context, db *gorm.DB) (model.User, error) {
|
|
value, exists := c.Get(types.LoginUserCache)
|
|
if exists {
|
|
return value.(model.User), nil
|
|
}
|
|
|
|
userId, ok := c.Get(types.LoginUserID)
|
|
if !ok {
|
|
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
|
|
}
|