From 4a1e4abf6ed1deafdb293d3aa85d099ef5c63a1a Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Tue, 14 Jun 2022 11:28:23 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/BscCoinContractEvent.java |  125 ++++++++++++++++++++++++++++++++++++-----
 1 files changed, 108 insertions(+), 17 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 355909e..0e51a29 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
@@ -1,5 +1,7 @@
 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;
@@ -7,20 +9,34 @@
 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.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;
 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;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
+import org.web3j.utils.Numeric;
 
 import javax.annotation.Resource;
 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 BscCoinContractEvent implements ContractEventService {
 
@@ -33,55 +49,130 @@
     @Resource
     private DappFundFlowDao dappFundFlowDao;
 
-    private final String ADDRESS = "0x977a9ddfb965a9a3416fa72ca7f91c4949c18f25";
+    @Resource
+    private RedisUtils redisUtils;
+
+    @Autowired
+    private DappSystemService dappSystemService;
+
+    @Resource
+    private DappOnlineTransferDao dappOnlineTransferDao;
+
+    @Resource
+    private ChainProducer chainProducer;
+
+    public static void main(String[] args) {
+        System.out.println(Numeric.toBigInt("0x12e4e85"));
+    }
 
     @Transactional(rollbackFor = Exception.class)
     @Override
     public void compile(EthUsdtContract.TransferEventResponse e) {
-        int decimals = ChainService.getInstance(ChainEnum.BSC_TFC.name()).decimals();
         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();
+
             // 合约创建时,同时创建地址
             if ("0x0000000000000000000000000000000000000000".equals(e.from)) {
                 DappMemberEntity toMember = dappMemberDao.selectByAddress(e.to, null);
                 if (toMember == null) {
-                    toMember = dappMemberService.insertMember(e.to, "0");
+                    toMember = dappMemberService.insertMember(e.to, "0", "BSC", "admin");
                 }
                 return;
             }
 
+            // 更新余额
             DappMemberEntity fromMember = dappMemberService.findByAddress(e.from, null);
-            if (fromMember == null) {
-                DappMemberEntity sourceAddress = dappMemberDao.selectByAddress(ChainEnum.BSC_TFC.getAddress(), null);
-                fromMember = dappMemberService.insertMember(e.from, sourceAddress.getInviteId());
+            if (fromMember != null) {
+                BigDecimal balance = ChainService.getInstance(ChainEnum.BSC_TFC.name()).balanceOf(e.from);
+                fromMember.setBalance(balance);
+                dappMemberService.updateById(fromMember);
+            } else {
+                fromMember = dappMemberService.insertMember(e.from, null);
             }
 
-            // 如果转账对象地址不为源池地址,则创建用户
-            if (!e.to.equals(ADDRESS)) {
+            BigInteger tokens = e.tokens;
+            BigDecimal amount = new BigDecimal(tokens.toString()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
+
+            // 如果转账对象地址不为源池地址,注册该用户并绑定上下级关系
+            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());
                 }
+
+//                if (toMember.getActiveStatus() == 2) {
+//                    toMember.setActiveStatus(1);
+//                }
+                // 更新余额
+                BigDecimal balance = ChainService.getInstance(ChainEnum.BSC_TFC.name()).balanceOf(e.to);
+                toMember.setBalance(balance);
+                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(ADDRESS)) {
-                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);
+
                 // 更改状态为已同步
                 fundFlow.setStatus(2);
                 fundFlow.setTargetAmount(transferAmount);
                 dappFundFlowDao.updateById(fundFlow);
 
-                // TODO 划点分配
+                // 划点分配
+                chainProducer.sendDitribProfit(fundFlow.getId());
 
-                // TODO 自动转账
-//                String hash = ChainService.getInstance(ChainEnum.BSC_USDT.name()).transfer(e.from, transferAmount);
-//                fundFlow.setToHash(hash);
-//                dappFundFlowDao.updateById(fundFlow);
+                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());
+                map.put("type", "flow");
+
+                // 发送转账消息
+                chainProducer.sendOnlineTransfer(JSONObject.toJSONString(map));
             }
 
         }

--
Gitblit v1.9.1