90 lines
1.7 KiB
Plaintext
90 lines
1.7 KiB
Plaintext
package user
|
|
|
|
import (
|
|
"common/cache"
|
|
"common/connection"
|
|
"goutil/logUtilPlus"
|
|
"sync"
|
|
)
|
|
|
|
// 用户缓存对象
|
|
var userMap = make(map[int64]*User)
|
|
var rwmu sync.RWMutex
|
|
|
|
// GetUserByID 根据用户id获取用户信息
|
|
func GetUserByID(UserID int64) (*User, error) {
|
|
|
|
//判断缓存是否存在
|
|
var user *User
|
|
|
|
func() *User {
|
|
rwmu.RLock()
|
|
defer rwmu.RUnlock()
|
|
ok := true
|
|
if user, ok = userMap[UserID]; ok {
|
|
return user
|
|
}
|
|
return nil
|
|
}()
|
|
|
|
if user != nil {
|
|
return user, nil
|
|
}
|
|
|
|
result := connection.GetUserDB().First(&user, UserID)
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
|
|
//添加缓存
|
|
func() {
|
|
rwmu.Lock()
|
|
defer rwmu.Unlock()
|
|
user.Cache = cache.NewCache()
|
|
userMap[user.ID] = user
|
|
}()
|
|
|
|
return user, nil
|
|
}
|
|
|
|
// AddUserCache 添加用户缓存
|
|
func AddUserCache(user *User) {
|
|
rwmu.Lock()
|
|
defer rwmu.Unlock()
|
|
user.Cache = cache.NewCache()
|
|
userMap[user.ID] = user
|
|
}
|
|
|
|
// AddUser 添加用户
|
|
// AddUser 添加新的用户到数据库中。
|
|
// 参数 User: 包含用户信息的对象。
|
|
// 返回值: 插入操作影响的行数和可能发生的错误。
|
|
func AddUser(User *User) (int64, error) {
|
|
|
|
//处理一些验证
|
|
|
|
//写入缓存
|
|
AddUserCache(User)
|
|
|
|
// 写入到数据库
|
|
result := connection.GetUserDB().Create(&User) // 通过数据的指针来创建
|
|
|
|
if result.Error != nil {
|
|
logUtilPlus.ErrorLog("添加用户失败 错误信息：", result.Error.Error())
|
|
}
|
|
return User.ID, nil
|
|
}
|
|
|
|
// Login 用户登录
|
|
func Login(account string, password string) (*User, error) {
|
|
|
|
//这里优先验证缓存数据
|
|
|
|
var User User
|
|
result := connection.GetUserDB().Where("account = ? AND password = ?", account, password).First(&User)
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
return &User, nil
|
|
}
|