diff --git a/internal/controller/wx/wx_v1_we_chat_event.go b/internal/controller/wx/wx_v1_we_chat_event.go index 82616f1..43a98c9 100644 --- a/internal/controller/wx/wx_v1_we_chat_event.go +++ b/internal/controller/wx/wx_v1_we_chat_event.go @@ -29,8 +29,7 @@ func (c *ControllerV1) WeChatEvent(ctx context.Context, req *v1.WeChatEventReq) switch req.Event { case "subscribe": key := strings.TrimPrefix(req.EventKey, "qrscene_") - split := strings.Split(req.EventKey, "_") - out, err := service.User().Login(ctx, &model.UserLoginIn{OpenId: unionid, StoreCode: split[0]}) + out, err := service.User().Login(ctx, &model.UserLoginIn{OpenId: unionid, SceneId: key}) if err != nil { return nil, err } @@ -39,8 +38,7 @@ func (c *ControllerV1) WeChatEvent(ctx context.Context, req *v1.WeChatEventReq) } return nil, nil case "SCAN": - split := strings.Split(req.EventKey, "_") - out, err := service.User().Login(ctx, &model.UserLoginIn{OpenId: unionid, StoreCode: split[0]}) + out, err := service.User().Login(ctx, &model.UserLoginIn{OpenId: unionid, SceneId: req.EventKey}) if err != nil { return nil, err } diff --git a/internal/logic/user/user.go b/internal/logic/user/user.go index 98d8573..c28e820 100644 --- a/internal/logic/user/user.go +++ b/internal/logic/user/user.go @@ -6,6 +6,7 @@ import ( "github.com/gogf/gf/v2/frame/g" "github.com/gogf/gf/v2/os/glog" "github.com/gogf/gf/v2/os/gtime" + "github.com/gogf/gf/v2/util/gconv" "github.com/gogf/gf/v2/util/grand" "server/internal/consts" "server/internal/dao" @@ -19,6 +20,7 @@ import ( "server/utility/gamelife" "server/utility/jwt" "server/utility/sms" + "strings" ) type sUser struct{} @@ -53,74 +55,109 @@ func New() service.IUser { } func (s *sUser) Login(ctx context.Context, in *model.UserLoginIn) (out *model.UserLoginOut, err error) { + // Fetch role information value, err := dao.Roles.Ctx(ctx).Where(do.Roles{Code: consts.UserRoleCode}).Fields(dao.Roles.Columns().Code, dao.Roles.Columns().Id).One() if err != nil { return nil, ecode.Fail.Sub("查找角色失败") } - // 根据 OpenId 查找用户 - exist, err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId}).Exist() - if err != nil { - return nil, ecode.Fail.Sub("查找用户失败") + + split := strings.Split(in.SceneId, "_") + if len(split) < 3 { + return nil, ecode.Fail.Sub("SceneId格式错误") } + storeId := gconv.Int64(split[0]) + var xyUserId string var userId int64 - v, err := dao.Stores.Ctx(ctx).Where(do.Stores{StoreCode: in.StoreCode}).Fields(dao.Stores.Columns().Id).Value() - if err != nil { - return nil, ecode.Fail.Sub("查找门店失败") - } - if !exist { - // 用户不存在,创建新用户 - // 生成 username: qy_ + 8位随机字母数字 - var username string - for { - randomStr := grand.Str("abcdefghijklmnopqrstuvwxyz0123456789", 8) - username = "qy_" + randomStr - // 检查 username 是否唯一 - count, err := dao.Users.Ctx(ctx).Where(do.Users{Username: username}).Count() + + if split[1] == "0" { + // Case 1: split[1] == "0", use in.OpenId to find or create user + exist, err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId}).Exist() + if err != nil { + return nil, ecode.Fail.Sub("查找用户失败") + } + + if !exist { + // User does not exist, create new user + var username string + for { + randomStr := grand.Str("abcdefghijklmnopqrstuvwxyz0123456789", 8) + username = "qy_" + randomStr + count, err := dao.Users.Ctx(ctx).Where(do.Users{Username: username}).Count() + if err != nil { + return nil, ecode.Fail.Sub("检查用户名失败") + } + if count == 0 { + break // Username is unique + } + } + password, err := encrypt.EncryptPassword(consts.DefaultPassword) if err != nil { - return nil, ecode.Fail.Sub("检查用户名失败") + return nil, ecode.Fail.Sub("加密密码失败") } - if count == 0 { - break // username 唯一,退出循环 + user := &entity.Users{ + WxOpenId: in.OpenId, + Username: username, + Nickname: username, + PasswordHash: password, + Avatar: consts.DefaultUserAvatar, + FirstVisitAt: gtime.Now(), + LastLoginAt: gtime.Now(), + WxPopenId: utility.GenerateUserID("WX"), + QqPopenId: utility.GenerateUserID("QQ"), + RoleId: value[dao.Roles.Columns().Id].Int64(), + LastLoginStoreId: storeId, + } + result, err := dao.Users.Ctx(ctx).Insert(user) + if err != nil { + return nil, ecode.Fail.Sub("创建用户失败") + } + userId, err = result.LastInsertId() + if err != nil { + return nil, ecode.Fail.Sub("获取用户ID失败") + } + } else { + // User exists, update last login store ID and time + var user entity.Users + if err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId}).Scan(&user); err != nil { + return nil, ecode.Fail.Sub("查找用户失败") + } + userId = user.Id + if _, err := dao.Users.Ctx(ctx).Where(do.Users{Id: userId}).Update(do.Users{ + LastLoginAt: gtime.Now(), + LastLoginStoreId: storeId, + }); err != nil { + return nil, ecode.Fail.Sub("更新登录时间失败") } } - password, err := encrypt.EncryptPassword(consts.DefaultPassword) - if err != nil { - return nil, ecode.Fail.Sub("加密密码失败") - } - user := &entity.Users{ - WxOpenId: in.OpenId, - Username: username, - Nickname: username, - PasswordHash: password, - Avatar: consts.DefaultUserAvatar, - FirstVisitAt: gtime.Now(), - LastLoginAt: gtime.Now(), - WxPopenId: utility.GenerateUserID("WX"), - QqPopenId: utility.GenerateUserID("QQ"), - RoleId: value[dao.Roles.Columns().Id].Int64(), - LastLoginStoreId: v.Int64(), - } - result, err := dao.Users.Ctx(ctx).Insert(user) - if err != nil { - return nil, ecode.Fail.Sub("创建用户失败") - } - userId, err = result.LastInsertId() - if err != nil { - return nil, ecode.Fail.Sub("获取用户ID失败") - } - } else { - // 用户存在,更新最后登录时间 + } else if split[1] == "1" { + // Case 2: split[1] == "1", use xyUserId to find user and bind in.OpenId + xyUserId = split[2] var user entity.Users - if err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId}).Scan(&user); err != nil { + exist, err := dao.Users.Ctx(ctx).Where(do.Users{XyUserId: xyUserId}).Exist() + if err != nil { + return nil, ecode.Fail.Sub("查找用户失败") + } + if !exist { + return nil, ecode.Fail.Sub("参数错误:用户不存在") + } + + // User exists, bind in.OpenId and update last login store ID + if err := dao.Users.Ctx(ctx).Where(do.Users{XyUserId: xyUserId}).Scan(&user); err != nil { return nil, ecode.Fail.Sub("查找用户失败") } userId = user.Id - if _, err := dao.Users.Ctx(ctx).Where(do.Users{Id: userId}).Update(do.Users{LastLoginAt: gtime.Now(), LastLoginStoreId: v.Int64()}); err != nil { - return nil, ecode.Fail.Sub("更新登录时间失败") + if _, err := dao.Users.Ctx(ctx).Where(do.Users{Id: userId}).Update(do.Users{ + WxOpenId: in.OpenId, + LastLoginAt: gtime.Now(), + LastLoginStoreId: storeId, + }); err != nil { + return nil, ecode.Fail.Sub("绑定OpenId失败") } + } else { + return nil, ecode.Fail.Sub("无效的SceneId类型") } - // 生成 token + // Generate token token, err := jwt.GenerateToken(&jwt.TokenIn{ UserId: userId, Role: value[dao.Roles.Columns().Code].String(), @@ -132,7 +169,7 @@ func (s *sUser) Login(ctx context.Context, in *model.UserLoginIn) (out *model.Us out = &model.UserLoginOut{ Token: token, } - return + return out, nil } func (s *sUser) WeChatLogin(ctx context.Context, in *model.WeChatLogin) (out *model.WeChatLoginOut, err error) { return diff --git a/internal/model/user.go b/internal/model/user.go index 71d95c8..6256c6b 100644 --- a/internal/model/user.go +++ b/internal/model/user.go @@ -59,8 +59,8 @@ type LoginCache struct { } type UserLoginIn struct { - OpenId string - StoreCode string // 门店编码,登录时根据门店编码查询,填写最近一次登录门店 id + OpenId string + SceneId string } type WeChatLogin struct { UUID string // 使用的是微信的唯一 id,而不是openid,保证微信服务号,小程序绑定的系统用户为同一个