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 | 109 +++++++++++++++++++++++++++++++++++++++++++++++------- 1 files changed, 94 insertions(+), 15 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 32c165d..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 @@ -9,10 +9,19 @@ 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; @@ -23,7 +32,11 @@ 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 { @@ -39,7 +52,14 @@ @Resource private RedisUtils redisUtils; - private final String ADDRESS = "0x977a9ddfb965a9a3416fa72ca7f91c4949c18f25"; + @Autowired + private DappSystemService dappSystemService; + + @Resource + private DappOnlineTransferDao dappOnlineTransferDao; + + @Resource + private ChainProducer chainProducer; public static void main(String[] args) { System.out.println(Numeric.toBigInt("0x12e4e85")); @@ -49,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(); // 合约创建时,同时创建地址 @@ -61,39 +89,90 @@ 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