From 03fa3ef5588ce2da29374693c77235457f655ebb Mon Sep 17 00:00:00 2001 From: zhangli <123879394@qq.com> Date: Sun, 24 Aug 2025 13:04:28 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=BC=95=E8=8D=90=E4=BA=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../sczx/user/service/impl/UserServiceImpl.java | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) 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);