From f7fc7c82d2d683a3e90e88ddbf8831bfa3800e11 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Tue, 31 May 2022 16:48:25 +0800
Subject: [PATCH] add online-tranfer, finish mining、distrib profit

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java |   74 +++++++++++++++++++++++++++++++++++++
 1 files changed, 74 insertions(+), 0 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 447a45d..7e41717 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
@@ -4,10 +4,14 @@
 import cc.mrbird.febs.common.entity.QueryRequest;
 import cc.mrbird.febs.common.exception.FebsException;
 import cc.mrbird.febs.common.utils.FebsUtil;
+import cc.mrbird.febs.common.utils.LoginUserUtil;
+import cc.mrbird.febs.common.utils.RedisUtils;
 import cc.mrbird.febs.common.utils.ShareCodeUtil;
 import cc.mrbird.febs.dapp.chain.ChainService;
 import cc.mrbird.febs.dapp.chain.ContractChainService;
 import cc.mrbird.febs.dapp.dto.ApproveDto;
+import cc.mrbird.febs.dapp.dto.ConnectDto;
+import cc.mrbird.febs.dapp.dto.PriceSettingDto;
 import cc.mrbird.febs.dapp.entity.DappMemberEntity;
 import cc.mrbird.febs.dapp.entity.DappTransferRecordEntity;
 import cc.mrbird.febs.dapp.entity.DappWalletCoinEntity;
@@ -18,6 +22,7 @@
 import cc.mrbird.febs.dapp.mapper.DappWalletMineDao;
 import cc.mrbird.febs.dapp.service.DappMemberService;
 import cc.mrbird.febs.system.entity.User;
+import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
@@ -28,6 +33,8 @@
 import org.springframework.transaction.annotation.Transactional;
 
 import java.math.BigDecimal;
+import java.util.HashMap;
+import java.util.Map;
 
 /**
  * @author 
@@ -41,6 +48,7 @@
     private final DappWalletCoinDao dappWalletCoinDao;
     private final DappWalletMineDao dappWalletMineDao;
     private final DappTransferRecordDao dappTransferRecordDao;
+    private final RedisUtils redisUtils;
 
     @Override
     @Transactional(rollbackFor = Exception.class)
@@ -109,6 +117,26 @@
         walletCoin.setAvailableAmount(BigDecimal.ZERO);
         walletCoin.setFrozenAmount(BigDecimal.ZERO);
         dappWalletCoinDao.insert(walletCoin);
+    }
+
+    @Override
+    public void connect(ConnectDto connectDto) {
+        DappMemberEntity member = dappMemberDao.selectByAddress(connectDto.getAddress(), null);
+
+        if (member == null) {
+            member = insertMember(connectDto.getAddress(), null);
+        }
+
+        String key = LoginUserUtil.getLoginKey(connectDto.getAddress(), connectDto.getNonce(), connectDto.getSign());
+        redisUtils.set(connectDto.getAddress(), member);
+
+        Map<Object, Object> keys = redisUtils.hmget(AppContants.REDIS_KEY_SIGN);
+        if (keys == null) {
+            keys = new HashMap<>();
+        }
+
+        keys.put(connectDto.getAddress(), key);
+        redisUtils.hmset(AppContants.REDIS_KEY_SIGN, keys);
     }
 
     @Override
@@ -220,4 +248,50 @@
         Page<DappTransferRecordEntity> page = new Page<>(request.getPageNum(), request.getPageSize());
         return dappTransferRecordDao.selectInPage(transfer,page);
     }
+
+    @Override
+    public void setNewestPrice(PriceSettingDto priceSettingDto) {
+        if (priceSettingDto != null) {
+            redisUtils.set(AppContants.REDIS_KEY_ETH_NEW_PRICE, priceSettingDto.getNewestPrice());
+        }
+    }
+
+    @Override
+    public DappMemberEntity insertMember(String address, String refererId) {
+        DappMemberEntity member = new DappMemberEntity();
+        member.setAddress(address);
+
+        dappMemberDao.insert(member);
+
+        member.setInviteId(ShareCodeUtil.toSerialCode(member.getId()));
+        member.setRefererId(refererId);
+        if (StrUtil.isNotBlank(refererId)){
+            boolean flag = false;
+            String parentId = refererId;
+            String ids = "";
+            while (!flag && StringUtils.isNotBlank(parentId)) {
+                if (StrUtil.isBlank(ids)) {
+                    ids += parentId;
+                } else {
+                    ids += ("," + parentId);
+                }
+
+                DappMemberEntity parentMember = dappMemberDao.selectMemberInfoByInviteId(parentId);
+                if (parentMember == null) {
+                    break;
+                }
+                parentId = parentMember.getRefererId();
+                if(StringUtils.isBlank(parentId)){
+                    break;
+                }
+                if (parentMember.getRefererId().equals(parentMember.getInviteId())) {
+                    flag = true;
+                }
+            }
+            member.setRefererIds(ids);
+        }
+        dappMemberDao.updateById(member);
+
+        return member;
+    }
 }

--
Gitblit v1.9.1