From 1041f9d218f170626a2af49fcaeb812b67ab31b8 Mon Sep 17 00:00:00 2001
From: xiaoyong931011 <15274802129@163.com>
Date: Thu, 01 Dec 2022 15:56:48 +0800
Subject: [PATCH] 20221130

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java |  256 +++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 239 insertions(+), 17 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 76aa551..f250aff 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
@@ -12,6 +12,7 @@
 import cc.mrbird.febs.dapp.chain.ContractChainService;
 import cc.mrbird.febs.dapp.dto.*;
 import cc.mrbird.febs.dapp.entity.*;
+import cc.mrbird.febs.dapp.enumerate.DataDictionaryEnum;
 import cc.mrbird.febs.dapp.mapper.*;
 import cc.mrbird.febs.dapp.service.DappWalletService;
 import cc.mrbird.febs.dapp.utils.BoxUtil;
@@ -22,6 +23,8 @@
 import cn.hutool.core.date.DateField;
 import cn.hutool.core.date.DateUnit;
 import cn.hutool.core.date.DateUtil;
+import cn.hutool.core.util.ObjectUtil;
+import cn.hutool.core.util.StrUtil;
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.Wrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
@@ -54,8 +57,11 @@
     private final RedisUtils redisUtils;
     private final DataDictionaryCustomMapper dataDictionaryCustomMapper;
     private final DappNftActivationDao dappNftActivationDao;
+    private final MemberCoinWithdrawDao memberCoinWithdrawDao;
 
     private final ChainProducer chainProducer;
+    private final DappSystemDao dappSystemDao;
+    private final DappSystemProfitDao dappSystemProfitDao;
 
     @Override
     public WalletInfoVo walletInfo() {
@@ -63,10 +69,11 @@
 
         DappMemberEntity memberInfo = dappMemberDao.selectById(member.getId());
         WalletInfoVo walletInfo = new WalletInfoVo();
-        List<DappMemberEntity> direct = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 1);
+        List<DappMemberEntity> direct = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 1, 1);
         DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
         DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId());
 
+        DappMineDataEntity mineData = dappSystemDao.selectMineDataForOne();
         walletInfo.setDirectCnt(direct.size());
         walletInfo.setInviteId(member.getActiveStatus() == 1 ? member.getInviteId() : "-");
         walletInfo.setBalance(walletCoin.getAvailableAmount());
@@ -74,7 +81,17 @@
         walletInfo.setOutCnt(memberInfo.getOutCnt());
         walletInfo.setProfit(dappFundFlowDao.selectProfitAmountByMemberId(member.getId()));
         walletInfo.setTfcBalance(walletMine.getAvailableAmount());
+        walletInfo.setSafePool(mineData.getSafePool());
 
+        walletInfo.setAccountType(memberInfo.getAccountType());
+        DappSystemProfit dappSystemProfit = dappSystemProfitDao.selectByMemberIdAndState(memberInfo.getId(),DappSystemProfit.STATE_IN);
+        walletInfo.setSystemProfitId(dappSystemProfit.getId());
+        BigDecimal directProfit = dappFundFlowDao.selectSumAmountByMemberIdAndTypeAndStatus(memberInfo.getId(),3,2);
+        walletInfo.setDirectProfit(directProfit);
+        BigDecimal levelProfit = dappFundFlowDao.selectSumAmountByMemberIdAndTypeAndStatus(memberInfo.getId(),4,2);
+        walletInfo.setLevelProfit(levelProfit);
+        BigDecimal luckyProfit = dappFundFlowDao.selectSumAmountByMemberIdAndTypeAndStatus(memberInfo.getId(),7,2);
+        walletInfo.setLuckyProfit(luckyProfit);
         return walletInfo;
     }
 
@@ -88,7 +105,7 @@
             dappFundFlowEntity.setType(recordInPageDto.getType());
         }
         dappFundFlowEntity.setMemberId(member.getId());
-        dappFundFlowEntity.setStatus(2);
+//        dappFundFlowEntity.setStatus(2);
 
         IPage<DappFundFlowEntity> records = dappFundFlowDao.selectInPage(page, dappFundFlowEntity);
         return records.getRecords();
@@ -118,9 +135,20 @@
         return dappAccountMoneyChangeDao.selectInPage(change, page);
     }
 
+    /**
+     * // 第一次{amount: val, fee: 0, txHash: '', type: 1, buyType: 2}
+     *       // 成功{type: 1, txHash: result.transactionHash, id: res.data, flag: 'success', buyType: 2}
+     *       // 失败{type: 1, id: res.data, flag: 'fail', buyType: 2}
+     */
     @Override
     public Long transfer(TransferDto transferDto) {
         DappMemberEntity member = LoginUserUtil.getAppUser();
+//        DappMemberEntity member = dappMemberDao.selectById(transferDto.getMemberId());
+
+        String upgrade = redisUtils.getString("APP_UPGRADE");
+        if ("upgrade".equals(upgrade)) {
+            throw new FebsException("功能升级中");
+        }
 
         if (transferDto.getType() != 2) {
             member = dappMemberDao.selectById(member.getId());
@@ -135,7 +163,7 @@
                 throw new FebsException("Balance Not Enough");
             }
 
-            updateWalletCoinWithLock(transferDto.getAmount().negate(), member.getId());
+            updateWalletCoinWithLock(transferDto.getAmount(), member.getId(), 2);
 
             DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), transferDto.getAmount().negate(), 1, 2, transferDto.getFee(), transferDto.getTxHash());
             dappFundFlowDao.insert(fundFlow);
@@ -155,8 +183,52 @@
 
             if ("success".equals(transferDto.getFlag())) {
                 DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId());
+                if(DappFundFlowEntity.WITHDRAW_STATUS_AGREE == flow.getStatus()){
+                    return null;
+                }
+                //是否已经加入动能
+                DappSystemProfit dappSystemProfitIng = dappSystemProfitDao.selectByMemberIdAndState(member.getId(), DappSystemProfit.STATE_IN);
+                if(ObjectUtil.isNotEmpty(dappSystemProfitIng)){
+                    return null;
+                }
+
+                //插入一条会员入列记录,即加入动能队列
+                DappSystemProfit dappSystemProfit = new DappSystemProfit(member.getId(), transferDto.getAmount());
+                dappSystemProfitDao.insert(dappSystemProfit);
+
                 flow.setFromHash(transferDto.getTxHash());
+                flow.setSystemProfitId(dappSystemProfit.getId());
+                flow.setStatus(DappFundFlowEntity.WITHDRAW_STATUS_AGREE);
                 dappFundFlowDao.updateById(flow);
+                //升级成为Agent
+                dappMemberDao.updateMemberAccountType(DataDictionaryEnum.AGENT.getCode(),member.getId());
+                dappMemberDao.updateMemberActiveStatus(1,member.getId());
+
+                //直接拿走0.05个BNB放入技术方
+                DataDictionaryCustom systemProfit = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(DataDictionaryEnum.SYSTEM_PROFIT.getType(), DataDictionaryEnum.SYSTEM_PROFIT.getCode());
+                String systemProfitStr = StrUtil.isEmpty(systemProfit.getValue()) ? "0.05" : systemProfit.getValue();
+                DappFundFlowEntity systemProfitFlow = new DappFundFlowEntity(1L, new BigDecimal(systemProfitStr), 2, 1, BigDecimal.ZERO, transferDto.getTxHash(),dappSystemProfit.getId());
+                dappFundFlowDao.insert(systemProfitFlow);
+                //发送转币消息
+                chainProducer.sendBnbTransferMsg(systemProfitFlow.getId());
+                //直接返利30%给直接上级
+                DappMemberEntity dappMemberEntity = dappMemberDao.selectById(member.getId());
+                String refererId = dappMemberEntity.getRefererId();
+                DappMemberEntity refererMember = dappMemberDao.selectMemberInfoByInviteId(refererId);
+
+                DataDictionaryCustom directProfitSet = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(DataDictionaryEnum.DIRECT_PROFIT.getType(), DataDictionaryEnum.DIRECT_PROFIT.getCode());
+                BigDecimal directProfitStr = new BigDecimal(StrUtil.isEmpty(directProfitSet.getValue()) ? "0.3" : directProfitSet.getValue());
+                BigDecimal subtract = transferDto.getAmount().subtract(new BigDecimal(systemProfitStr));
+                BigDecimal directProfit = subtract.multiply(directProfitStr).setScale(2,BigDecimal.ROUND_DOWN);
+                
+                DappFundFlowEntity fundFlow = new DappFundFlowEntity(refererMember.getId(), directProfit, 3, 1, BigDecimal.ZERO, transferDto.getTxHash(),dappSystemProfit.getId());
+                dappFundFlowDao.insert(fundFlow);
+                //发送转币消息
+                chainProducer.sendBnbTransferMsg(fundFlow.getId());
+                //层级奖励30%
+                chainProducer.sendLevelProfitMsg(dappSystemProfit.getId());
+                //发送一个消息,计算当前是否有人可以出局
+                chainProducer.sendMemberOutMsg(dappSystemProfit.getId());
             } else {
                 DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId());
                 if (flow.getStatus() == 1) {
@@ -169,17 +241,35 @@
 
     @Override
     public BigDecimal calPrice(PriceDto priceDto) {
-        String priceStr = redisUtils.getString(AppContants.REDIS_KEY_TFC_NEW_PRICE);
+//        String priceStr = redisUtils.getString(AppContants.REDIS_KEY_TFC_NEW_PRICE);
+        DataDictionaryCustom symbolPrice = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(DataDictionaryEnum.SYMBOL_PRICE.getType(), DataDictionaryEnum.SYMBOL_PRICE.getCode());
+        DataDictionaryCustom serviceFeeDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(DataDictionaryEnum.WITHDRAW_SERVICE_FEE.getType(), DataDictionaryEnum.WITHDRAW_SERVICE_FEE.getCode());
 
-        return priceDto.getAmount().multiply(new BigDecimal("0.1")).multiply(new BigDecimal(priceStr));
+        BigDecimal amount = priceDto.getAmount();
+        if (priceDto.getAmount() == null) {
+            amount = BigDecimal.ZERO;
+        }
+        return amount.multiply(new BigDecimal(serviceFeeDic.getValue()).divide(BigDecimal.valueOf(100), 8, RoundingMode.HALF_DOWN)).divide(new BigDecimal(symbolPrice.getValue()), 2, RoundingMode.HALF_UP);
     }
 
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public void withdraw(WithdrawDto withdrawDto) {
         DappMemberEntity member = LoginUserUtil.getAppUser();
 
-        updateWalletCoinWithLock(withdrawDto.getAmount().negate(), member.getId());
-        updateWalletMineWithLock(withdrawDto.getFee().negate(), member.getId());
+        DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
+        if (walletCoin.getAvailableAmount().compareTo(withdrawDto.getAmount()) < 0) {
+            throw new FebsException("Balance Not Enough");
+        }
+
+        DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId());
+        if (walletMine.getAvailableAmount().compareTo(withdrawDto.getFee()) < 0) {
+            throw new FebsException("DMD Not Enough");
+        }
+
+        updateWalletCoinWithLock(withdrawDto.getAmount(), member.getId(), 2);
+        updateWalletMineWithLock(withdrawDto.getFee(), member.getId(), 2);
+
 
         DappFundFlowEntity feeFlow = new DappFundFlowEntity(member.getId(), withdrawDto.getFee().negate(), 7, 2, null, null);
         dappFundFlowDao.insert(feeFlow);
@@ -187,24 +277,46 @@
         DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), withdrawDto.getAmount().negate(), 5, 2, withdrawDto.getFee(), null);
         dappFundFlowDao.insert(fundFlow);
 
+        MemberCoinWithdrawEntity memberCoinWithdraw = new MemberCoinWithdrawEntity();
+        memberCoinWithdraw.setMemberId(member.getId());
+        memberCoinWithdraw.setAddress(member.getAddress());
+        memberCoinWithdraw.setAmount(withdrawDto.getAmount());
+        memberCoinWithdraw.setFeeAmount(withdrawDto.getFee());
+        memberCoinWithdraw.setStatus(MemberCoinWithdrawEntity.STATUS_YES);
+        memberCoinWithdraw.setSymbol("USDT");
+        memberCoinWithdraw.setFlowId(fundFlow.getId());
+        memberCoinWithdrawDao.insert(memberCoinWithdraw);
+
         String hash = ChainService.getInstance(ChainEnum.BSC_USDT.name()).transfer(member.getAddress(), withdrawDto.getAmount());
         fundFlow.setToHash(hash);
         dappFundFlowDao.updateById(fundFlow);
 
-        Map<String, Object> map = new HashMap<>();
-        map.put("fee", withdrawDto.getFee());
-        map.put("address", ChainEnum.BSC_TFC_FEE.getAddress());
-        chainProducer.sendTfcFee(JSONObject.toJSONString(map));
+        chainProducer.sendFeeDistributeMsg(fundFlow.getId());
+//        Map<String, Object> map = new HashMap<>();
+//        map.put("fee", withdrawDto.getFee());
+//        map.put("address", ChainEnum.BSC_TFC_FEE.getAddress());
+//        map.put("flow", feeFlow.getId());
+//        chainProducer.sendTfcFee(JSONObject.toJSONString(map));
     }
 
     @Override
-    public void updateWalletCoinWithLock(BigDecimal amount, Long memberId) {
+    public void updateWalletCoinWithLock(BigDecimal amount, Long memberId, int type) {
         boolean isSuccess = false;
         while(!isSuccess) {
             DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(memberId);
 
-            walletCoin.setTotalAmount(walletCoin.getTotalAmount().add(amount));
-            walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().add(amount));
+            // 加
+            if (type == 1) {
+                walletCoin.setTotalAmount(walletCoin.getTotalAmount().add(amount));
+                walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().add(amount));
+                // 减
+            } else {
+                if (amount.compareTo(walletCoin.getAvailableAmount()) > 0) {
+                    throw new FebsException("Balance Not Enough");
+                }
+                walletCoin.setTotalAmount(walletCoin.getTotalAmount().subtract(amount));
+                walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().subtract(amount));
+            }
 
             int i = dappWalletCoinDao.updateWithLock(walletCoin);
             if (i > 0) {
@@ -214,13 +326,21 @@
     }
 
     @Override
-    public void updateWalletMineWithLock(BigDecimal amount, Long memberId) {
+    public void updateWalletMineWithLock(BigDecimal amount, Long memberId, int type) {
         boolean isSuccess = false;
         while(!isSuccess) {
             DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(memberId);
 
-            walletMine.setTotalAmount(walletMine.getTotalAmount().add(amount));
-            walletMine.setAvailableAmount(walletMine.getAvailableAmount().add(amount));
+            if (type == 1) {
+                walletMine.setTotalAmount(walletMine.getTotalAmount().add(amount));
+                walletMine.setAvailableAmount(walletMine.getAvailableAmount().add(amount));
+            } else {
+                if (amount.compareTo(walletMine.getAvailableAmount()) > 0) {
+                    throw new FebsException("DMD Not Enough");
+                }
+                walletMine.setTotalAmount(walletMine.getTotalAmount().subtract(amount));
+                walletMine.setAvailableAmount(walletMine.getAvailableAmount().subtract(amount));
+            }
 
             int i = dappWalletMineDao.updateWithLock(walletMine);
             if (i > 0) {
@@ -228,4 +348,106 @@
             }
         }
     }
+
+    @Override
+    public void addFrozenAmountWithLock(BigDecimal amount, Long memberId) {
+        boolean isSuccess = false;
+        while(!isSuccess) {
+            DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(memberId);
+
+            walletCoin.setTotalAmount(walletCoin.getTotalAmount().add(amount));
+            walletCoin.setFrozenAmount(walletCoin.getFrozenAmount().add(amount));
+
+            int i = dappWalletCoinDao.updateWithLock(walletCoin);
+            if (i > 0) {
+                isSuccess = true;
+
+                DappFundFlowEntity frozenAmount = new DappFundFlowEntity(memberId, amount, 9, 2, null, null);
+                dappFundFlowDao.insert(frozenAmount);
+            }
+        }
+    }
+
+    @Override
+    public void releaseFrozenAmountWithLock(Long memberId) {
+        boolean isSuccess = false;
+        while(!isSuccess) {
+            DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(memberId);
+
+            if (walletCoin.getFrozenAmount().compareTo(BigDecimal.ZERO) <= 0) {
+                return;
+            }
+            BigDecimal frozen = walletCoin.getFrozenAmount();
+
+            walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().add(frozen));
+            walletCoin.setFrozenAmount(walletCoin.getFrozenAmount().subtract(frozen));
+
+            int i = dappWalletCoinDao.updateWithLock(walletCoin);
+            if (i > 0) {
+                isSuccess = true;
+
+                DappFundFlowEntity releaseFrozen = new DappFundFlowEntity(memberId, frozen, 10, 2, null, null);
+                dappFundFlowDao.insert(releaseFrozen);
+
+                DappFundFlowEntity releaseFrozenNegate = new DappFundFlowEntity(memberId, frozen.negate(), 9, 2, null, null);
+                dappFundFlowDao.insert(releaseFrozenNegate);
+            }
+        }
+    }
+
+    @Override
+    public DappWalletCoinEntity findByMemberId(Long memberId) {
+        return dappWalletCoinDao.selectByMemberId(memberId);
+    }
+
+    @Override
+    public void transferAgain(TransferDto transferDto) {
+        Long memberId = transferDto.getMemberId();
+        DappMemberEntity member = dappMemberDao.selectById(memberId);
+        String upgrade = redisUtils.getString("APP_UPGRADE");
+        if ("upgrade".equals(upgrade)) {
+            throw new FebsException("功能升级中");
+        }
+        if ("success".equals(transferDto.getFlag())) {
+            //是否已经加入动能
+            DappSystemProfit dappSystemProfitIng = dappSystemProfitDao.selectByMemberIdAndState(member.getId(), DappSystemProfit.STATE_IN);
+            if(ObjectUtil.isNotEmpty(dappSystemProfitIng)){
+                return;
+            }
+            //插入一条会员入列记录,即加入动能队列
+            DappSystemProfit dappSystemProfit = new DappSystemProfit(member.getId(), transferDto.getAmount());
+            dappSystemProfitDao.insert(dappSystemProfit);
+            DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId());
+            flow.setFromHash(transferDto.getTxHash());
+            flow.setSystemProfitId(dappSystemProfit.getId());
+            flow.setStatus(DappFundFlowEntity.WITHDRAW_STATUS_AGREE);
+            dappFundFlowDao.updateById(flow);
+
+            //直接拿走0.05个BNB放入技术方
+            DataDictionaryCustom systemProfit = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(DataDictionaryEnum.SYSTEM_PROFIT.getType(), DataDictionaryEnum.SYSTEM_PROFIT.getCode());
+            String systemProfitStr = StrUtil.isEmpty(systemProfit.getValue()) ? "0.05" : systemProfit.getValue();
+            DappFundFlowEntity systemProfitFlow = new DappFundFlowEntity(1L, new BigDecimal(systemProfitStr), 2, 1, BigDecimal.ZERO, transferDto.getTxHash(),dappSystemProfit.getId());
+            dappFundFlowDao.insert(systemProfitFlow);
+            //发送转币消息
+            chainProducer.sendBnbTransferMsg(systemProfitFlow.getId());
+            //直接返利30%给直接上级
+            DappMemberEntity dappMemberEntity = dappMemberDao.selectById(member.getId());
+            String refererId = dappMemberEntity.getRefererId();
+            DappMemberEntity refererMember = dappMemberDao.selectMemberInfoByInviteId(refererId);
+
+            DataDictionaryCustom directProfitSet = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(DataDictionaryEnum.DIRECT_PROFIT.getType(), DataDictionaryEnum.DIRECT_PROFIT.getCode());
+            BigDecimal directProfitStr = new BigDecimal(StrUtil.isEmpty(directProfitSet.getValue()) ? "0.3" : directProfitSet.getValue());
+            BigDecimal directProfit = (transferDto.getAmount().subtract(new BigDecimal(systemProfitStr))).multiply(directProfitStr).setScale(2,BigDecimal.ROUND_DOWN);
+
+            DappFundFlowEntity fundFlow = new DappFundFlowEntity(refererMember.getId(), directProfit, 3, 1, BigDecimal.ZERO, transferDto.getTxHash(),dappSystemProfit.getId());
+            dappFundFlowDao.insert(fundFlow);
+            //发送转币消息
+            chainProducer.sendBnbTransferMsg(fundFlow.getId());
+            //层级奖励30%
+            chainProducer.sendLevelProfitMsg(dappSystemProfit.getId());
+            //发送一个消息,计算当前是否有人可以出局
+            chainProducer.sendMemberOutMsg(dappSystemProfit.getId());
+
+        }
+    }
 }

--
Gitblit v1.9.1