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