From d0dd32f897c2b6786f4da8b860b7dcb58a71ac62 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Fri, 26 Aug 2022 16:11:44 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java |  245 +++++++++++++++++++------------------------------
 1 files changed, 95 insertions(+), 150 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 be99e2a..1fe9315 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
@@ -19,6 +19,7 @@
 import cc.mrbird.febs.dapp.vo.ActiveNftListVo;
 import cc.mrbird.febs.dapp.vo.WalletInfoVo;
 import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.date.DateField;
 import cn.hutool.core.date.DateUnit;
 import cn.hutool.core.date.DateUtil;
 import com.baomidou.mybatisplus.core.conditions.Wrapper;
@@ -58,21 +59,17 @@
     public WalletInfoVo walletInfo() {
         DappMemberEntity member = LoginUserUtil.getAppUser();
 
-        Map<String, BigDecimal> map = dappFundFlowDao.selectAmountTotalByType(member.getId());
+        DappMemberEntity memberInfo = dappMemberDao.selectById(member.getId());
         WalletInfoVo walletInfo = new WalletInfoVo();
         List<DappMemberEntity> direct = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 1);
-        List<DappMemberEntity> notDirect = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 2);
-        BigDecimal childHoldAmount = dappMemberDao.selectChildHoldAmount(member.getInviteId());
-
         DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
 
-        walletInfo.setTotalChild(notDirect.size());
         walletInfo.setDirectCnt(direct.size());
-        walletInfo.setTotalChildCoin(childHoldAmount);
-        walletInfo.setTeamReward(map.get("teamReward"));
-        walletInfo.setMiningAmount(map.get("mine"));
         walletInfo.setInviteId(member.getInviteId());
-        walletInfo.setBoxCnt(walletCoin.getBoxCnt());
+        walletInfo.setBalance(walletCoin.getAvailableAmount());
+        walletInfo.setHasBuy(memberInfo.getActiveStatus());
+        walletInfo.setOutCnt(1);
+        walletInfo.setProfit(BigDecimal.TEN);
         return walletInfo;
     }
 
@@ -218,127 +215,36 @@
     public Long transfer(TransferDto transferDto) {
         DappMemberEntity member = LoginUserUtil.getAppUser();
 
-        if (member.getActiveStatus() == 2) {
-            throw new FebsException("请联系邀请人转币激活");
+        member = dappMemberDao.selectById(member.getId());
+        if (member.getActiveStatus() == 1) {
+            throw new FebsException("Do not repeat purchase");
         }
 
-        // 每日出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());
+        if (transferDto.getBuyType() == 1) {
+            DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
+            if (transferDto.getAmount().compareTo(walletCoin.getAvailableAmount()) > 0) {
+                throw new FebsException("Balance Not Enough");
+            }
+            walletCoin.setTotalAmount(walletCoin.getTotalAmount().subtract(transferDto.getAmount()));
+            walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().subtract(transferDto.getAmount()));
 
-        // 用户24小时可购买USDT
-        BigDecimal buyUsdtMax = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_IDO_USDT_MAX_BUY_DAILY + 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);
-
-        String hasStart = redisUtils.getString(AppContants.SYSTEM_START_FLAG);
-        if (transferDto.getId() == null) {
-            if (DappFundFlowEntity.TYPE_SALE == transferDto.getType()) {
-                if (!"start".equals(hasStart)) {
-                    throw new FebsException(MessageSourceUtils.getString("transfer_msg_001"));
-                }
-
-                if (transferDto.getAmount().multiply(transferDto.getPrice()).compareTo(usdtRemain) > 0) {
-                    throw new FebsException(MessageSourceUtils.getString("transfer_msg_002"));
-                }
-
-                if (transferDto.getAmount().compareTo(saleCoinRemain) > 0) {
-                    throw new FebsException(MessageSourceUtils.getString("transfer_msg_003"));
-                }
-
-                usdtRemain = usdtRemain.subtract(transferDto.getAmount().multiply(transferDto.getPrice()));
-                saleCoinRemain = saleCoinRemain.subtract(transferDto.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()) {
-                // 购买时,前端传来的amount是USDT,卖出amount是TFC
-                BigDecimal usdtAmount = transferDto.getAmount();
-                BigDecimal coinAmount = transferDto.getAmount().divide(transferDto.getPrice(), 6, RoundingMode.HALF_UP);
-                transferDto.setAmount(coinAmount);
-
-                if ("start".equals(hasStart)) {
-                    if (transferDto.getAmount().compareTo(buyCoinRemain) > 0) {
-                        throw new FebsException(MessageSourceUtils.getString("transfer_msg_004"));
-                    }
-                    buyCoinRemain = buyCoinRemain.subtract(transferDto.getAmount());
-
-                    // 修改当日可购买量
-                    redisUtils.set(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN, buyCoinRemain);
-                    // 如果系统还没有启动,则判断铸池中的剩余量
-                } else {
-                    // 最少购买
-//                    DataDictionaryCustom dic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.DIC_TYPE_SYSTEM_SETTING, AppContants.DIC_VALUE_MAKER_MIN_LIMIT);
-//                    if (transferDto.getAmount().compareTo(new BigDecimal(dic.getValue())) < 0) {
-//                        throw new FebsException(MessageSourceUtils.getString("transfer_msg_005"));
-//                    }
-
-                    if (buyUsdtMax.compareTo(usdtAmount) < 0) {
-                        throw new FebsException(MessageSourceUtils.getString("transfer_msg_007"));
-                    }
-
-                    if (transferDto.getAmount().compareTo(makeCoinRemain) > 0) {
-                        throw new FebsException(MessageSourceUtils.getString("transfer_msg_006"));
-                    }
-                    makeCoinRemain = makeCoinRemain.subtract(transferDto.getAmount());
-                    buyUsdtMax = buyUsdtMax.subtract(usdtAmount);
-
-                    // 修改每日最大购买USDT量
-                    redisUtils.set(AppContants.REDIS_KEY_IDO_USDT_MAX_BUY_DAILY + member.getAddress(), buyUsdtMax);
-
-                    // 修改铸池量
-                    redisUtils.set(AppContants.REDIS_KEY_MAKE_POOL_CNT, makeCoinRemain);
-                }
+            dappWalletCoinDao.updateById(walletCoin);
+        } else {
+            if (transferDto.getId() == null) {
+                DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), transferDto.getAmount(), transferDto.getType(), 1, transferDto.getFee(), transferDto.getTxHash());
+                dappFundFlowDao.insert(fundFlow);
+                return fundFlow.getId();
             }
 
-            DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), transferDto.getAmount(), transferDto.getType(), 1, transferDto.getFee(), transferDto.getTxHash());
-            fundFlow.setNewestPrice(transferDto.getPrice());
-            dappFundFlowDao.insert(fundFlow);
-            return fundFlow.getId();
-        }
-
-        if ("success".equals(transferDto.getFlag())) {
-            DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId());
-
-            flow.setFromHash(transferDto.getTxHash());
-            dappFundFlowDao.updateById(flow);
-        } else {
-            DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId());
-            if (flow.getStatus() == 1) {
-                if (DappFundFlowEntity.TYPE_BUY == flow.getType()) {
-                    // 购买时,前端传来的amount是USDT,卖出amount是TFC
-                    BigDecimal usdtAmount = transferDto.getAmount();
-                    BigDecimal coinAmount = transferDto.getAmount().divide(flow.getNewestPrice(), 6, RoundingMode.HALF_UP);
-                    transferDto.setAmount(coinAmount);
-
-                    if ("start".equals(hasStart)) {
-                        buyCoinRemain = buyCoinRemain.add(flow.getAmount());
-
-                        // 修改当日可购买量
-                        redisUtils.set(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN, buyCoinRemain);
-                    } else {
-                        makeCoinRemain = makeCoinRemain.add(flow.getAmount());
-                        buyUsdtMax = buyUsdtMax.add(usdtAmount);
-
-                        // 修改铸池量
-                        redisUtils.set(AppContants.REDIS_KEY_MAKE_POOL_CNT, makeCoinRemain);
-                        redisUtils.set(AppContants.REDIS_KEY_IDO_USDT_MAX_BUY_DAILY + member.getAddress(), buyUsdtMax);
-                    }
-                } else {
-                    usdtRemain = usdtRemain.add(transferDto.getAmount().multiply(transferDto.getPrice()));
-                    saleCoinRemain = saleCoinRemain.add(transferDto.getAmount());
-
-                    // 修改当日U剩余量
-                    redisUtils.set(AppContants.REDIS_KEY_USDT_OUT_LIMIT_REMAIN, usdtRemain);
-                    // 修改用户24小时可售量
-                    redisUtils.set(AppContants.REDIS_KEY_COIN_REMAIN + member.getAddress(), saleCoinRemain);
+            if ("success".equals(transferDto.getFlag())) {
+                DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId());
+                flow.setFromHash(transferDto.getTxHash());
+                dappFundFlowDao.updateById(flow);
+            } else {
+                DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId());
+                if (flow.getStatus() == 1) {
+                    dappFundFlowDao.deleteById(transferDto.getId());
                 }
-                dappFundFlowDao.deleteById(transferDto.getId());
             }
         }
         return null;
@@ -369,28 +275,23 @@
         HashMap<String, BigDecimal> map = new HashMap<>();
         map.put("x", x);
         map.put("y", y);
-        map.put("nft", AppContants.NFT_ACTIVE_PRICE);
         return map;
     }
 
     @Override
-    public int boxSurprise() {
+    public ActiveNftListVo boxSurprise() {
+        DappMemberEntity member = LoginUserUtil.getAppUser();
 
-        // 功能升级
-        throw new FebsException(MessageSourceUtils.getString("box_surprise_002"));
+        DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
+        if (walletCoin.getBoxCnt() < 1) {
+            throw new FebsException(MessageSourceUtils.getString("box_surprise_001"));
+        }
 
-//        DappMemberEntity member = LoginUserUtil.getAppUser();
-//
-//        DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
-//        if (walletCoin.getBoxCnt() < 1) {
-//            throw new FebsException(MessageSourceUtils.getString("box_surprise_001"));
-//        }
-//
-//        walletCoin.setBoxCnt(walletCoin.getBoxCnt() - 1);
-//        dappWalletCoinDao.updateById(walletCoin);
-//
-//        BoxUtil.Box box = BoxUtil.openBox();
-//
+        walletCoin.setBoxCnt(walletCoin.getBoxCnt() - 1);
+        dappWalletCoinDao.updateById(walletCoin);
+
+        BoxUtil.Box box = BoxUtil.openBox();
+
 //        DappFundFlowEntity boxFundFlow = new DappFundFlowEntity(member.getId(), new BigDecimal(box.getIndex()), 6, 2, BigDecimal.ZERO);
 //        dappFundFlowDao.insert(boxFundFlow);
 //
@@ -407,25 +308,62 @@
 //                log.error("发放卡牌错误", e);
 //            }
 //        }).start();
-//
-//        return box.getIndex();
+
+        Date time = new Date();
+        Date expire = DateUtil.offset(time, DateField.HOUR, 2);
+        DappNftActivation nftActivation = new DappNftActivation();
+        nftActivation.setMemberId(member.getId());
+        nftActivation.setCount(box.getIndex());
+        nftActivation.setOpenTime(time);
+        nftActivation.setExpireTime(expire);
+        nftActivation.setStatus(1);
+        dappNftActivationDao.insert(nftActivation);
+
+        ActiveNftListVo nft = new ActiveNftListVo();
+        nft.setCount(box.getIndex());
+        nft.setId(nftActivation.getId());
+        nft.setRemain(DateUtil.between(time, expire, DateUnit.SECOND));
+        return nft;
     }
 
     @Override
     @Transactional(rollbackFor = Exception.class)
-    public void activeNft(ActiveDto activeDto) {
+    public synchronized void activeNft(ActiveDto activeDto) {
         DappMemberEntity member = LoginUserUtil.getAppUser();
-        DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), BigDecimal.ONE, 8, 1, BigDecimal.ZERO, activeDto.getTxHash());
-        fundFlow.setNewestPrice(AppContants.NFT_ACTIVE_PRICE);
-
-        dappFundFlowDao.insert(fundFlow);
 
         DappNftActivation nftActive = dappNftActivationDao.selectById(activeDto.getId());
+        if (nftActive == null) {
+            throw new FebsException(MessageSourceUtils.getString("nft_active_003"));
+        }
+
+        if (nftActive.getCount() < activeDto.getCount()) {
+            throw new FebsException(MessageSourceUtils.getString("nft_active_004"));
+        }
+
+        if (DateUtil.between(new Date(), nftActive.getExpireTime(), DateUnit.SECOND) < 0) {
+            throw new FebsException(MessageSourceUtils.getString("nft_active_005"));
+        }
+
         if (nftActive.getStatus() != 1) {
             throw new FebsException(MessageSourceUtils.getString("nft_active_001"));
         }
 
-        nftActive.setStatus(2);
+        PriceDto priceDto = new PriceDto();
+        priceDto.setAmount(BigDecimal.ZERO);
+        Map<String, BigDecimal> price = calPrice(priceDto);
+        DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), new BigDecimal(activeDto.getCount()), 8, 1, BigDecimal.ZERO, activeDto.getTxHash());
+
+        fundFlow.setTargetAmount(price.get("x").multiply(new BigDecimal(activeDto.getCount())));
+        fundFlow.setNewestPrice(AppContants.NFT_ACTIVE_PRICE.multiply(BigDecimal.valueOf(nftActive.getCount())));
+        dappFundFlowDao.insert(fundFlow);
+
+        int count = nftActive.getCount() - activeDto.getCount();
+        if (count == 0) {
+            nftActive.setStatus(3);
+        }
+
+        nftActive.setCount(count);
+        nftActive.setHash(activeDto.getTxHash());
         dappNftActivationDao.updateById(nftActive);
     }
 
@@ -435,8 +373,9 @@
         List<ActiveNftListVo> list = new ArrayList<>();
 
         UpdateWrapper<DappNftActivation> query = new UpdateWrapper<>();
-        query.eq("status", 3);
+        query.eq("status", 1);
         query.eq("member_id", member.getId());
+        query.ge("expire_time", new Date());
         List<DappNftActivation> nftActivations = dappNftActivationDao.selectList(query);
 
         if (CollUtil.isEmpty(nftActivations)) {
@@ -447,10 +386,16 @@
         nftActivations.forEach(nft -> {
             ActiveNftListVo nftVo = new ActiveNftListVo();
             nftVo.setId(nft.getId());
-            nftVo.setRemain(DateUtil.between(nft.getOpenTime(), now, DateUnit.SECOND, false));
+            nftVo.setRemain(DateUtil.between(now, nft.getExpireTime(), DateUnit.SECOND, false));
+            nftVo.setCount(nft.getCount());
             list.add(nftVo);
         });
 
         return list;
     }
+
+    @Override
+    public void withdraw(WithdrawDto withdrawDto) {
+
+    }
 }

--
Gitblit v1.9.1