From 6562e04677044d4c15ee303d922bfeabe56320fb Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Tue, 07 Jun 2022 19:50:34 +0800
Subject: [PATCH] test and fix some bugs

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/BscUsdtContractEvent.java |   45 ++++++++++++++++++++++++++++++++++-----------
 1 files changed, 34 insertions(+), 11 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 ad0ad6d..e4a5c7d 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
@@ -13,6 +13,7 @@
 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.dapp.utils.OnlineTransferUtil;
 import cc.mrbird.febs.rabbit.producer.ChainProducer;
 import cn.hutool.core.util.StrUtil;
 import com.alibaba.fastjson.JSONObject;
@@ -58,7 +59,16 @@
 
         redisUtils.set(AppContants.REDIS_KEY_BLOCK_USDT_NUM, e.log.getBlockNumber());
         // 判断对方打款地址是否为源池地址
-        if (ChainEnum.BSC_TFC_SOURCE.getAddress().equals(e.to)) {
+        if (ChainEnum.BSC_TFC_SOURCE.getAddress().toLowerCase().equals(e.to)) {
+            log.info("触发USDT合约监听事件");
+
+            // 如果得到触发,则休眠10秒。 因为此处监听器触发可能优先于前端调用transfer接口
+            try {
+                Thread.sleep(10000);
+            } catch (InterruptedException ex) {
+                ex.printStackTrace();
+            }
+
             ContractChainService sourceUsdtInstance = ChainService.getInstance(ChainEnum.BSC_USDT.name());
             int decimals = sourceUsdtInstance.decimals();
             int tfcDecimals = ChainService.getInstance(ChainEnum.BSC_TFC.name()).decimals();
@@ -77,8 +87,13 @@
             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);
+            BigDecimal amount = new BigDecimal(tokens.toString()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
             DappFundFlowEntity fundFlow = dappFundFlowDao.selectByFromHash(e.log.getTransactionHash(), 1);
+
+            if (fundFlow == null) {
+                log.info("本地无交易:{}", e.log.getTransactionHash());
+                return;
+            }
 
             BigDecimal newPrice = fundFlow.getNewestPrice();
             BigDecimal transferAmount = amount.divide(newPrice, tfcDecimals, RoundingMode.HALF_DOWN);
@@ -91,8 +106,7 @@
 
             // 如果系统会开启,则使用自动打款
             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);
+                OnlineTransferUtil.addTransfer(e.from, transferAmount, 1, 1, ChainEnum.BSC_TFC_MAKE.name(), "BSC", fundFlow.getId().toString());
 
                 Map<String, String> map = new HashMap<>();
                 map.put("batchNo", fundFlow.getId().toString());
@@ -100,15 +114,24 @@
 
                 // 发送转账消息
                 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());
+                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");
+                // 若源池中的USDT达到或超过8万U,则启动整个系统
+                BigDecimal balance = sourceUsdtInstance.balanceOf(ChainEnum.BSC_USDT_SOURCE.getAddress());
+                if (target.compareTo(balance) < 1) {
+                    redisUtils.set(AppContants.SYSTEM_START_FLAG, "start");
+                }
+            } else {
+                OnlineTransferUtil.addTransfer(e.from, transferAmount, 1, 1, ChainEnum.BSC_TFC_SOURCE.name(), AppContants.SYMBOL_COIN, fundFlow.getId().toString());
+
+                Map<String, String> map = new HashMap<>();
+                map.put("batchNo", fundFlow.getId().toString());
+                map.put("type", "flow");
+
+                // 发送转账消息
+                chainProducer.sendOnlineTransfer(JSONObject.toJSONString(map));
             }
         }
     }

--
Gitblit v1.9.1