From ada1169b0bcabc1eadb621d2215f157234e4031a Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Sat, 11 Jun 2022 22:57:12 +0800
Subject: [PATCH] fix to test

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappSystemServiceImpl.java |  292 +++++++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 275 insertions(+), 17 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappSystemServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappSystemServiceImpl.java
index 7f5491b..08b5435 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappSystemServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappSystemServiceImpl.java
@@ -7,25 +7,31 @@
 import cc.mrbird.febs.dapp.chain.ChainService;
 import cc.mrbird.febs.dapp.chain.ContractChainService;
 import cc.mrbird.febs.dapp.dto.SystemDto;
-import cc.mrbird.febs.dapp.entity.DappMemberEntity;
-import cc.mrbird.febs.dapp.entity.DappMineDataEntity;
-import cc.mrbird.febs.dapp.entity.DappReturnRatioEntity;
-import cc.mrbird.febs.dapp.mapper.DappReturnRatioDao;
-import cc.mrbird.febs.dapp.mapper.DappSystemDao;
+import cc.mrbird.febs.dapp.entity.*;
+import cc.mrbird.febs.dapp.mapper.*;
 import cc.mrbird.febs.dapp.service.DappSystemService;
+import cc.mrbird.febs.dapp.utils.OnlineTransferUtil;
 import cc.mrbird.febs.dapp.vo.RedisTransferPoolVo;
+import cc.mrbird.febs.dapp.vo.SlipSettingVo;
+import cc.mrbird.febs.job.SystemTradeJob;
+import cc.mrbird.febs.rabbit.producer.ChainProducer;
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.date.DateTime;
+import cn.hutool.core.date.DateUnit;
+import cn.hutool.core.date.DateUtil;
 import cn.hutool.core.util.RandomUtil;
 import cn.hutool.core.util.StrUtil;
+import com.alibaba.fastjson.JSONObject;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
 import java.math.BigDecimal;
+import java.math.BigInteger;
 import java.math.RoundingMode;
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 /**
  * @author 
@@ -39,17 +45,23 @@
     private final DappSystemDao dappSystemDao;
     private final DappReturnRatioDao dappReturnRatioDao;
     private final RedisUtils redisUtils;
+    private final DappFundFlowDao dappFundFlowDao;
+    private final DappMemberDao dappMemberDao;
+    private final DataDictionaryCustomMapper dataDictionaryCustomMapper;
+    private final DappOnlineTransferDao dappOnlineTransferDao;
+    private final ChainProducer chainProducer;
 
-    private final String[] ADDRESS_PREFIX = {"T", "0x"};
+
     @Override
     public Map<String, Object> findTotalInComeAndList() {
-        DappMineDataEntity dataMine = dappSystemDao.selectMineDataForOne();
-
+        List<DappMemberEntity> addresses = dappMemberDao.selectList(null);
+        DappMemberEntity mineAddress = dappMemberDao.selectByAddress(ChainEnum.BSC_TFC.getAddress(), null);
+        BigDecimal coinCnt = dappMemberDao.selectAllBalance("normal");
 
         Map<String, Object> result = new HashMap<>();
-        result.put("total", 6000);
-        result.put("addreessCnt", 1234);
-        result.put("coinCnt", 4000);
+        result.put("total", mineAddress.getBalance());
+        result.put("addreessCnt", addresses.size());
+        result.put("coinCnt", coinCnt);
         return result;
     }
 
@@ -71,6 +83,27 @@
         DappMemberEntity member = LoginUserUtil.getAppUser();
 
         SystemDto system = new SystemDto();
+        String hasStart = redisUtils.getString(AppContants.SYSTEM_START_FLAG);
+        // 启动未启动
+        if (!"start".equals(hasStart)) {
+            system.setFeeRatio(BigDecimal.TEN);
+            system.setUsdtRemain(BigDecimal.ZERO);
+            system.setUsdtTotal(BigDecimal.ZERO);
+            system.setSaleRemain(BigDecimal.ZERO);
+            system.setSaleTotal(BigDecimal.ZERO);
+            Object makePoolObj = redisUtils.get(AppContants.REDIS_KEY_MAKE_POOL_CNT);
+            BigDecimal balance;
+            if (makePoolObj == null) {
+                balance = ChainService.getInstance(ChainEnum.BSC_TFC_MAKE.name()).balanceOf(ChainEnum.BSC_TFC_MAKE.getAddress());
+                redisUtils.set(AppContants.REDIS_KEY_MAKE_POOL_CNT, balance);
+            } else {
+                balance = (BigDecimal) makePoolObj;
+            }
+            system.setBuyRemain(balance);
+            system.setBuyTotal(balance);
+
+            return system;
+        }
 
         RedisTransferPoolVo transferPool = (RedisTransferPoolVo) redisUtils.get(AppContants.REDIS_KEY_TRANSFER_POOL_VOL);
         BigDecimal poolRemain = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN);
@@ -80,13 +113,17 @@
 
         ContractChainService instance = ChainService.getInstance(ChainEnum.BSC_TFC.name());
         BigDecimal balance = instance.balanceOf(member.getAddress());
-        Object o = redisUtils.get(AppContants.REDIS_KEY_COIN_REMAIN);
+
+        Object o = redisUtils.get(AppContants.REDIS_KEY_COIN_REMAIN + member.getAddress());
 
         BigDecimal coinTotal = balance.multiply(BigDecimal.valueOf(0.3)).setScale(instance.decimals(), RoundingMode.HALF_DOWN);
         BigDecimal remain;
         if (o == null) {
+            DateTime tomorrow = DateUtil.beginOfDay(DateUtil.tomorrow());
+            long time = DateUtil.between(new Date(), tomorrow, DateUnit.SECOND, true);
+
             remain = coinTotal;
-            redisUtils.set(AppContants.REDIS_KEY_COIN_REMAIN, remain);
+            redisUtils.set(AppContants.REDIS_KEY_COIN_REMAIN + member.getAddress(), remain, time);
         } else {
             remain = (BigDecimal) o;
         }
@@ -104,4 +141,225 @@
         system.setFeeRatio(BigDecimal.TEN);
         return system;
     }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public void tradeProfitDistribute(Long id) {
+        DappFundFlowEntity fundflow = dappFundFlowDao.selectById(id);
+        String batchNo = RandomUtil.randomString(16) + id.toString();
+
+        BigDecimal fee = fundflow.getFee();
+        DappMemberEntity member = dappMemberDao.selectById(fundflow.getMemberId());
+
+        if (StrUtil.isBlank(member.getRefererId()) || "0".equals(member.getRefererId())) {
+            return;
+        }
+        List<String> inviteIds = StrUtil.split(member.getRefererId(), ',');
+        List<DappMemberEntity> parents = dappMemberDao.selectParentsList(inviteIds, 6);
+
+        DataDictionaryCustom miniHoldCoin = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.DIC_TYPE_SYSTEM_SETTING, AppContants.DIC_VALUE_MINI_HOLD_COIN_LIMIT);
+        DataDictionaryCustom slipPointSetting = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.DIC_TYPE_SLIP_POINT_SETTING, AppContants.DIC_TYPE_SLIP_POINT_SETTING);
+        SlipSettingVo slipSetting = JSONObject.parseObject(slipPointSetting.getValue(), SlipSettingVo.class);
+
+        // 源池
+        ContractChainService instance = ChainService.getInstance(ChainEnum.BSC_TFC_SOURCE.name());
+        BigDecimal destroy = slipSetting.getDestroyPoint().divide(slipSetting.getAllPoint(), 2, RoundingMode.HALF_DOWN).multiply(fee);
+        log.info("销毁数量:{}, 比例:{}", destroy, slipSetting.getDestroyPoint());
+
+        // 销毁
+        OnlineTransferUtil.addTransfer(AppContants.DESTROY_ADDRESS, destroy, fundflow.getType(), 3, ChainEnum.BSC_TFC_SOURCE.name(), AppContants.SYMBOL_COIN, batchNo);
+
+        // 总分发金额
+        BigDecimal distrbAmount = fee.subtract(destroy);
+        // 技术金额
+        BigDecimal techAmount = distrbAmount.multiply(slipSetting.getTechProp().divide(BigDecimal.valueOf(100), 2, RoundingMode.HALF_UP));
+        // 节点金额
+        BigDecimal nodeAmount = distrbAmount.subtract(techAmount);
+        log.info("总分发金额:{}, 技术:{}, 节点:{}", distrbAmount, techAmount, nodeAmount);
+
+        if (CollUtil.isNotEmpty(parents)) {
+            int i = 1;
+            for (DappMemberEntity parent : parents) {
+                BigDecimal balance = instance.balanceOf(parent.getAddress());
+                log.info("地址:{}, 余额:{}", parent.getAddress(), balance);
+
+                // 若地址持币少了指定数量
+                if (balance.compareTo(new BigDecimal(miniHoldCoin.getValue())) < 0) {
+                    continue;
+                }
+
+                DataDictionaryCustom distribDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.DIC_TYPE_DISTRIBUTE_PROP, String.valueOf(i));
+                if (distribDic == null) {
+                    break;
+                }
+
+                BigDecimal total = distrbAmount.multiply(new BigDecimal(distribDic.getValue()).divide(BigDecimal.valueOf(100), 2, RoundingMode.HALF_UP));
+
+                OnlineTransferUtil.addTransfer(parent.getAddress(), total, fundflow.getType(), 1, ChainEnum.BSC_TFC_SOURCE.name(), AppContants.SYMBOL_COIN, batchNo);
+
+                DappFundFlowEntity distribFlow = new DappFundFlowEntity(parent.getId(), total, 4, 2, BigDecimal.ZERO);
+                dappFundFlowDao.insert(distribFlow);
+
+                nodeAmount = nodeAmount.subtract(total);
+                i++;
+            }
+        }
+
+        // 若节点金额还有剩余,则进入技术金额
+        techAmount = techAmount.add(nodeAmount);
+        OnlineTransferUtil.addTransfer(ChainEnum.BSC_TFC_TECH.getAddress(), techAmount, fundflow.getType(), 2, ChainEnum.BSC_TFC_SOURCE.name(), AppContants.SYMBOL_COIN, batchNo);
+
+        Map<String, String> map = new HashMap<>();
+        map.put("batchNo", batchNo);
+        map.put("type", "batch");
+
+        // 发送转账消息
+        chainProducer.sendOnlineTransfer(JSONObject.toJSONString(map));
+    }
+
+    @Override
+    public void mining() {
+        ContractChainService contract = ChainService.getInstance(ChainEnum.BSC_TFC.name());
+        // 发行量
+        BigInteger totalSupply = contract.totalSupply();
+
+        BigDecimal totalUnHold = contract.balanceOf(AppContants.DESTROY_ADDRESS);
+        List<DappMemberEntity> admin = dappMemberDao.selectMemberByAccountType("admin");
+        for (DappMemberEntity member : admin) {
+            totalUnHold = totalUnHold.add(contract.balanceOf(member.getAddress()));
+        }
+
+        Object transferPoolObj = redisUtils.get(AppContants.REDIS_KEY_MINE_TRANSFER_POOL_TRADE_CNT);
+
+        // 全网持币量
+        BigDecimal allInternet = new BigDecimal(totalSupply).subtract(totalUnHold);
+        // 中转池交易量
+        BigDecimal transferPoll = (BigDecimal) transferPoolObj;
+
+        List<DappMemberEntity> allMembers = dappMemberDao.selectList(null);
+        if (CollUtil.isEmpty(allMembers)) {
+            return;
+        }
+        DataDictionaryCustom miniHoldCoin = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.DIC_TYPE_SYSTEM_SETTING, AppContants.DIC_VALUE_MINI_HOLD_COIN_LIMIT);
+
+        String batchNo = RandomUtil.randomString(32);
+        ContractChainService instance = ChainService.getInstance(ChainEnum.BSC_TFC.name());
+        for (DappMemberEntity member : allMembers) {
+            if ("admin".equals(member.getAccountType())) {
+                continue;
+            }
+
+            BigDecimal balance = instance.balanceOf(member.getAddress());
+            if (balance.compareTo(new BigDecimal(miniHoldCoin.getValue())) < 1) {
+                continue;
+            }
+
+            // 挖矿数量
+            BigDecimal mine = transferPoll.multiply(balance.divide(allInternet, instance.decimals(), RoundingMode.HALF_UP));
+            if (mine.compareTo(BigDecimal.ZERO) < 1) {
+                continue;
+            }
+
+            DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), mine, 3, 2, BigDecimal.ZERO);
+            dappFundFlowDao.insert(fundFlow);
+            OnlineTransferUtil.addTransfer(member.getAddress(), mine, 3, 1, ChainEnum.BSC_TFC.name(), AppContants.SYMBOL_COIN, batchNo);
+
+            member.setBalance(balance);
+            dappMemberDao.updateById(member);
+        }
+
+        Map<String, String> map = new HashMap<>();
+        map.put("batchNo", batchNo);
+        map.put("type", "batch");
+
+        // 发送转账消息
+        chainProducer.sendOnlineTransfer(JSONObject.toJSONString(map));
+    }
+
+    @Override
+    public void onlineTransfer(String str) {
+        JSONObject jsonObject = JSONObject.parseObject(str);
+        String batchNo = jsonObject.getString("batchNo");
+        List<DappOnlineTransferEntity> transferList = dappOnlineTransferDao.selectByBatchNo(batchNo);
+
+        if (CollUtil.isEmpty(transferList)) {
+            return;
+        }
+
+        for (DappOnlineTransferEntity transfer : transferList) {
+            if (transfer.getHasFinish() == 1) {
+                continue;
+            }
+
+            String txHash = ChainService.getInstance(transfer.getFromType()).transfer(transfer.getAddress(), transfer.getAmount());
+            while (StrUtil.isEmpty(txHash)) {
+                // 休眠2s
+                try {
+                    Thread.sleep(2000);
+                } catch (InterruptedException e) {
+                    e.printStackTrace();
+                }
+                txHash = ChainService.getInstance(transfer.getFromType()).transfer(transfer.getAddress(), transfer.getAmount());
+            }
+
+            transfer.setTxHash(txHash);
+            transfer.setHasFinish(1);
+            transfer.setUpdateTime(new Date());
+            dappOnlineTransferDao.updateById(transfer);
+
+            if ("flow".equals(jsonObject.getString("type"))) {
+                DappFundFlowEntity fundFlow = dappFundFlowDao.selectById(Long.parseLong(batchNo));
+                fundFlow.setToHash(txHash);
+                dappFundFlowDao.updateById(fundFlow);
+            }
+        }
+    }
+
+    @Override
+    public void userBuyReward(Long id) {
+        DappFundFlowEntity fundFlow = dappFundFlowDao.selectById(id);
+
+        DappMemberEntity member = dappMemberDao.selectById(fundFlow.getMemberId());
+        DappMemberEntity referer = dappMemberDao.selectMemberInfoByInviteId(member.getRefererId());
+        if (referer == null || "admin".equals(referer.getAccountType())) {
+            return;
+        }
+
+        DataDictionaryCustom dic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.DIC_TYPE_SYSTEM_SETTING, AppContants.DIC_VALUE_USER_BUY_REWARD);
+        BigDecimal prop = new BigDecimal(dic.getValue()).divide(new BigDecimal("100"), 2, RoundingMode.HALF_UP);
+        BigDecimal reward = fundFlow.getAmount().multiply(prop);
+
+        BigDecimal balance = ChainService.getInstance(ChainEnum.BSC_TFC_REWARD.name()).balanceOf(ChainEnum.BSC_TFC_REWARD.getAddress());
+        if (balance.compareTo(reward) < 1) {
+            return;
+        }
+
+        DappFundFlowEntity rewardFlow = new DappFundFlowEntity(referer.getId(), reward, 4, 2, BigDecimal.ZERO);
+        dappFundFlowDao.insert(rewardFlow);
+
+        OnlineTransferUtil.addTransfer(referer.getAddress(), reward, 4, 1, ChainEnum.BSC_TFC_REWARD.name(), AppContants.SYMBOL_COIN, rewardFlow.getId().toString());
+
+        Map<String, String> map = new HashMap<>();
+        map.put("batchNo", rewardFlow.getId().toString());
+        map.put("type", "batch");
+        chainProducer.sendOnlineTransfer(JSONObject.toJSONString(map));
+    }
+
+    @Autowired
+    private SystemTradeJob systemTradeJob;
+
+    @Override
+    public void startSystem() {
+        log.info("启动系统");
+
+        Thread thread = new Thread(new Runnable() {
+            @Override
+            public void run() {
+                systemTradeJob.transferPoolVol();
+                systemTradeJob.sourcePoolUsdtOutLimit();
+            }
+        });
+
+        thread.start();
+    }
 }

--
Gitblit v1.9.1