From f252c8e23b48c2117e6a0a0373546585d0824fe2 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Tue, 30 Mar 2021 18:52:54 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/utils/CalculateUtil.java |   45 ++++++++++++++++++++++++++++++++++++---------
 1 files changed, 36 insertions(+), 9 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/utils/CalculateUtil.java b/src/main/java/com/xcong/excoin/utils/CalculateUtil.java
index 183a4a4..d4c2dd2 100644
--- a/src/main/java/com/xcong/excoin/utils/CalculateUtil.java
+++ b/src/main/java/com/xcong/excoin/utils/CalculateUtil.java
@@ -80,7 +80,7 @@
 
     /**
      * 全仓模式 -- 预估强平价
-     * 开仓价 - (权益 - 其他币种成本)/当前币种成本 * (开仓价 * 杠杆)
+     * 开仓价 - (权益 - 其他币种成本)/当前币种成本 * (开仓价 * 1/杠杆)
      */
     public static BigDecimal getForceSetPriceForWhole(String currentSymbol, @NotNull MemberEntity memberEntity) {
         ContractHoldOrderDao holdOrderDao = SpringContextHolder.getBean(ContractHoldOrderDao.class);
@@ -97,8 +97,6 @@
             for (String symbol : symbols) {
                 // 其他币种成本
                 BigDecimal totalBondAmount = BigDecimal.ZERO;
-                // 当前币种手续费
-                BigDecimal symbolFeeAmount = BigDecimal.ZERO;
                 // 当前币种保证金
                 BigDecimal symbolBondAmount = BigDecimal.ZERO;
 
@@ -116,8 +114,7 @@
                         if (holdOrderEntity.getOpeningType() == ContractHoldOrderEntity.OPENING_TYPE_MORE) {
                             isAloneLess = false;
                         }
-                        symbolFeeAmount = symbolFeeAmount.add(holdOrderEntity.getOpeningFeeAmount());
-                        symbolBondAmount = symbolBondAmount.add(bondAmount);
+                        symbolBondAmount = symbolBondAmount.add(bondAmount.subtract(holdOrderEntity.getOpeningFeeAmount()));
 
                         if (openPrice.compareTo(BigDecimal.ZERO) == 0) {
                             openPrice = holdOrderEntity.getOpeningPrice();
@@ -132,8 +129,8 @@
                 }
 
 //                log.info("{}, {}, {}, {}, {}, {}", totalBondAmount, symbolBondAmount, symbolFeeAmount, openPrice, profitOrLoss, leverRatio);
-                BigDecimal equity = walletContract.getTotalBalance().add(profitOrLoss);
-                BigDecimal sub = equity.subtract(symbolFeeAmount).subtract(totalBondAmount);
+                BigDecimal equity = walletContract.getTotalBalance().add(profitOrLoss).subtract(walletContract.getFrozenBalance());
+                BigDecimal sub = equity.subtract(totalBondAmount);
 //                log.info("sub -- {}", sub);
                 if (sub.compareTo(symbolBondAmount) <= 0) {
                     BigDecimal multi = BigDecimal.valueOf(10);
@@ -166,7 +163,37 @@
         return result;
     }
 
-    private static BigDecimal calProfitOrLoss(ContractHoldOrderEntity holdOrderEntity, MemberEntity memberEntity) {
+    public static BigDecimal calForcePriceForWhole(MemberEntity memberEntity, ContractHoldOrderEntity contractHoldOrderEntity) {
+        ContractHoldOrderDao holdOrderDao = SpringContextHolder.getBean(ContractHoldOrderDao.class);
+        MemberWalletContractDao walletContractDao = SpringContextHolder.getBean(MemberWalletContractDao.class);
+
+        Long memberId = memberEntity.getId();
+        MemberWalletContractEntity walletContract = walletContractDao.findWalletContractByMemberIdAndSymbol(memberId, CoinTypeEnum.USDT.name());
+        List<ContractHoldOrderEntity> holdOrderEntities = holdOrderDao.selectHoldOrderListForWholeByMemberIdAndSymbol(memberId, null);
+        BigDecimal forcePrice = BigDecimal.ZERO;
+        if (CollUtil.isNotEmpty(holdOrderEntities)) {
+            BigDecimal totalBondAmount = BigDecimal.ZERO;
+            BigDecimal totalProfitOrLoss = BigDecimal.ZERO;
+            for (ContractHoldOrderEntity holdOrderEntity : holdOrderEntities) {
+                totalBondAmount = totalBondAmount.add(holdOrderEntity.getBondAmount());
+                totalProfitOrLoss =  totalProfitOrLoss.add(calProfitOrLoss(holdOrderEntity, memberEntity));
+            }
+
+            BigDecimal divideChild = walletContract.getTotalBalance().subtract(totalBondAmount).add(totalProfitOrLoss);
+            BigDecimal divideParent = contractHoldOrderEntity.getSymbolSku().multiply(new BigDecimal(contractHoldOrderEntity.getSymbolCntSale()));
+
+            BigDecimal divide = divideChild.divide(divideParent, 8, BigDecimal.ROUND_DOWN);
+
+            if (ContractHoldOrderEntity.OPENING_TYPE_MORE == contractHoldOrderEntity.getOpeningType()) {
+                forcePrice = contractHoldOrderEntity.getOpeningPrice().subtract(divide);
+            } else {
+                forcePrice = contractHoldOrderEntity.getOpeningPrice().add(divide);
+            }
+        }
+        return forcePrice;
+    }
+
+    public static BigDecimal calProfitOrLoss(ContractHoldOrderEntity holdOrderEntity, MemberEntity memberEntity) {
         CacheSettingUtils cacheSettingUtils = SpringContextHolder.getBean(CacheSettingUtils.class);
         RedisUtils redisUtils = SpringContextHolder.getBean(RedisUtils.class);
         BigDecimal lotNumber = cacheSettingUtils.getSymbolSku(holdOrderEntity.getSymbol());
@@ -289,6 +316,6 @@
             redisUtils.set(AppContants.HOLD_BOND_RATIO, tradeSetting.getHoldBondRatio());
         }
 
-        return contractHoldOrder.getOpeningPrice().multiply(new BigDecimal(contractHoldOrder.getSymbolCntSale())).multiply(holdBondRatio).multiply(contractHoldOrder.getSymbolSku());
+        return contractHoldOrder.getOpeningPrice().multiply(new BigDecimal(contractHoldOrder.getSymbolCntSale())).multiply(holdBondRatio.multiply(new BigDecimal(100))).multiply(contractHoldOrder.getSymbolSku()).divide(new BigDecimal(contractHoldOrder.getLeverRatio()), 8, BigDecimal.ROUND_DOWN);
     }
 }

--
Gitblit v1.9.1