From 73cfce48624bc8558d387ee8b3204cad8d4bbedd Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 13 Jun 2022 18:19:54 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java |   21 ++++++++++++++-------
 1 files changed, 14 insertions(+), 7 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 0dcae82..829727f 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
@@ -9,6 +9,7 @@
 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;
@@ -217,8 +218,8 @@
         // 铸池中的币的剩余量
         BigDecimal makeCoinRemain = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_MAKE_POOL_CNT);
 
+        String hasStart = redisUtils.getString(AppContants.SYSTEM_START_FLAG);
         if (transferDto.getId() == null) {
-            String hasStart = redisUtils.getString(AppContants.SYSTEM_START_FLAG);
             if (DappFundFlowEntity.TYPE_SALE == transferDto.getType()) {
                 if (!"start".equals(hasStart)) {
                     throw new FebsException("系统还未启动");
@@ -266,12 +267,18 @@
                 // 修改用户24小时可售量
                 redisUtils.set(AppContants.REDIS_KEY_COIN_REMAIN + member.getAddress(), saleCoinRemain);
             } else if (DappFundFlowEntity.TYPE_BUY == transferDto.getType()) {
-                buyCoinRemain = buyCoinRemain.subtract(flow.getAmount());
+                if ("start".equals(hasStart)) {
+                    buyCoinRemain = buyCoinRemain.subtract(flow.getAmount());
 
-                // 修改当日可购买量
-                redisUtils.set(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN, buyCoinRemain);
+                    // 修改当日可购买量
+                    redisUtils.set(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN, buyCoinRemain);
+                } else {
+                    makeCoinRemain = makeCoinRemain.subtract(flow.getAmount());
+
+                    // 修改铸池量
+                    redisUtils.set(AppContants.REDIS_KEY_MAKE_POOL_CNT, makeCoinRemain);
+                }
             } else {
-
                 throw new FebsException("参数错误");
             }
         } else {
@@ -281,7 +288,7 @@
     }
 
     @Override
-    public Map<String, BigDecimal> calPrice() {
+    public Map<String, BigDecimal> calPrice(PriceDto priceDto) {
         DappMemberEntity member = LoginUserUtil.getAppUser();
 
 
@@ -299,7 +306,7 @@
         // 源池代币剩余数量
         BigDecimal sourceCoin = tfcInstance.balanceOf(ChainEnum.BSC_USDT_SOURCE.getAddress());
         // 用户代币剩余数量
-        BigDecimal coin = tfcInstance.balanceOf(member.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);
 

--
Gitblit v1.9.1