diff --git a/src/main/java/com/sczx/user/service/impl/UserServiceImpl.java b/src/main/java/com/sczx/user/service/impl/UserServiceImpl.java index d75ac75..ee02f39 100644 --- a/src/main/java/com/sczx/user/service/impl/UserServiceImpl.java +++ b/src/main/java/com/sczx/user/service/impl/UserServiceImpl.java @@ -120,6 +120,23 @@ public class UserServiceImpl implements IUserService { } baseUserRepo.saveOrUpdate(newUserPO); + if(Objects.nonNull(wxMiniProgramRegRequest.getReferralUserId())){ + LambdaQueryWrapper queryReferralWrapper = new LambdaQueryWrapper<>(); + queryReferralWrapper.eq(BaseUserReferralPO::getUserId, baseUserPO.getId()).last( " limit 1"); + BaseUserReferralPO baseUserReferralPO = baseUserReferralRepo.getOne(queryReferralWrapper); + if(Objects.isNull(baseUserReferralPO)){ + baseUserReferralPO = new BaseUserReferralPO(); + BaseUserPO referralUserPO = baseUserRepo.getById(wxMiniProgramRegRequest.getReferralUserId()); + baseUserReferralPO.setUserId(baseUserPO.getId()); + baseUserReferralPO.setUserPhoneNumber(phoneNumber); + baseUserReferralPO.setReferralUserId(wxMiniProgramRegRequest.getReferralUserId()); + baseUserReferralPO.setReferralUserPhoneNumber(referralUserPO.getPhoneNumber()); + //TODO-获取推荐奖励 + //保存推荐信息 + baseUserReferralRepo.save(baseUserReferralPO); + } + } + return getLoginResponse(openid,MiniProgramTypeEnum.WECHAT.getType()); }catch (Exception e){ log.error("微信小程序注册失败", e);