From 2ccbfa3ef257c53f9fbc80ab7e7459b09bbfa3d3 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 30 May 2022 18:30:51 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/BscUsdtContractEvent.java |   61 +++++++++++++++++++++---------
 1 files changed, 43 insertions(+), 18 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..3c5f2c5 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,10 +1,9 @@
 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.chain.*;
 import cc.mrbird.febs.dapp.entity.DappFundFlowEntity;
 import cc.mrbird.febs.dapp.entity.DappMemberEntity;
 import cc.mrbird.febs.dapp.mapper.DappFundFlowDao;
@@ -26,37 +25,63 @@
 public class BscUsdtContractEvent implements ContractEventService {
 
     @Resource
+    private RedisUtils redisUtils;
+
+    @Resource
     private DappMemberService dappMemberService;
 
     @Resource
     private DappFundFlowDao dappFundFlowDao;
 
-    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)) {
+        ContractChainService sourceUsdtInstance = ChainService.getInstance(ChainEnum.BSC_USDT.name());
+        int decimals = sourceUsdtInstance.decimals();
+        int tfcDecimals = ChainService.getInstance(ChainEnum.BSC_TFC.name()).decimals();
+
+        // 判断对方打款地址是否为源池地址
+        if (ChainEnum.BSC_TFC_SOURCE.getAddress().equals(e.to)) {
             DappMemberEntity fromMember = dappMemberService.findByAddress(e.from, null);
+            // 如果此时fromMember为null,则说明该用户未经过转账绑定关系,而是直接注册并购买币,则将关系绑定到顶级账户
             if (fromMember == null) {
-                fromMember = dappMemberService.insertMember(e.from, null);
+                DappMemberEntity toAddress = dappMemberService.findByAddress(e.to, null);
+                fromMember = dappMemberService.insertMember(e.from, toAddress.getInviteId());
             }
 
+            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);
+            DappFundFlowEntity fundFlow = dappFundFlowDao.selectByFromHash(e.log.getTransactionHash(), 1);
 
-            DappFundFlowEntity fundFlow = new DappFundFlowEntity(fromMember.getId(), amount, 2, null, null, e.log.getTransactionHash());
-            dappFundFlowDao.insert(fundFlow);
+            BigDecimal newPrice = fundFlow.getNewestPrice();
+            BigDecimal transferAmount = amount.divide(newPrice, tfcDecimals, RoundingMode.HALF_DOWN);
 
-            // TODO price
-            BigDecimal newPrice = BigDecimal.valueOf(1);
-
-            BigDecimal transferAmount = amount.multiply(newPrice);
-            String hash = ChainService.getInstance(ChainEnum.BSC_TFC.name()).transfer(e.from, transferAmount);
-            fundFlow.setToHash(hash);
+            // 更改状态为已同步
+            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");
+            }
         }
     }
+
+    public static void main(String[] args) {
+        System.out.println(ChainService.getInstance(ChainEnum.BSC_USDT.name()).balanceOf("0x9DDE1834683D642D4D077498DC7fbdb8CF70E8FE"));
+    }
 }

--
Gitblit v1.9.1