From c6ebbc7a11ca228addb0cca7c5f34d2d5ae3249b Mon Sep 17 00:00:00 2001 From: liujie <1793218484@qq.com> Date: 星期三, 18 六月 2025 17:48:11 +0800 Subject: [PATCH] 更新优化 --- ruoyi-applet/src/main/java/com/ruoyi/web/controller/api/WxLoginController.java | 19 ++++++++++++++++--- 1 files changed, 16 insertions(+), 3 deletions(-) diff --git a/ruoyi-applet/src/main/java/com/ruoyi/web/controller/api/WxLoginController.java b/ruoyi-applet/src/main/java/com/ruoyi/web/controller/api/WxLoginController.java index 4a16eb7..e6854aa 100644 --- a/ruoyi-applet/src/main/java/com/ruoyi/web/controller/api/WxLoginController.java +++ b/ruoyi-applet/src/main/java/com/ruoyi/web/controller/api/WxLoginController.java @@ -98,9 +98,15 @@ tbUser.setPhone(loginBody.getPhone()); tbUser.setStatus(1); tbUser.setUserName(loginBody.getPhone()); - tbUser.setInviteId(loginBody.getInviteId()); + tbUser.setInviteId(loginBody.getInviteUserId()); tbUserService.save(tbUser); user = tbUser; + ajax.put("firstRegister",1); + }else { + ajax.put("firstRegister",0); + if(StringUtils.isEmpty(user.getInviteId()) && !StringUtils.isEmpty(loginBody.getInviteUserId())){ + user.setInviteId(loginBody.getInviteUserId()); + } } LoginUser loginUser = new LoginUser(); loginUser.setUserId(Long.valueOf(user.getId())); @@ -111,10 +117,11 @@ return ajax; } - @ApiOperation(value = "通过code获得openid,获取用户信息",tags = {"登录模块"}) + @ApiOperation(value = "微信登录 通过code获得openid,获取用户信息",tags = {"登录模块"}) @PostMapping("/openIdByJsCode") public AjaxResult openIdByJsCode(@RequestBody AppletUserEncrypteData data) { log.info("<<<<<<<<换取openid开始<<<<<<<<:{}", data.getCode()); + AjaxResult ajax = AjaxResult.success(); WxAppletTools appletTools = new WxAppletTools(wxRestTemplate, wxConfig, redisService); Code2SessionRespBody body = appletTools.getOpenIdByJscode2session(new Code2SessionResqBody().build(data.getCode())); String openid = body.getOpenid(); @@ -135,20 +142,26 @@ user.setInviteId(data.getInviteUserId()); user.setStatus(1); tbUserService.save(user); + ajax.put("firstRegister",1); }else { + ajax.put("firstRegister",0); + if(StringUtils.isEmpty(user.getInviteId()) && !StringUtils.isEmpty(data.getInviteUserId())){ + user.setInviteId(data.getInviteUserId()); + } user.setOpenId(appletUserDecodeData.getOpenId()); user.updateById(); } if(user.getStatus()==2){ return AjaxResult.error("登录失败,当前账号已被冻结"); } - AjaxResult ajax = AjaxResult.success(); + LoginUser loginUser = new LoginUser(); loginUser.setUserId(Long.valueOf(user.getId())); loginUser.setUser(new SysUser()); ajax.put(Constants.TOKEN, tokenService.createToken(loginUser)); ajax.put("username",user.getUserName()); ajax.put("avatar",user.getAvatar()); + return ajax; } -- Gitblit v1.7.1