From 92dd39c0ecc2f5ecd92b730aad036004576d5dfc Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Thu, 09 May 2024 17:30:08 +0800
Subject: [PATCH] 合约监听

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/BscCoinContractEvent.java |  245 ++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 200 insertions(+), 45 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 c8ba7de..3900f4a 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,90 +1,245 @@
 package cc.mrbird.febs.dapp.service.impl;
 
+import cc.mrbird.febs.common.contants.AppContants;
+import cc.mrbird.febs.common.enumerates.FlowTypeEnum;
+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.entity.DappMemberEntity;
+import cc.mrbird.febs.dapp.entity.*;
+import cc.mrbird.febs.dapp.mapper.DappFundFlowDao;
 import cc.mrbird.febs.dapp.mapper.DappMemberDao;
+import cc.mrbird.febs.dapp.mapper.DappMemberFundMapper;
+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.ObjectUtil;
 import cn.hutool.core.util.StrUtil;
+import com.alibaba.fastjson.JSONObject;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+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 {
 
-    @Autowired
+    @Resource
     private DappMemberDao dappMemberDao;
+    @Resource
+    private DappMemberFundMapper dappMemberFundMapper;
 
-    private final String ADDRESS = "0x977a9ddfb965a9a3416fa72ca7f91c4949c18f25";
-    int decimals = ChainService.getInstance(ChainEnum.BSC_TFC.name()).decimals();
+    @Resource
+    private DappMemberService dappMemberService;
 
+    @Resource
+    private DappFundFlowDao dappFundFlowDao;
+
+    @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) {
         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", "BSC", "admin");
+                }
                 return;
             }
 
-            DappMemberEntity fromMember = dappMemberDao.selectByAddress(e.from, null);
-            if (fromMember == null) {
-                fromMember = insertMember(e.from, null);
+            // 更新余额
+            DappMemberEntity fromMember = dappMemberService.findByAddress(e.from, null);
+            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 = insertMember(e.to, fromMember.getInviteId());
+                    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;
+            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);
+
+                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));
             }
 
         }
     }
 
-    private DappMemberEntity insertMember(String address, String refererId) {
-        DappMemberEntity member = new DappMemberEntity();
-        member.setAddress(address);
-
-        dappMemberDao.insert(member);
-
-        member.setInviteId(ShareCodeUtil.toSerialCode(member.getId()));
-        member.setRefererId(refererId);
-        if (StrUtil.isNotBlank(refererId)){
-            boolean flag = false;
-            String parentId = refererId;
-            String ids = "";
-            while (!flag && StringUtils.isNotBlank(parentId)) {
-                if (StrUtil.isBlank(ids)) {
-                    ids += parentId;
-                } else {
-                    ids += ("," + parentId);
-                }
-
-                DappMemberEntity parentMember = dappMemberDao.selectMemberInfoByInviteId(parentId);
-                if (parentMember == null) {
-                    break;
-                }
-                parentId = parentMember.getRefererId();
-                if(StringUtils.isBlank(parentId)){
-                    break;
-                }
-                if (parentMember.getRefererId().equals(parentMember.getInviteId())) {
-                    flag = true;
-                }
-            }
-            member.setRefererIds(ids);
+    @Override
+    public void coinReward(EthUsdtContract.CoinRewardEventResponse e) {
+        if (e.adr == null) {
+            return;
         }
-        dappMemberDao.updateById(member);
+        /**
+         * 根据地址去查询数据库是否已经注册
+         * 生成对应的产矿记录
+         * 生成对应的产矿流水
+         */
+        String address = e.adr;
+        String blockHash = e.blockHash;
+        DappMemberEntity dappMemberEntity = dappMemberDao.selectByAddress(address, null);
+        if(ObjectUtil.isEmpty(dappMemberEntity)){
+            return;
+        }
+        int decimals = ChainService.getInstance(ChainEnum.BSC_GFA.name()).decimals();
+        BigDecimal amount = new BigDecimal(e.amount.toString()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
+        BigDecimal price = new BigDecimal(e.price.toString()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
+        BigDecimal usdtTarget = new BigDecimal(e.usdtTarget.toString()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
+        BigDecimal coinTarget = new BigDecimal(e.coinTarget.toString()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
+        BigDecimal usdtCoin = new BigDecimal(e.usdtCoin.toString()).divide(BigDecimal.TEN.pow(decimals), decimals, RoundingMode.HALF_DOWN);
+        BigInteger lastMineTime = e.lastMineTime;
 
-        return member;
+        QueryWrapper<DappMemberFund> objectQueryWrapper = new QueryWrapper<>();
+        objectQueryWrapper.eq("block_hash",blockHash);
+        objectQueryWrapper.eq("address",address);
+        objectQueryWrapper.eq("price",price);
+        objectQueryWrapper.eq("usdt_target",usdtTarget);
+        objectQueryWrapper.eq("coin_target",coinTarget);
+        List<DappMemberFund> dappMemberFunds = dappMemberFundMapper.selectList(objectQueryWrapper);
+        if(CollUtil.isEmpty(dappMemberFunds)){
+            Long memberId = dappMemberEntity.getId();
+            DappMemberFund dappMemberFund = new DappMemberFund();
+            dappMemberFund.setMemberId(memberId);
+            dappMemberFund.setAddress(address);
+            dappMemberFund.setBlockHash(blockHash);
+            dappMemberFund.setAmount(amount);
+            dappMemberFund.setPrice(price);
+            dappMemberFund.setUsdtTarget(usdtTarget);
+            dappMemberFund.setCoinTarget(coinTarget);
+            dappMemberFund.setUsdtCoin(usdtCoin);
+            dappMemberFund.setLastMineTime(lastMineTime.toString());
+            dappMemberFundMapper.insert(dappMemberFund);
+
+            DappFundFlowEntity fundFlowTuiJian1 = new DappFundFlowEntity(
+                    memberId,
+                    usdtTarget,
+                    FlowTypeEnum.ZHIYA_CHA_CHU.getValue(),
+                    DappFundFlowEntity.WITHDRAW_STATUS_AGREE,
+                    StrUtil.format("金本位{},价格{}。",usdtTarget,price));
+
+            dappFundFlowDao.insert(fundFlowTuiJian1);
+
+            DappFundFlowEntity fundFlowTuiJian2 = new DappFundFlowEntity(
+                    memberId,
+                    coinTarget,
+                    FlowTypeEnum.ZHIYA_CHA_CHU.getValue(),
+                    DappFundFlowEntity.WITHDRAW_STATUS_AGREE,
+                    StrUtil.format("币本位{},价格{}。",coinTarget,price));
+            dappFundFlowDao.insert(fundFlowTuiJian2);
+            log.info("本地交易:{},amount-{},price-{}", e.adr,amount,price);
+        }
     }
 }

--
Gitblit v1.9.1