From 21c4edde92e2c7bade302db52329422821f0a8d5 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 08 Jun 2022 19:23:45 +0800
Subject: [PATCH] fix some problem

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/BscCoinContractEvent.java |   60 +++++++++++++++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 53 insertions(+), 7 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/BscCoinContractEvent.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/BscCoinContractEvent.java
index 8ee15b5..0831463 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/BscCoinContractEvent.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/BscCoinContractEvent.java
@@ -10,6 +10,7 @@
 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.DappTransferRecordEntity;
 import cc.mrbird.febs.dapp.mapper.DappFundFlowDao;
 import cc.mrbird.febs.dapp.mapper.DappMemberDao;
 import cc.mrbird.febs.dapp.mapper.DappOnlineTransferDao;
@@ -17,8 +18,10 @@
 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;
 import org.springframework.stereotype.Service;
@@ -30,8 +33,10 @@
 import java.math.BigInteger;
 import java.math.RoundingMode;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 
+@Slf4j
 @Service
 public class BscCoinContractEvent implements ContractEventService {
 
@@ -64,7 +69,15 @@
     @Override
     public void compile(EthUsdtContract.TransferEventResponse e) {
         if (e.to != null) {
+            log.info("触发TFC监听");
+
+            try {
+                Thread.sleep(5000);
+            } catch (InterruptedException ex) {
+                ex.printStackTrace();
+            }
             redisUtils.set(AppContants.REDIS_KEY_BLOCK_COIN_NUM, e.log.getBlockNumber());
+            redisUtils.set(AppContants.REDIS_KEY_BLOCK_ETH_INCREMENT_NUM, e.log.getBlockNumber());
             int decimals = ChainService.getInstance(ChainEnum.BSC_TFC.name()).decimals();
 
             // 合约创建时,同时创建地址
@@ -83,8 +96,11 @@
 //                return;
 //            }
 
+            BigInteger tokens = e.tokens;
+            BigDecimal amount = BigDecimal.valueOf(tokens.intValue()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
+
             // 如果转账对象地址不为源池地址,且对象地址在系统中存在并未激活,则激活用户
-            if (!e.to.equals(ChainEnum.BSC_TFC_SOURCE.getAddress())) {
+            if (!e.to.equals(ChainEnum.BSC_TFC_SOURCE.getAddress().toLowerCase())) {
                 DappMemberEntity toMember = dappMemberDao.selectByAddress(e.to, null);
                 if (toMember == null) {
 //                    toMember = dappMemberService.insertMember(e.to, fromMember.getInviteId());
@@ -95,13 +111,43 @@
                     toMember.setActiveStatus(1);
                     dappMemberService.updateById(toMember);
                 }
+
+                OnlineTransferUtil.addTransferRecord(e.from, e.to, amount, e.log.getTransactionHash(), DappTransferRecordEntity.TRANSFER_SOURCE_FLAG_ONLINE, AppContants.SYMBOL_COIN);
             }
 
             // 卖币
-            if (e.to.equals(ChainEnum.BSC_TFC_SOURCE.getAddress())) {
-                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);
+            if (e.to.equals(ChainEnum.BSC_TFC_SOURCE.getAddress().toLowerCase())) {
+                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, AppContants.SYMBOL_COIN);
+                        log.info("本地无交易:{}", e.log.getTransactionHash());
+                        return;
+                    }
+
+                    boolean hasFlow = false;
+                    for (DappFundFlowEntity flow : flows) {
+                        if (flow.getStatus() == 1) {
+                            if (amount.compareTo(flow.getTargetAmount()) == 0) {
+                                hasFlow = true;
+                                fundFlow = flow;
+                                fundFlow.setFromHash(e.log.getTransactionHash());
+                                break;
+                            }
+                        }
+                    }
+
+                    if (!hasFlow) {
+                        OnlineTransferUtil.addTransferRecord(e.from, e.to, amount, e.log.getTransactionHash(), DappTransferRecordEntity.TRANSFER_SOURCE_FLAG_ONLINE, AppContants.SYMBOL_COIN);
+                        return;
+                    }
+                }
+                OnlineTransferUtil.addTransferRecord(e.from, e.to, amount, e.log.getTransactionHash(), DappTransferRecordEntity.TRANSFER_SOURCE_FLAG_APPLICATION, AppContants.SYMBOL_COIN);
 
                 BigDecimal newPrice = fundFlow.getNewestPrice();
                 BigDecimal transferAmount = amount.multiply(newPrice);
@@ -112,9 +158,9 @@
                 dappFundFlowDao.updateById(fundFlow);
 
                 // 划点分配
-                dappSystemService.tradeProfitDistribute(fundFlow.getId());
+                chainProducer.sendDitribProfit(fundFlow.getId());
 
-                OnlineTransferUtil.addTransfer(e.from, transferAmount, fundFlow.getType(), 1, ChainEnum.BSC_TFC_SOURCE.name(), AppContants.SYMBOL_COIN, fundFlow.getId().toString());
+                OnlineTransferUtil.addTransfer(e.from, transferAmount, fundFlow.getType(), 1, ChainEnum.BSC_USDT_SOURCE.name(), AppContants.SYMBOL_USDT, fundFlow.getId().toString());
 
                 Map<String, String> map = new HashMap<>();
                 map.put("batchNo", fundFlow.getId().toString());

--
Gitblit v1.9.1