115 lines
3.2 KiB
Go
115 lines
3.2 KiB
Go
package operation
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"fmt"
|
||
"game/common/model"
|
||
"game/common/model/user"
|
||
"game/common/proto/pb"
|
||
"game/common/utils"
|
||
"github.com/fox/fox/log"
|
||
"github.com/go-redis/redis/v8"
|
||
"gorm.io/gorm"
|
||
"strconv"
|
||
)
|
||
|
||
type UserAccountOp struct {
|
||
db *gorm.DB
|
||
accountRedis *redis.Client
|
||
accountOp *model.TableOp[user.UserAccount]
|
||
}
|
||
|
||
func NewUserAccountOp() *UserAccountOp {
|
||
return &UserAccountOp{
|
||
db: UserDB,
|
||
accountRedis: AccountRedis,
|
||
accountOp: model.NewTableOp[user.UserAccount](UserDB, AccountRedis),
|
||
}
|
||
}
|
||
|
||
func (s *UserAccountOp) redisKey(username string) string {
|
||
return fmt.Sprintf("username:%s", username)
|
||
}
|
||
|
||
func (s *UserAccountOp) GetUserAccount(username string) (*user.UserAccount, pb.ErrCode) {
|
||
sUid, err := s.accountRedis.Get(context.Background(), s.redisKey(username)).Result()
|
||
if err != nil {
|
||
if errors.Is(err, redis.Nil) {
|
||
var us user.UserAccount
|
||
err = s.db.Where("username = ?", username).First(&us).Error
|
||
if err != nil {
|
||
log.ErrorF("find user:%v err:%v", username, err)
|
||
return nil, pb.ErrCode_SystemErr
|
||
}
|
||
} else {
|
||
log.ErrorF("find user:%v err:%v", username, err)
|
||
return nil, pb.ErrCode_SystemErr
|
||
}
|
||
}
|
||
uid, _ := strconv.ParseInt(sUid, 10, 64)
|
||
if uid < 0 {
|
||
log.ErrorF("get user account:%v failed, uid is: %d", username, uid)
|
||
return nil, pb.ErrCode_SystemErr
|
||
}
|
||
return s.accountOp.Find(uint(uid))
|
||
}
|
||
|
||
// 创建用户
|
||
func (s *UserAccountOp) CreateUserAccount(us *user.UserAccount) (*user.UserAccount, pb.ErrCode) {
|
||
// 密码加密
|
||
hashedPassword, err := utils.Password(us.Password)
|
||
if err != nil {
|
||
log.ErrorF("username :%v generate password err:%v", us.Username, err)
|
||
return nil, pb.ErrCode_SystemErr
|
||
}
|
||
us.Password = hashedPassword
|
||
var code pb.ErrCode
|
||
us, code = s.accountOp.Create(us)
|
||
if code != pb.ErrCode_OK {
|
||
return nil, code
|
||
}
|
||
s.accountRedis.Set(context.Background(), s.redisKey(us.Username), us.ID, model.TableExpire)
|
||
return us, pb.ErrCode_OK
|
||
}
|
||
|
||
// 更新密码
|
||
func (s *UserAccountOp) UpdateUserPassword(us *user.UserAccount) (*user.UserAccount, pb.ErrCode) {
|
||
// 密码加密
|
||
hashedPassword, err := utils.Password(us.Password)
|
||
if err != nil {
|
||
log.ErrorF("username :%v generate password err:%v", us.Username, err)
|
||
return nil, pb.ErrCode_SystemErr
|
||
}
|
||
var code pb.ErrCode
|
||
us, code = s.accountOp.Update(us.ID, map[string]any{"password": hashedPassword})
|
||
if code != pb.ErrCode_OK {
|
||
s.accountRedis.Expire(context.Background(), s.redisKey(us.Username), model.TableExpire)
|
||
}
|
||
return us, code
|
||
}
|
||
|
||
//// 记录登录日志
|
||
//func (s *UserAccountOp) recordLoginLog(userID uint, ip, deviceID string, success bool, failReason string) {
|
||
// logEntry := user.UserLoginLog{
|
||
// PlayerID: userID,
|
||
// LoginIP: ip,
|
||
// DeviceInfo: deviceID,
|
||
// LoginResult: success,
|
||
// FailReason: failReason,
|
||
// }
|
||
//
|
||
// if err := s.logDb.Create(&logEntry).Error; err != nil {
|
||
// log.ErrorF("记录登录日志失败: %v", err)
|
||
// }
|
||
//}
|
||
//
|
||
//// 生成JWT令牌(简化版)
|
||
//func generateToken(userID uint, username string) (string, error) {
|
||
// _ = userID
|
||
// _ = username
|
||
// // 这里应该使用JWT库生成实际令牌
|
||
// // 简化实现，实际项目中请使用安全的JWT实现
|
||
// return "generated-token-placeholder", nil
|
||
//}
|