From 06d334585a2174726e3574ea939e10beec620bae Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Sat, 02 Apr 2022 17:55:39 +0800 Subject: [PATCH] fix some bug and problem and add agent return --- src/main/java/cc/mrbird/febs/dapp/service/impl/DappUserServiceImpl.java | 14 +++++++++++++- 1 files changed, 13 insertions(+), 1 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappUserServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappUserServiceImpl.java index 3e75e13..f4f40ac 100644 --- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappUserServiceImpl.java +++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappUserServiceImpl.java @@ -48,9 +48,22 @@ @Override @Transactional(rollbackFor = Exception.class) public void add(User user) { + User currentUser = FebsUtil.getCurrentUser(); DappMemberEntity member = dappMemberDao.selectMemberInfoByInviteId(user.getInviteId()); if(member == null) { throw new FebsException("邀请码错误"); + } + + if (currentUser.getDeptId() == null) { + DappUserMemberRelateEntity relate = dappUserMemberRelateDao.selectByUserId(currentUser.getUserId()); + if (!relate.getInviteId().equals(member.getRefererId())) { + throw new FebsException("该用户不是直推"); + } + } + + DappUserMemberRelateEntity userMemberRelate = dappUserMemberRelateDao.selectByInviteId(user.getInviteId()); + if (userMemberRelate != null) { + throw new FebsException("邀请码用户已存在"); } user.setCreateTime(new Date()); @@ -66,7 +79,6 @@ userRole.setUserId(user.getUserId()); userRoleMapper.insert(userRole); - User currentUser = FebsUtil.getCurrentUser(); DappUserMemberRelateEntity relate = new DappUserMemberRelateEntity(); relate.setMemberId(member.getId()); relate.setUserId(user.getUserId()); -- Gitblit v1.9.1