From e1992f204a135a347a89a5859a8fe54033401b7e Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 30 May 2022 16:34:05 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/BscUsdtContractEvent.java |   47 +++++++++++++++++++++++++++++++++++------------
 1 files changed, 35 insertions(+), 12 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 0d3d668..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,22 +25,31 @@
 public class BscUsdtContractEvent implements ContractEventService {
 
     @Resource
+    private RedisUtils redisUtils;
+
+    @Resource
     private DappMemberService dappMemberService;
 
     @Resource
     private DappFundFlowDao dappFundFlowDao;
 
-    private final String ADDRESS = "0x977a9ddfb965a9a3416fa72ca7f91c4949c18f25";
 
     @Override
     public void compile(EthUsdtContract.TransferEventResponse e) {
-        int decimals = ChainService.getInstance(ChainEnum.BSC_USDT.name()).decimals();
+        ContractChainService sourceUsdtInstance = ChainService.getInstance(ChainEnum.BSC_USDT.name());
+        int decimals = sourceUsdtInstance.decimals();
         int tfcDecimals = ChainService.getInstance(ChainEnum.BSC_TFC.name()).decimals();
-        if (ADDRESS.equals(e.to)) {
+
+        // 判断对方打款地址是否为源池地址
+        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);
@@ -55,10 +63,25 @@
             fundFlow.setTargetAmount(transferAmount);
             dappFundFlowDao.updateById(fundFlow);
 
-//            BigDecimal transferAmount = amount.multiply(newPrice);
-//            String hash = ChainService.getInstance(ChainEnum.BSC_TFC.name()).transfer(e.from, transferAmount);
-//            fundFlow.setToHash(hash);
-//            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