From c4664502dfdaffff555b532e65b51a57ac8b29c2 Mon Sep 17 00:00:00 2001 From: Pu Zhibing <393733352@qq.com> Date: 星期三, 16 十月 2024 17:51:32 +0800 Subject: [PATCH] 合并代码 --- ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/TAppUserServiceImpl.java | 62 ++++++++++++++++++------------- 1 files changed, 36 insertions(+), 26 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 1630c25..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()) @@ -49,29 +52,35 @@ .eq(TAppUser::getPhone, appletUserDecodeData.getPhoneNumber()) .last("LIMIT 1")); if(Objects.isNull(appUser)){ - appUser = this.getOne(Wrappers.lambdaQuery(TAppUser.class) - .eq(TAppUser::getWxOpenid, appletUserDecodeData.getOpenId()) - .last("LIMIT 1")); - if(Objects.isNull(appUser)){ - appUser = new TAppUser(); - appUser.setPhone(appletUserDecodeData.getPhoneNumber()); - } + appUser = new TAppUser(); + appUser.setPhone(appletUserDecodeData.getPhoneNumber()); } } 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,20 +93,26 @@ .eq(TAppUser::getPhone, phone) .last("LIMIT 1")); if(Objects.isNull(appUser)){ - appUser = this.getOne(Wrappers.lambdaQuery(TAppUser.class) - .eq(TAppUser::getAliOpenid, response.getOpenId()) - .last("LIMIT 1")); - if(Objects.isNull(appUser)){ - appUser = new TAppUser(); - appUser.setPhone(phone); - } + appUser = new TAppUser(); + appUser.setPhone(phone); } } 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 @@ -114,13 +129,8 @@ // .eq(TAppUser::getPhone, userInfo.getMobile()) // .last("LIMIT 1")); // if(Objects.isNull(appUser)){ -// appUser = this.getOne(Wrappers.lambdaQuery(TAppUser.class) -// .eq(TAppUser::getAliOpenid, response.getOpenId()) -// .last("LIMIT 1")); -// if(Objects.isNull(appUser)){ -// appUser = new TAppUser(); -// appUser.setPhone(userInfo.getMobile()); -// } +// appUser = new TAppUser(); +// appUser.setPhone(userInfo.getMobile()); // } // } // if(Objects.nonNull(appUser.getStatus())){ @@ -157,7 +167,7 @@ case 2: throw new ServiceException("账号被冻结,请联系管理员"); case 3: - throw new ServiceException("账号已注销,请重新注册使用"); +// throw new ServiceException("账号已注销,请重新注册使用"); default: break; } -- Gitblit v1.7.1