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 |  194 ++++++++++++++++++++++++++++++++++++------------
 1 files changed, 145 insertions(+), 49 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 f235f25..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
@@ -10,19 +10,26 @@
 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.*;
 
@@ -44,16 +51,17 @@
     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;
     }
 
@@ -75,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);
@@ -84,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 + 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 + member.getAddress(), remain);
+            redisUtils.set(AppContants.REDIS_KEY_COIN_REMAIN + member.getAddress(), remain, time);
         } else {
             remain = (BigDecimal) o;
         }
@@ -113,6 +146,7 @@
     @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());
@@ -133,9 +167,7 @@
         log.info("销毁数量:{}, 比例:{}", destroy, slipSetting.getDestroyPoint());
 
         // 销毁
-//        instance.transfer(AppContants.DESTROY_ADDRESS, destroy);
-        DappOnlineTransferEntity destroyRecord = new DappOnlineTransferEntity(AppContants.DESTROY_ADDRESS, destroy, fundflow.getType(), 3, ChainEnum.BSC_TFC_SOURCE.name(), "TFC", fundflow.getId().toString());
-        dappOnlineTransferDao.insert(destroyRecord);
+        OnlineTransferUtil.addTransfer(AppContants.DESTROY_ADDRESS, destroy, fundflow.getType(), 3, ChainEnum.BSC_TFC_SOURCE.name(), AppContants.SYMBOL_COIN, batchNo);
 
         // 总分发金额
         BigDecimal distrbAmount = fee.subtract(destroy);
@@ -145,39 +177,41 @@
         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);
 
-        BigDecimal preNode = slipSetting.getNodeProp().divide(BigDecimal.valueOf(100), 2, RoundingMode.HALF_UP);
-        log.info("每份:{}", preNode);
-        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;
+                }
 
-            // 若地址持币少了指定数量 或者 不为创始节点,则不参与节点滑点
-            if (balance.compareTo(new BigDecimal(miniHoldCoin.getValue())) < 0 || parent.getNodeType() == 2) {
-                return;
+                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++;
             }
-
-            DataDictionaryCustom distribDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.DIC_TYPE_DISTRIBUTE_PROP, String.valueOf(i));
-            BigDecimal total = new BigDecimal(distribDic.getValue()).multiply(preNode);
-
-//            instance.transfer(parent.getAddress(), total);
-            DappOnlineTransferEntity distribRecord = new DappOnlineTransferEntity(parent.getAddress(), total, fundflow.getType(), 1, ChainEnum.BSC_TFC_SOURCE.name(), "TFC", fundflow.getId().toString());
-            dappOnlineTransferDao.insert(distribRecord);
-
-            nodeAmount = nodeAmount.subtract(total);
         }
 
         // 若节点金额还有剩余,则进入技术金额
         techAmount = techAmount.add(nodeAmount);
-//        instance.transfer(ChainEnum.BSC_TFC_TECH.getAddress(), techAmount);
-
-        DappOnlineTransferEntity techRecord = new DappOnlineTransferEntity(ChainEnum.BSC_TFC_TECH.getAddress(), techAmount, fundflow.getType(), 2, ChainEnum.BSC_TFC_SOURCE.name(), "TFC", fundflow.getId().toString());
-        dappOnlineTransferDao.insert(techRecord);
+        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", fundflow.getId().toString());
-        map.put("type", "flow");
+        map.put("batchNo", batchNo);
+        map.put("type", "batch");
 
         // 发送转账消息
         chainProducer.sendOnlineTransfer(JSONObject.toJSONString(map));
@@ -185,16 +219,21 @@
 
     @Override
     public void mining() {
-        log.info("挖矿");
+        ContractChainService contract = ChainService.getInstance(ChainEnum.BSC_TFC.name());
+        // 发行量
+        BigInteger totalSupply = contract.totalSupply();
 
-        Object o = redisUtils.get(AppContants.REDIS_KEY_MINE_ALL_INTERNET_CNT);
-        if (o == null) {
-            return;
+        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 = (BigDecimal) o;
+        // 全网持币量
+        BigDecimal allInternet = new BigDecimal(totalSupply).subtract(totalUnHold);
+        // 中转池交易量
         BigDecimal transferPoll = (BigDecimal) transferPoolObj;
 
         List<DappMemberEntity> allMembers = dappMemberDao.selectList(null);
@@ -221,8 +260,12 @@
                 continue;
             }
 
-            DappOnlineTransferEntity tfc = new DappOnlineTransferEntity(member.getAddress(), mine, 3, 1, "TFC", ChainEnum.BSC_TFC.name(), batchNo);
-            dappOnlineTransferDao.insert(tfc);
+            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<>();
@@ -248,22 +291,75 @@
                 continue;
             }
 
-            // TODO 线上转账,已注释
-//            String txHash = ChainService.getInstance(transfer.getFromType()).transfer(transfer.getAddress(), transfer.getAmount());
-//            transfer.setTxHash(txHash);
+            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.setStatus(3);
-//                fundFlow.setToHash(txHash);
-
+                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