From a146097a15d719c60af3cdeb4de1f21aa7b5ca8d Mon Sep 17 00:00:00 2001 From: Pu Zhibing <393733352@qq.com> Date: 星期五, 20 九月 2024 10:47:53 +0800 Subject: [PATCH] Merge branch 'master' of http://120.76.84.145:10101/gitblit/r/java/mx_charging_pile --- ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/TAppUserServiceImpl.java | 35 ++++++++++++++++++++++++++++++----- 1 files changed, 30 insertions(+), 5 deletions(-) diff --git a/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/TAppUserServiceImpl.java b/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/TAppUserServiceImpl.java index d3351ef..66ff91a 100644 --- a/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/TAppUserServiceImpl.java +++ b/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/TAppUserServiceImpl.java @@ -10,6 +10,7 @@ import com.ruoyi.account.api.model.TAppUser; import com.ruoyi.account.mapper.TAppUserMapper; import com.ruoyi.account.service.TAppUserService; +import com.ruoyi.account.service.TInviteUserService; import com.ruoyi.account.wx.model.WeixinProperties; import com.ruoyi.account.wx.pojo.AppletUserDecodeData; import com.ruoyi.common.core.exception.ServiceException; @@ -35,8 +36,10 @@ public class TAppUserServiceImpl extends ServiceImpl<TAppUserMapper, TAppUser> implements TAppUserService { @Autowired private TokenService tokenService; + @Autowired + private TInviteUserService inviteUserService; @Override - public Map<String, Object> wxLogin(AppletUserDecodeData appletUserDecodeData) { + public Map<String, Object> wxLogin(AppletUserDecodeData appletUserDecodeData,Long inviteUserId) { // 通过手机号查询用户,是否已存在手动导入用户,包含支付宝用户 TAppUser appUser = this.getOne(Wrappers.lambdaQuery(TAppUser.class) .eq(TAppUser::getPhone, appletUserDecodeData.getPhoneNumber()) @@ -54,19 +57,30 @@ } } if(Objects.nonNull(appUser.getStatus())){ - throwInfo(appUser.getStatus()); + switch (appUser.getStatus()){ + case 1: + break; + case 2: + throw new ServiceException("账号被冻结,请联系管理员"); + case 3: + appUser = new TAppUser(); + appUser.setPhone(appletUserDecodeData.getPhoneNumber()); + break; + } } + appUser.setInviteUserId(inviteUserId); appUser.setAvatar(appletUserDecodeData.getAvatarUrl()); appUser.setCity(appletUserDecodeData.getCity()); appUser.setName(appletUserDecodeData.getNickName()); appUser.setProvince(appletUserDecodeData.getProvince()); appUser.setWxOpenid(appletUserDecodeData.getOpenId()); this.saveOrUpdate(appUser); + inviteUserService.saveInviteUser(appUser.getId(), inviteUserId); return this.getUserInfo(appUser); } @Override - public Map<String, Object> aliLogin(AlipaySystemOauthTokenResponse response,String phone) { + public Map<String, Object> aliLogin(AlipaySystemOauthTokenResponse response,String phone,Long inviteUserId) { // 通过手机号查询用户,是否已存在手动导入用户,包含微信用户 TAppUser appUser = this.getOne(Wrappers.lambdaQuery(TAppUser.class) .eq(TAppUser::getPhone, phone) @@ -84,10 +98,21 @@ } } if(Objects.nonNull(appUser.getStatus())){ - throwInfo(appUser.getStatus()); + switch (appUser.getStatus()){ + case 1: + break; + case 2: + throw new ServiceException("账号被冻结,请联系管理员"); + case 3: + appUser = new TAppUser(); + appUser.setPhone(phone); + break; + } } + appUser.setInviteUserId(inviteUserId); appUser.setAliOpenid(response.getOpenId()); this.saveOrUpdate(appUser); + inviteUserService.saveInviteUser(appUser.getId(), inviteUserId); return this.getUserInfo(appUser); } // @Override @@ -142,7 +167,7 @@ case 2: throw new ServiceException("账号被冻结,请联系管理员"); case 3: - throw new ServiceException("账号已注销,请重新注册使用"); +// throw new ServiceException("账号已注销,请重新注册使用"); default: break; } -- Gitblit v1.7.1