From 5264c0165e3dbdeec367081ea68e18bffc4bd060 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 29 Aug 2022 16:17:25 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/BscUsdtContractEvent.java |  104 +++++++++++++++++++++++++++++++---------------------
 1 files changed, 62 insertions(+), 42 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 3c5f2c5..abbc930 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
@@ -4,12 +4,15 @@
 import cc.mrbird.febs.common.utils.RedisUtils;
 import cc.mrbird.febs.common.utils.ShareCodeUtil;
 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.mapper.DappFundFlowDao;
-import cc.mrbird.febs.dapp.mapper.DappMemberDao;
+import cc.mrbird.febs.dapp.entity.*;
+import cc.mrbird.febs.dapp.mapper.*;
 import cc.mrbird.febs.dapp.service.DappMemberService;
+import cc.mrbird.febs.dapp.service.DappSystemService;
+import cc.mrbird.febs.dapp.utils.OnlineTransferUtil;
+import cc.mrbird.febs.rabbit.producer.ChainProducer;
+import cn.hutool.core.collection.CollUtil;
 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;
@@ -19,6 +22,9 @@
 import java.math.BigDecimal;
 import java.math.BigInteger;
 import java.math.RoundingMode;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
 
 @Slf4j
 @Service
@@ -26,62 +32,76 @@
 
     @Resource
     private RedisUtils redisUtils;
-
-    @Resource
-    private DappMemberService dappMemberService;
-
     @Resource
     private DappFundFlowDao dappFundFlowDao;
+    @Resource
+    private DappWalletCoinDao dappWalletCoinDao;
+    @Resource
+    private ChainProducer chainProducer;
 
 
     @Override
     public void compile(EthUsdtContract.TransferEventResponse e) {
-        ContractChainService sourceUsdtInstance = ChainService.getInstance(ChainEnum.BSC_USDT.name());
-        int decimals = sourceUsdtInstance.decimals();
-        int tfcDecimals = ChainService.getInstance(ChainEnum.BSC_TFC.name()).decimals();
+        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)) {
-            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());
+        if (ChainEnum.BSC_USDT.getAddress().toLowerCase().equals(e.to)) {
+            log.info("触发USDT合约监听事件");
+            redisUtils.set(AppContants.REDIS_KEY_BLOCK_ETH_INCREMENT_NUM, e.log.getBlockNumber());
+
+            // 如果得到触发,则休眠10秒。 因为此处监听器触发可能优先于前端调用transfer接口
+            try {
+                Thread.sleep(10000);
+            } catch (InterruptedException ex) {
+                ex.printStackTrace();
             }
 
-            String hasStart = redisUtils.getString(AppContants.SYSTEM_START_FLAG);
+            ContractChainService sourceUsdtInstance = ChainService.getInstance(ChainEnum.BSC_USDT.name());
+            int decimals = sourceUsdtInstance.decimals();
+            if (e.from.equals("0xaa25aa7a19f9c426e07dee59b12f944f4d9f1dd3")) {
+                return;
+            }
 
             BigInteger tokens = e.tokens;
-            BigDecimal amount = BigDecimal.valueOf(tokens.intValue()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
-            DappFundFlowEntity fundFlow = dappFundFlowDao.selectByFromHash(e.log.getTransactionHash(), 1);
+            BigDecimal amount = new BigDecimal(tokens.toString()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
 
-            BigDecimal newPrice = fundFlow.getNewestPrice();
-            BigDecimal transferAmount = amount.divide(newPrice, tfcDecimals, RoundingMode.HALF_DOWN);
+            DappFundFlowEntity fundFlow = dappFundFlowDao.selectByFromHash(e.log.getTransactionHash(), null);
+            if (fundFlow != null && fundFlow.getStatus() != 1) {
+                return;
+            }
 
+            if (fundFlow == null) {
+                List<DappFundFlowEntity> flows = dappFundFlowDao.selectFundFlowListByAddress(e.from, 1);
+                if (CollUtil.isEmpty(flows)) {
+                    OnlineTransferUtil.addTransferRecord(e.from, e.to, amount, e.log.getTransactionHash(), DappTransferRecordEntity.TRANSFER_SOURCE_FLAG_ONLINE, "USDT");
+                    log.info("本地无交易:{}", e.log.getTransactionHash());
+                    return;
+                }
+
+                for (DappFundFlowEntity flow : flows) {
+                    if (flow.getStatus() == 1) {
+                        if (amount.compareTo(flow.getAmount().multiply(flow.getNewestPrice()).setScale(4, RoundingMode.HALF_UP)) == 0) {
+                            fundFlow = flow;
+                            fundFlow.setFromHash(e.log.getTransactionHash());
+                            break;
+                        }
+                    }
+                }
+            }
+
+            if (fundFlow == null) {
+                return;
+            }
+
+            fundFlow.setAmount(fundFlow.getAmount().negate());
             // 更改状态为已同步
             fundFlow.setStatus(2);
-            fundFlow.setTargetAmount(transferAmount);
             dappFundFlowDao.updateById(fundFlow);
 
-            // 如果系统会开启,则使用自动打款
-            if (!"start".equals(hasStart)) {
-                String hash = ChainService.getInstance(ChainEnum.BSC_TFC_MAKE.name()).transfer(e.from, transferAmount);
-
-                // 更新为已打款
-                fundFlow.setStatus(3);
-                fundFlow.setToHash(hash);
-                dappFundFlowDao.updateById(fundFlow);
-            }
-
-            // 若源池中的USDT达到或超过8万U,则启动整个系统
-            BigDecimal balance = sourceUsdtInstance.balanceOf(ChainEnum.BSC_USDT_SOURCE.getAddress());
-            if (BigDecimal.valueOf(80000).compareTo(balance) < 1) {
-                redisUtils.set(AppContants.SYSTEM_START_FLAG, "start");
-            }
+            chainProducer.sendAchieveTreeMsg(fundFlow.getMemberId());
         }
-    }
-
-    public static void main(String[] args) {
-        System.out.println(ChainService.getInstance(ChainEnum.BSC_USDT.name()).balanceOf("0x9DDE1834683D642D4D077498DC7fbdb8CF70E8FE"));
     }
 }

--
Gitblit v1.9.1