From c0989001fbddc1b6e24783c3a4b8cb8775eab1fe Mon Sep 17 00:00:00 2001
From: wzy <wzy19931122ai@163.com>
Date: Sun, 06 Nov 2022 19:32:01 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/BscUsdtContractEvent.java |   95 +++++++++++++++++++++++++++++++++++------------
 1 files changed, 70 insertions(+), 25 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 7b1c321..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
@@ -1,16 +1,18 @@
 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.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.chain.*;
+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;
@@ -20,43 +22,86 @@
 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
 public class BscUsdtContractEvent implements ContractEventService {
 
     @Resource
-    private DappMemberService dappMemberService;
-
+    private RedisUtils redisUtils;
     @Resource
     private DappFundFlowDao dappFundFlowDao;
+    @Resource
+    private DappWalletCoinDao dappWalletCoinDao;
+    @Resource
+    private ChainProducer chainProducer;
 
-    private final String ADDRESS = "0x977a9ddfb965a9a3416fa72ca7f91c4949c18f25";
-    int decimals = ChainService.getInstance(ChainEnum.BSC_USDT.name()).decimals();
-    int tfcDecimals = ChainService.getInstance(ChainEnum.BSC_TFC.name()).decimals();
 
     @Override
     public void compile(EthUsdtContract.TransferEventResponse e) {
-        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_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();
+            }
+
+            ContractChainService sourceUsdtInstance = ChainService.getInstance(ChainEnum.BSC_USDT.name());
+            int decimals = sourceUsdtInstance.decimals();
+            if (e.from.equals("0xaa25aa7a19f9c426e07dee59b12f944f4d9f1dd3")) {
+                return;
             }
 
             BigInteger tokens = e.tokens;
+            BigDecimal amount = new BigDecimal(tokens.toString()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
 
-            BigDecimal amount = BigDecimal.valueOf(tokens.intValue()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
+            DappFundFlowEntity fundFlow = dappFundFlowDao.selectByFromHash(e.log.getTransactionHash(), null);
+            if (fundFlow != null && fundFlow.getStatus() != 1) {
+                return;
+            }
 
-            DappFundFlowEntity fundFlow = new DappFundFlowEntity(fromMember.getId(), amount, 2, null, null, e.log.getTransactionHash());
-            dappFundFlowDao.insert(fundFlow);
+            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;
+                }
 
-            // TODO price
-            BigDecimal newPrice = BigDecimal.valueOf(1);
+                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;
+                        }
+                    }
+                }
+            }
 
-            BigDecimal transferAmount = amount.multiply(newPrice);
-            String hash = ChainService.getInstance(ChainEnum.BSC_TFC.name()).transfer(e.from, transferAmount);
-            fundFlow.setToHash(hash);
+            if (fundFlow == null) {
+                return;
+            }
+
+            fundFlow.setAmount(fundFlow.getAmount().negate());
+            // 更改状态为已同步
+            fundFlow.setStatus(2);
             dappFundFlowDao.updateById(fundFlow);
+
+            chainProducer.sendAchieveTreeMsg(fundFlow.getMemberId());
         }
     }
 }

--
Gitblit v1.9.1