From b7e79b1a0a954a6d15dc45d835e0024b6b51c211 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 06 Jun 2022 16:01:34 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/BscUsdtContractEvent.java |   84 ++++++++++++++++++++++++++++++++++-------
 1 files changed, 69 insertions(+), 15 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/BscUsdtContractEvent.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/BscUsdtContractEvent.java
index 0d3d668..6e5ece7 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/BscUsdtContractEvent.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/BscUsdtContractEvent.java
@@ -1,16 +1,21 @@
 package cc.mrbird.febs.dapp.service.impl;
 
+import cc.mrbird.febs.common.contants.AppContants;
+import cc.mrbird.febs.common.utils.RedisUtils;
 import cc.mrbird.febs.common.utils.ShareCodeUtil;
-import cc.mrbird.febs.dapp.chain.ChainEnum;
-import cc.mrbird.febs.dapp.chain.ChainService;
-import cc.mrbird.febs.dapp.chain.ContractEventService;
-import cc.mrbird.febs.dapp.chain.EthUsdtContract;
+import cc.mrbird.febs.dapp.chain.*;
 import cc.mrbird.febs.dapp.entity.DappFundFlowEntity;
 import cc.mrbird.febs.dapp.entity.DappMemberEntity;
+import cc.mrbird.febs.dapp.entity.DappOnlineTransferEntity;
+import cc.mrbird.febs.dapp.entity.DataDictionaryCustom;
 import cc.mrbird.febs.dapp.mapper.DappFundFlowDao;
 import cc.mrbird.febs.dapp.mapper.DappMemberDao;
+import cc.mrbird.febs.dapp.mapper.DappOnlineTransferDao;
+import cc.mrbird.febs.dapp.mapper.DataDictionaryCustomMapper;
 import cc.mrbird.febs.dapp.service.DappMemberService;
+import cc.mrbird.febs.rabbit.producer.ChainProducer;
 import cn.hutool.core.util.StrUtil;
+import com.alibaba.fastjson.JSONObject;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -20,10 +25,15 @@
 import java.math.BigDecimal;
 import java.math.BigInteger;
 import java.math.RoundingMode;
+import java.util.HashMap;
+import java.util.Map;
 
 @Slf4j
 @Service
 public class BscUsdtContractEvent implements ContractEventService {
+
+    @Resource
+    private RedisUtils redisUtils;
 
     @Resource
     private DappMemberService dappMemberService;
@@ -31,17 +41,40 @@
     @Resource
     private DappFundFlowDao dappFundFlowDao;
 
-    private final String ADDRESS = "0x977a9ddfb965a9a3416fa72ca7f91c4949c18f25";
+    @Resource
+    private ChainProducer chainProducer;
+
+    @Resource
+    private DappOnlineTransferDao dappOnlineTransferDao;
+    @Resource
+    private DataDictionaryCustomMapper dataDictionaryCustomMapper;
+
 
     @Override
     public void compile(EthUsdtContract.TransferEventResponse e) {
-        int decimals = ChainService.getInstance(ChainEnum.BSC_USDT.name()).decimals();
-        int tfcDecimals = ChainService.getInstance(ChainEnum.BSC_TFC.name()).decimals();
-        if (ADDRESS.equals(e.to)) {
-            DappMemberEntity fromMember = dappMemberService.findByAddress(e.from, null);
-            if (fromMember == null) {
-                fromMember = dappMemberService.insertMember(e.from, null);
+        if (e.to == null) {
+            return;
+        }
+
+        redisUtils.set(AppContants.REDIS_KEY_BLOCK_USDT_NUM, e.log.getBlockNumber());
+        // 判断对方打款地址是否为源池地址
+        if (ChainEnum.BSC_TFC_SOURCE.getAddress().equals(e.to)) {
+            ContractChainService sourceUsdtInstance = ChainService.getInstance(ChainEnum.BSC_USDT.name());
+            int decimals = sourceUsdtInstance.decimals();
+            int tfcDecimals = ChainService.getInstance(ChainEnum.BSC_TFC.name()).decimals();
+
+            if (e.from.equals("0xaa25aa7a19f9c426e07dee59b12f944f4d9f1dd3")) {
+                return;
             }
+
+            DappMemberEntity fromMember = dappMemberService.findByAddress(e.from, null);
+            // 如果此时fromMember为null,则说明该用户未经过转账绑定关系,而是直接注册并购买币,则将关系绑定到顶级账户
+            if (fromMember == null) {
+                DappMemberEntity toAddress = dappMemberService.findByAddress(e.to, null);
+                fromMember = dappMemberService.insertMember(e.from, toAddress.getInviteId());
+            }
+
+            String hasStart = redisUtils.getString(AppContants.SYSTEM_START_FLAG);
 
             BigInteger tokens = e.tokens;
             BigDecimal amount = BigDecimal.valueOf(tokens.intValue()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
@@ -55,10 +88,31 @@
             fundFlow.setTargetAmount(transferAmount);
             dappFundFlowDao.updateById(fundFlow);
 
-//            BigDecimal transferAmount = amount.multiply(newPrice);
-//            String hash = ChainService.getInstance(ChainEnum.BSC_TFC.name()).transfer(e.from, transferAmount);
-//            fundFlow.setToHash(hash);
-//            dappFundFlowDao.updateById(fundFlow);
+            // 如果系统会开启,则使用自动打款
+            if (!"start".equals(hasStart)) {
+                DappOnlineTransferEntity bsc = new DappOnlineTransferEntity(e.from, transferAmount, 1, 1, ChainEnum.BSC_TFC_MAKE.name(), "BSC", fundFlow.getId().toString());
+                dappOnlineTransferDao.insert(bsc);
+
+                Map<String, String> map = new HashMap<>();
+                map.put("batchNo", fundFlow.getId().toString());
+                map.put("type", "flow");
+
+                // 发送转账消息
+                chainProducer.sendOnlineTransfer(JSONObject.toJSONString(map));
+            }
+
+            DataDictionaryCustom dic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.DIC_TYPE_SYSTEM_SETTING, AppContants.DIC_VALUE_SYSTEM_START_TARGET);
+            BigDecimal target = new BigDecimal(dic.getValue());
+
+            // 若源池中的USDT达到或超过8万U,则启动整个系统
+            BigDecimal balance = sourceUsdtInstance.balanceOf(ChainEnum.BSC_USDT_SOURCE.getAddress());
+            if (target.compareTo(balance) < 1) {
+                redisUtils.set(AppContants.SYSTEM_START_FLAG, "start");
+            }
         }
     }
+
+    public static void main(String[] args) {
+        System.out.println(ChainService.getInstance(ChainEnum.BSC_USDT.name()).balanceOf("0x9DDE1834683D642D4D077498DC7fbdb8CF70E8FE"));
+    }
 }

--
Gitblit v1.9.1