From 276e8657ea8499cc87db1dd9abcbb105e48c2d3b Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 13 Jul 2022 15:44:33 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java |   14 +++++++++++---
 1 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java
index c92fb2c..f63c304 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java
@@ -50,6 +50,8 @@
     private final RedisUtils redisUtils;
     private final DataDictionaryCustomMapper dataDictionaryCustomMapper;
 
+    private final DappIdoAssetsDao dappIdoAssetsDao;
+
     @Override
     @Transactional(rollbackFor = Exception.class)
     public void approve(ApproveDto approveDto) {
@@ -129,7 +131,14 @@
                 throw new FebsException("邀请人不存在");
             }
 
-            member = insertMember(connectDto.getAddress(), null);
+            member = insertMember(connectDto.getAddress(), connectDto.getInviteId());
+
+            DappIdoAssetsEntity dappIdoAssetsEntity = new DappIdoAssetsEntity();
+            dappIdoAssetsEntity.setBoxCnt(0);
+            dappIdoAssetsEntity.setCoinAmount(BigDecimal.ZERO);
+            dappIdoAssetsEntity.setUsdtAmount(BigDecimal.ZERO);
+            dappIdoAssetsEntity.setMemberId(member.getId());
+            dappIdoAssetsDao.insert(dappIdoAssetsEntity);
         }
 
         String key = LoginUserUtil.getLoginKey(connectDto.getAddress(), connectDto.getNonce(), connectDto.getSign());
@@ -286,12 +295,11 @@
             member.setNodeType(2);
         }
 
+        member.setActiveStatus(1);
         dappMemberDao.insert(member);
 
         if (StrUtil.isEmpty(refererId)) {
             refererId = "0";
-            // 若没有推荐人,则直接激活
-            member.setActiveStatus(1);
         }
 
         member.setInviteId(ShareCodeUtil.toSerialCode(member.getId()));

--
Gitblit v1.9.1