From 21c4edde92e2c7bade302db52329422821f0a8d5 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 08 Jun 2022 19:23:45 +0800
Subject: [PATCH] fix some problem

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java |   64 ++++++++++++++++++++++++++++---
 1 files changed, 57 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 e6cf50d..03eb81a 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
@@ -31,7 +31,7 @@
 import java.util.Map;
 
 /**
- * @author 
+ * @author
  * @date 2022-03-18
  **/
 @Slf4j
@@ -201,11 +201,43 @@
     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());
+        BigDecimal buyCoinRemain = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN);
+        // 铸池中的币的剩余量
+        BigDecimal makeCoinRemain = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_MAKE_POOL_CNT);
+
         if (transferDto.getId() == null) {
+            String hasStart = redisUtils.getString(AppContants.SYSTEM_START_FLAG);
             if (DappFundFlowEntity.TYPE_SALE == transferDto.getType()) {
-                String hasStart = redisUtils.getString(AppContants.SYSTEM_START_FLAG);
                 if (!"start".equals(hasStart)) {
                     throw new FebsException("系统还未启动");
+                }
+
+                if (transferDto.getAmount().multiply(transferDto.getPrice()).compareTo(usdtRemain) > 0) {
+                    throw new FebsException("超过当日出U剩余量");
+                }
+
+                if (transferDto.getAmount().compareTo(saleCoinRemain) > 0) {
+                    throw new FebsException("超出24小时内可售量");
+                }
+
+            } else if (DappFundFlowEntity.TYPE_BUY == transferDto.getType()) {
+                if ("start".equals(hasStart)) {
+                    if (transferDto.getAmount().compareTo(buyCoinRemain) > 0) {
+                        throw new FebsException("超过当日可购买量");
+                    }
+                    // 如果系统还没有启动,则判断铸池中的剩余量
+                } else {
+                    if (transferDto.getAmount().compareTo(makeCoinRemain) > 0) {
+                        throw new FebsException("可购数量不足");
+                    }
                 }
             }
 
@@ -220,10 +252,27 @@
 
             flow.setFromHash(transferDto.getTxHash());
             dappFundFlowDao.updateById(flow);
+
+            if (DappFundFlowEntity.TYPE_SALE == transferDto.getType()) {
+                usdtRemain = usdtRemain.subtract(flow.getAmount().multiply(flow.getNewestPrice()));
+                saleCoinRemain = saleCoinRemain.subtract(flow.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()) {
+                buyCoinRemain = buyCoinRemain.subtract(flow.getAmount());
+
+                // 修改当日可购买量
+                redisUtils.set(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN, buyCoinRemain);
+            } else {
+
+                throw new FebsException("参数错误");
+            }
         } else {
             dappFundFlowDao.deleteById(transferDto.getId());
         }
-
         return null;
     }
 
@@ -238,12 +287,13 @@
         BigDecimal sourceCoin = tfcInstance.balanceOf(ChainEnum.BSC_USDT_SOURCE.getAddress());
         // 用户代币剩余数量
         BigDecimal coin = tfcInstance.balanceOf(member.getAddress());
-        BigDecimal x = sourceU.divide(sourceCoin, tfcInstance.decimals(), RoundingMode.HALF_DOWN);
-        BigDecimal y = sourceU.divide(sourceCoin.add(coin), tfcInstance.decimals(), RoundingMode.HALF_DOWN);
+        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", BigDecimal.ONE);
-        map.put("y", BigDecimal.valueOf(0.8D));
+        map.put("x", x);
+        map.put("y", y);
         return map;
     }
 }

--
Gitblit v1.9.1