From ae50866f34212c007643c1744f78426676885c08 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Tue, 09 Aug 2022 16:18:08 +0800
Subject: [PATCH] fix:chainEnum and ChainListenerJob

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java |  242 ++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 233 insertions(+), 9 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
index 8e1cfa1..0a44401 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
@@ -1,15 +1,24 @@
 package cc.mrbird.febs.dapp.service.impl;
 
+import cc.mrbird.febs.common.configure.i18n.MessageSourceUtils;
 import cc.mrbird.febs.common.contants.AppContants;
 import cc.mrbird.febs.common.entity.QueryRequest;
 import cc.mrbird.febs.common.exception.FebsException;
+import cc.mrbird.febs.common.utils.FebsUtil;
 import cc.mrbird.febs.common.utils.LoginUserUtil;
 import cc.mrbird.febs.common.utils.RedisUtils;
+import cc.mrbird.febs.dapp.chain.ChainEnum;
+import cc.mrbird.febs.dapp.chain.ChainService;
+import cc.mrbird.febs.dapp.chain.ContractChainService;
+import cc.mrbird.febs.dapp.dto.PriceDto;
 import cc.mrbird.febs.dapp.dto.RecordInPageDto;
+import cc.mrbird.febs.dapp.dto.TransferDto;
 import cc.mrbird.febs.dapp.dto.WalletOperateDto;
 import cc.mrbird.febs.dapp.entity.*;
 import cc.mrbird.febs.dapp.mapper.*;
+import cc.mrbird.febs.dapp.service.DappSystemService;
 import cc.mrbird.febs.dapp.service.DappWalletService;
+import cc.mrbird.febs.dapp.utils.BoxUtil;
 import cc.mrbird.febs.dapp.vo.WalletInfoVo;
 import cn.hutool.core.collection.CollUtil;
 import com.baomidou.mybatisplus.core.metadata.IPage;
@@ -20,10 +29,13 @@
 import org.springframework.transaction.annotation.Transactional;
 
 import java.math.BigDecimal;
+import java.math.RoundingMode;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 /**
- * @author 
+ * @author
  * @date 2022-03-18
  **/
 @Slf4j
@@ -37,18 +49,28 @@
     private final DappFundFlowDao dappFundFlowDao;
     private final DappAccountMoneyChangeDao dappAccountMoneyChangeDao;
     private final RedisUtils redisUtils;
+    private final DataDictionaryCustomMapper dataDictionaryCustomMapper;
+    private final DappSystemService dappSystemService;
 
     @Override
     public WalletInfoVo walletInfo() {
         DappMemberEntity member = LoginUserUtil.getAppUser();
 
-        DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
-        DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId());
-
+        Map<String, BigDecimal> map = dappFundFlowDao.selectAmountTotalByType(member.getId());
         WalletInfoVo walletInfo = new WalletInfoVo();
-        walletInfo.setTotalMine(walletMine.getTotalAmount());
-        walletInfo.setAvailableMine(walletMine.getAvailableAmount());
-        walletInfo.setAvailableWallet(walletCoin.getAvailableAmount());
+        List<DappMemberEntity> direct = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 1);
+        List<DappMemberEntity> notDirect = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 2);
+        BigDecimal childHoldAmount = dappMemberDao.selectChildHoldAmount(member.getInviteId());
+
+        DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
+
+        walletInfo.setTotalChild(notDirect.size());
+        walletInfo.setDirectCnt(direct.size());
+        walletInfo.setTotalChildCoin(childHoldAmount);
+        walletInfo.setTeamReward(map.get("teamReward"));
+        walletInfo.setMiningAmount(map.get("mine"));
+        walletInfo.setInviteId(member.getInviteId());
+        walletInfo.setBoxCnt(walletCoin.getBoxCnt());
         return walletInfo;
     }
 
@@ -112,7 +134,7 @@
         walletCoin.setFrozenAmount(walletCoin.getFrozenAmount().add(walletOperateDto.getAmount()));
         dappWalletCoinDao.updateById(walletCoin);
 
-        DappFundFlowEntity fund = new DappFundFlowEntity(member.getId(), walletOperateDto.getAmount(), AppContants.MONEY_TYPE_WITHDRAWAL, 1, fee);
+        DappFundFlowEntity fund = new DappFundFlowEntity(member.getId(), walletOperateDto.getAmount(), AppContants.MONEY_TYPE_WITHDRAWAL, 1, new BigDecimal(fee));
         dappFundFlowDao.insert(fund);
 
         DappAccountMoneyChangeEntity usdtChange = new DappAccountMoneyChangeEntity(member.getId(), preAmount, walletOperateDto.getAmount(), walletCoin.getAvailableAmount(), "USDT申请提现", AppContants.MONEY_TYPE_WITHDRAWAL);
@@ -123,8 +145,14 @@
     public List<DappFundFlowEntity> recordInPage(RecordInPageDto recordInPageDto) {
         Page<DappFundFlowEntity> page = new Page<>(recordInPageDto.getPageNum(), recordInPageDto.getPageSize());
 
+        DappMemberEntity member = LoginUserUtil.getAppUser();
         DappFundFlowEntity dappFundFlowEntity = new DappFundFlowEntity();
-        dappFundFlowEntity.setType(recordInPageDto.getType());
+        if (recordInPageDto.getType() != null && recordInPageDto.getType() != 0) {
+            dappFundFlowEntity.setType(recordInPageDto.getType());
+        }
+        dappFundFlowEntity.setMemberId(member.getId());
+        dappFundFlowEntity.setStatus(2);
+
         IPage<DappFundFlowEntity> records = dappFundFlowDao.selectInPage(page, dappFundFlowEntity);
         return records.getRecords();
     }
@@ -183,4 +211,200 @@
         Page<DappAccountMoneyChangeEntity> page = new Page<>(request.getPageNum(), request.getPageSize());
         return dappAccountMoneyChangeDao.selectInPage(change, page);
     }
+
+    @Override
+    public Long transfer(TransferDto transferDto) {
+        DappMemberEntity member = LoginUserUtil.getAppUser();
+
+        if (member.getActiveStatus() == 2) {
+            throw new FebsException("请联系邀请人转币激活");
+        }
+
+        // 每日出U剩余量(卖币)
+        BigDecimal usdtRemain = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_USDT_OUT_LIMIT_REMAIN);
+        // 用户24小时可出售量
+        BigDecimal saleCoinRemain = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_COIN_REMAIN + member.getAddress());
+
+        // 用户24小时可购买USDT
+        BigDecimal buyUsdtMax = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_IDO_USDT_MAX_BUY_DAILY + member.getAddress());
+
+        BigDecimal buyCoinRemain = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN);
+        // 铸池中的币的剩余量
+        BigDecimal makeCoinRemain = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_MAKE_POOL_CNT);
+
+        String hasStart = redisUtils.getString(AppContants.SYSTEM_START_FLAG);
+        if (transferDto.getId() == null) {
+            if (DappFundFlowEntity.TYPE_SALE == transferDto.getType()) {
+                if (!"start".equals(hasStart)) {
+                    throw new FebsException(MessageSourceUtils.getString("transfer_msg_001"));
+                }
+
+                if (transferDto.getAmount().multiply(transferDto.getPrice()).compareTo(usdtRemain) > 0) {
+                    throw new FebsException(MessageSourceUtils.getString("transfer_msg_002"));
+                }
+
+                if (transferDto.getAmount().compareTo(saleCoinRemain) > 0) {
+                    throw new FebsException(MessageSourceUtils.getString("transfer_msg_003"));
+                }
+
+                usdtRemain = usdtRemain.subtract(transferDto.getAmount().multiply(transferDto.getPrice()));
+                saleCoinRemain = saleCoinRemain.subtract(transferDto.getAmount());
+
+                // 修改当日U剩余量
+                redisUtils.set(AppContants.REDIS_KEY_USDT_OUT_LIMIT_REMAIN, usdtRemain);
+                // 修改用户24小时可售量
+                redisUtils.set(AppContants.REDIS_KEY_COIN_REMAIN + member.getAddress(), saleCoinRemain);
+            } else if (DappFundFlowEntity.TYPE_BUY == transferDto.getType()) {
+                // 购买时,前端传来的amount是USDT,卖出amount是TFC
+                BigDecimal usdtAmount = transferDto.getAmount();
+                BigDecimal coinAmount = transferDto.getAmount().divide(transferDto.getPrice(), 6, RoundingMode.HALF_UP);
+                transferDto.setAmount(coinAmount);
+
+                if ("start".equals(hasStart)) {
+                    if (transferDto.getAmount().compareTo(buyCoinRemain) > 0) {
+                        throw new FebsException(MessageSourceUtils.getString("transfer_msg_004"));
+                    }
+                    buyCoinRemain = buyCoinRemain.subtract(transferDto.getAmount());
+
+                    // 修改当日可购买量
+                    redisUtils.set(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN, buyCoinRemain);
+                    // 如果系统还没有启动,则判断铸池中的剩余量
+                } else {
+                    // 最少购买
+//                    DataDictionaryCustom dic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.DIC_TYPE_SYSTEM_SETTING, AppContants.DIC_VALUE_MAKER_MIN_LIMIT);
+//                    if (transferDto.getAmount().compareTo(new BigDecimal(dic.getValue())) < 0) {
+//                        throw new FebsException(MessageSourceUtils.getString("transfer_msg_005"));
+//                    }
+
+                    if (buyUsdtMax.compareTo(usdtAmount) < 0) {
+                        throw new FebsException(MessageSourceUtils.getString("transfer_msg_007"));
+                    }
+
+                    if (transferDto.getAmount().compareTo(makeCoinRemain) > 0) {
+                        throw new FebsException(MessageSourceUtils.getString("transfer_msg_006"));
+                    }
+                    makeCoinRemain = makeCoinRemain.subtract(transferDto.getAmount());
+                    buyUsdtMax = buyUsdtMax.subtract(usdtAmount);
+
+                    // 修改每日最大购买USDT量
+                    redisUtils.set(AppContants.REDIS_KEY_IDO_USDT_MAX_BUY_DAILY + member.getAddress(), buyUsdtMax);
+
+                    // 修改铸池量
+                    redisUtils.set(AppContants.REDIS_KEY_MAKE_POOL_CNT, makeCoinRemain);
+                }
+            }
+
+            DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), transferDto.getAmount(), transferDto.getType(), 1, transferDto.getFee(), transferDto.getTxHash());
+            fundFlow.setNewestPrice(transferDto.getPrice());
+            dappFundFlowDao.insert(fundFlow);
+            return fundFlow.getId();
+        }
+
+        if ("success".equals(transferDto.getFlag())) {
+            DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId());
+
+            flow.setFromHash(transferDto.getTxHash());
+            dappFundFlowDao.updateById(flow);
+        } else {
+            DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId());
+            if (flow.getStatus() == 1) {
+                if (DappFundFlowEntity.TYPE_BUY == flow.getType()) {
+                    // 购买时,前端传来的amount是USDT,卖出amount是TFC
+                    BigDecimal usdtAmount = transferDto.getAmount();
+                    BigDecimal coinAmount = transferDto.getAmount().divide(flow.getNewestPrice(), 6, RoundingMode.HALF_UP);
+                    transferDto.setAmount(coinAmount);
+
+                    if ("start".equals(hasStart)) {
+                        buyCoinRemain = buyCoinRemain.add(flow.getAmount());
+
+                        // 修改当日可购买量
+                        redisUtils.set(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN, buyCoinRemain);
+                    } else {
+                        makeCoinRemain = makeCoinRemain.add(flow.getAmount());
+                        buyUsdtMax = buyUsdtMax.add(usdtAmount);
+
+                        // 修改铸池量
+                        redisUtils.set(AppContants.REDIS_KEY_MAKE_POOL_CNT, makeCoinRemain);
+                        redisUtils.set(AppContants.REDIS_KEY_IDO_USDT_MAX_BUY_DAILY + member.getAddress(), buyUsdtMax);
+                    }
+                } else {
+                    usdtRemain = usdtRemain.add(transferDto.getAmount().multiply(transferDto.getPrice()));
+                    saleCoinRemain = saleCoinRemain.add(transferDto.getAmount());
+
+                    // 修改当日U剩余量
+                    redisUtils.set(AppContants.REDIS_KEY_USDT_OUT_LIMIT_REMAIN, usdtRemain);
+                    // 修改用户24小时可售量
+                    redisUtils.set(AppContants.REDIS_KEY_COIN_REMAIN + member.getAddress(), saleCoinRemain);
+                }
+                dappFundFlowDao.deleteById(transferDto.getId());
+            }
+        }
+        return null;
+    }
+
+    @Override
+    public Map<String, BigDecimal> calPrice(PriceDto priceDto) {
+        DappMemberEntity member = LoginUserUtil.getAppUser();
+
+        if (!dappSystemService.systemHasStart()) {
+            HashMap<String, BigDecimal> map = new HashMap<>();
+            map.put("x", new BigDecimal("0.05"));
+            map.put("y", new BigDecimal("0.05"));
+            return map;
+        }
+
+        ContractChainService tfcInstance = ChainService.getInstance(ChainEnum.BSC_TFC.name());
+        // u剩余数量
+        BigDecimal sourceU = ChainService.getInstance(ChainEnum.BSC_USDT.name()).balanceOf(ChainEnum.BSC_USDT_SOURCE.getAddress());
+        // 源池代币剩余数量
+        BigDecimal sourceCoin = tfcInstance.balanceOf(ChainEnum.BSC_USDT_SOURCE.getAddress());
+        // 用户卖出数量
+        BigDecimal coin = priceDto.getAmount();
+        BigDecimal x = sourceU.divide(sourceCoin, tfcInstance.decimals(), RoundingMode.HALF_UP);
+        BigDecimal y = sourceU.divide(sourceCoin.add(coin), tfcInstance.decimals(), RoundingMode.HALF_UP);
+
+        log.info("购买价格:{}, 出卖价格:{}", x, y);
+        HashMap<String, BigDecimal> map = new HashMap<>();
+        map.put("x", x);
+        map.put("y", y);
+        return map;
+    }
+
+    @Override
+    public int boxSurprise() {
+
+        // 功能升级
+        throw new FebsException(MessageSourceUtils.getString("box_surprise_002"));
+
+//        DappMemberEntity member = LoginUserUtil.getAppUser();
+//
+//        DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
+//        if (walletCoin.getBoxCnt() < 1) {
+//            throw new FebsException(MessageSourceUtils.getString("box_surprise_001"));
+//        }
+//
+//        walletCoin.setBoxCnt(walletCoin.getBoxCnt() - 1);
+//        dappWalletCoinDao.updateById(walletCoin);
+//
+//        BoxUtil.Box box = BoxUtil.openBox();
+//
+//        DappFundFlowEntity boxFundFlow = new DappFundFlowEntity(member.getId(), new BigDecimal(box.getIndex()), 6, 2, BigDecimal.ZERO);
+//        dappFundFlowDao.insert(boxFundFlow);
+//
+//        new Thread(() -> {
+//            try {
+//                int count = box.getIndex();
+//                while (count > 0) {
+//                    ChainService.getInstance(ChainEnum.BSC_NFT_SDC.name()).safeMintNFT(member.getAddress());
+//                    Thread.sleep(5000);
+//                    count--;
+//                }
+//            } catch (Exception e) {
+//                e.printStackTrace();
+//                log.error("发放卡牌错误", e);
+//            }
+//        }).start();
+//
+//        return box.getIndex();
+    }
 }

--
Gitblit v1.9.1