From 18b363581c21fc5a8fba893a30fd652b00433e77 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 06 Apr 2022 10:05:35 +0800
Subject: [PATCH] fix
---
src/main/java/cc/mrbird/febs/dapp/service/impl/DappUserServiceImpl.java | 16 ++++++++++++++--
1 files changed, 14 insertions(+), 2 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..b4a7d9f 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
@@ -26,7 +26,7 @@
import java.util.List;
/**
- * @author wzy
+ * @author
* @date 2022-03-24
**/
@Slf4j
@@ -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