From 4f14bc15fe02eff37fb0f7581c1b7b11f880405a Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Thu, 03 Sep 2020 20:51:16 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/utils/CalculateUtil.java |  188 +++++++++++++++++------------------------------
 1 files changed, 68 insertions(+), 120 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/utils/CalculateUtil.java b/src/main/java/com/xcong/excoin/utils/CalculateUtil.java
index 9a6b047..d687981 100644
--- a/src/main/java/com/xcong/excoin/utils/CalculateUtil.java
+++ b/src/main/java/com/xcong/excoin/utils/CalculateUtil.java
@@ -6,7 +6,6 @@
 import com.xcong.excoin.common.enumerates.CoinTypeEnum;
 import com.xcong.excoin.common.enumerates.RabbitPriceTypeEnum;
 import com.xcong.excoin.common.exception.GlobalException;
-import com.xcong.excoin.common.response.Result;
 import com.xcong.excoin.modules.contract.dao.ContractHoldOrderDao;
 import com.xcong.excoin.modules.contract.entity.ContractHoldOrderEntity;
 import com.xcong.excoin.modules.member.dao.MemberWalletContractDao;
@@ -22,7 +21,6 @@
 import javax.validation.constraints.NotNull;
 import java.math.BigDecimal;
 import java.math.RoundingMode;
-import java.util.ArrayList;
 import java.util.List;
 
 /**
@@ -85,142 +83,92 @@
 
     /**
      * 全仓模式 -- 预估强平价
-     * 面值*(多单张数*多单开仓价-空单张数*空单开仓价)-余额-已实现盈亏 / 面值*(多单张数-空单张数)-(维持保证金率+TAKER手续费)*面值*(开多张数+开空张数)
+     * 开仓价 - (权益 - 其他币种成本)/当前币种成本 * (开仓价 * 杠杆)
      */
-    public static void getForceSetPriceForWhole(@NotNull String symbol, @NotNull MemberEntity memberEntity) {
+    public static void getForceSetPriceForWhole(@NotNull String aa, @NotNull MemberEntity memberEntity) {
         ContractHoldOrderDao holdOrderDao = SpringContextHolder.getBean(ContractHoldOrderDao.class);
         MemberWalletContractDao walletContractDao = SpringContextHolder.getBean(MemberWalletContractDao.class);
 
         Long memberId = memberEntity.getId();
-        MemberWalletContractEntity walletContract = walletContractDao.findWalletContractByMemberIdAndSymbol(memberId, CoinTypeConvert.convertContractTypeToCoin(symbol));
-        List<ContractHoldOrderEntity> holdOrderEntities = holdOrderDao.selectHoldOrderListForWholeByMemberIdAndSymbol(memberId, symbol);
+        MemberWalletContractEntity walletContract = walletContractDao.findWalletContractByMemberIdAndSymbol(memberId, CoinTypeEnum.USDT.name());
+        List<ContractHoldOrderEntity> holdOrderEntities = holdOrderDao.selectHoldOrderListForWholeByMemberIdAndSymbol(memberId, null);
+        List<String> symbols = holdOrderDao.selectWholeHoldOrderSymbolsByMemberId(memberId);
         if (CollUtil.isNotEmpty(holdOrderEntities)) {
-            int maxCnt = 0;
-            int subCnt = 0;
-            BigDecimal openPrice = BigDecimal.ZERO;
-            int type = 1;
-            BigDecimal bondAmount = BigDecimal.ZERO;
-            BigDecimal sku = BigDecimal.ZERO;
 
-            Long id = 0L;
-            int operateNo = 0;
-            for (ContractHoldOrderEntity holdOrderEntity : holdOrderEntities) {
-                if (holdOrderEntity.getSymbolCntSale() >= maxCnt) {
-                    int temp = maxCnt;
-                    maxCnt = holdOrderEntity.getSymbolCntSale();
-                    subCnt = holdOrderEntity.getSymbolCntSale() - temp;
-                    openPrice = holdOrderEntity.getOpeningPrice();
-                    type = holdOrderEntity.getOpeningType();
+            for (String symbol : symbols) {
+                // 其他币种成本
+                BigDecimal totalBondAmount = BigDecimal.ZERO;
+                // 当前币种手续费
+                BigDecimal symbolFeeAmount = BigDecimal.ZERO;
+                // 当前币种保证金
+                BigDecimal symbolBondAmount = BigDecimal.ZERO;
 
-                    id = holdOrderEntity.getId();
-                    operateNo = holdOrderEntity.getOperateNo() + 1;
-                } else {
-                    subCnt = maxCnt - holdOrderEntity.getSymbolCntSale();
+                // 开仓均价
+                BigDecimal openPrice = BigDecimal.ZERO;
+                // 总盈亏
+                BigDecimal profitOrLoss = BigDecimal.ZERO;
+                // 杠杆
+                int leverRatio = 0;
+                for (ContractHoldOrderEntity holdOrderEntity : holdOrderEntities) {
+                    BigDecimal bondAmount = holdOrderEntity.getBondAmount().subtract(holdOrderEntity.getOpeningFeeAmount());
+
+                    if (symbol.equalsIgnoreCase(holdOrderEntity.getSymbol())) {
+                        symbolFeeAmount = symbolFeeAmount.add(holdOrderEntity.getOpeningFeeAmount());
+                        symbolBondAmount = symbolBondAmount.add(bondAmount);
+
+                        if (openPrice.compareTo(BigDecimal.ZERO) == 0) {
+                            openPrice = holdOrderEntity.getOpeningPrice();
+                        } else {
+                            openPrice = openPrice.add(holdOrderEntity.getOpeningPrice()).divide(BigDecimal.valueOf(2), 8, BigDecimal.ROUND_DOWN);
+                        }
+                        leverRatio = holdOrderEntity.getLeverRatio();
+                    } else {
+                        totalBondAmount = totalBondAmount.add(holdOrderEntity.getBondAmount());
+                    }
+                    profitOrLoss = profitOrLoss.add(calProfitOrLoss(holdOrderEntity, memberEntity));
                 }
 
-                bondAmount = bondAmount.add(holdOrderEntity.getBondAmount().subtract(holdOrderEntity.getOpeningFeeAmount()));
-                sku = holdOrderEntity.getSymbolSku();
-            }
-            BigDecimal forceSetPrice = BigDecimal.ZERO;
-            if (subCnt != 0) {
-                log.info("---->{}, {}", walletContract.getAvailableBalance(), bondAmount);
-                forceSetPrice = getForceSetPrice(walletContract.getAvailableBalance().add(bondAmount), openPrice, subCnt, sku, type, memberEntity);
-            }
+                log.info("{}, {}, {}, {}, {}, {}", totalBondAmount, symbolFeeAmount, symbolFeeAmount, openPrice, profitOrLoss, leverRatio);
+                BigDecimal sub = walletContract.getTotalBalance().add(profitOrLoss).subtract(symbolFeeAmount).subtract(totalBondAmount);
+                log.info("sub -- {}", sub);
+                BigDecimal divide = sub.divide(symbolBondAmount, 8, BigDecimal.ROUND_DOWN);
+                log.info("divide -- {}", divide);
+                BigDecimal divide2 = openPrice.divide(BigDecimal.valueOf(leverRatio), 8, BigDecimal.ROUND_DOWN);
+                log.info("divide2 -- {}", divide2);
 
-            log.info("强平价:{}", forceSetPrice);
-            for (ContractHoldOrderEntity updateHoldOrder : holdOrderEntities) {
-                updateHoldOrder.setForceClosingPrice(forceSetPrice);
-                updateHoldOrder.setIsCanClosing(ContractHoldOrderEntity.ORDER_CAN_CLOSING_Y);
-                updateHoldOrder.setOperateNo(operateNo);
-                holdOrderDao.updateById(updateHoldOrder);
-            }
-
-
-            if (forceSetPrice.compareTo(BigDecimal.ZERO) > 0) {
-                log.info("id:{}, type:{}, forceSetPrice:{}, symbol:{}, operateNo:{}", id, type, forceSetPrice, symbol, operateNo);
-                sendOrderBombMsg(id, type, forceSetPrice, symbol, operateNo, memberEntity.getId());
+                BigDecimal forcePrice = openPrice.subtract(divide.multiply(divide2));
+                log.info("forcePrice -- {}", forcePrice);
             }
         }
     }
 
-    /*
-    BigDecimal totalSetPrice = BigDecimal.ZERO;
-            int totalCnt = 0;
+    private 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());
+        BigDecimal newPrice = new BigDecimal(redisUtils.getString(CoinTypeConvert.convertToKey(holdOrderEntity.getSymbol())));
 
-            for (ContractHoldOrderEntity holdOrderEntity : holdOrderEntities) {
-                BigDecimal forceSetPrice = getForceSetPrice(walletContract.getTotalBalance().subtract(holdOrderEntity.getOpeningFeeAmount()), holdOrderEntity.getOpeningPrice(), holdOrderEntity.getSymbolCntSale(), holdOrderEntity.getSymbolSku(), holdOrderEntity.getOpeningType(), memberEntity);
-                log.info("订单强平价 : {}, 订单ID : {}, {}", forceSetPrice, holdOrderEntity.getId(), holdOrderEntity.getSymbolCntSale());
-                totalSetPrice = totalSetPrice.add(forceSetPrice.multiply(BigDecimal.valueOf(holdOrderEntity.getSymbolCntSale())));
-                totalCnt += holdOrderEntity.getSymbolCntSale();
-                log.info("-->{}", totalCnt);
+        // 盈亏
+        BigDecimal rewardRatio = BigDecimal.ZERO;
+        // 开多
+        if (ContractHoldOrderEntity.OPENING_TYPE_MORE == holdOrderEntity.getOpeningType()) {
+            // (最新价-开仓价)*规格*张数
+            rewardRatio = newPrice.subtract(holdOrderEntity.getOpeningPrice()).multiply(lotNumber).multiply(new BigDecimal(holdOrderEntity.getSymbolCntSale()));
+            // 开空
+        } else {
+            // (开仓价-最新价)*规格*张数
+            rewardRatio = holdOrderEntity.getOpeningPrice().subtract(newPrice).multiply(lotNumber).multiply(new BigDecimal(holdOrderEntity.getSymbolCntSale()));
+        }
+
+        if (memberEntity.getIsProfit() == MemberEntity.IS_PROFIT_Y) {
+            PlatformTradeSettingEntity tradeSettingEntity = cacheSettingUtils.getTradeSetting();
+            if (rewardRatio.compareTo(BigDecimal.ZERO) > -1) {
+                rewardRatio = rewardRatio.multiply(BigDecimal.ONE.subtract(tradeSettingEntity.getProfitParam()));
             }
+        }
 
-            BigDecimal forceSetPrice = totalSetPrice.divide(BigDecimal.valueOf(totalCnt), 8, BigDecimal.ROUND_DOWN);
-            log.info("强平价 : {}", forceSetPrice);
-            for (ContractHoldOrderEntity updateHoldOrder : holdOrderEntities) {
-                updateHoldOrder.setForceClosingPrice(forceSetPrice);
-                updateHoldOrder.setIsCanClosing(ContractHoldOrderEntity.ORDER_CAN_CLOSING_Y);
-                holdOrderDao.updateById(updateHoldOrder);
-
-                if (forceSetPrice.compareTo(BigDecimal.ZERO) >= 0) {
-                    sendOrderBombMsg(updateHoldOrder.getId(), updateHoldOrder.getOpeningType(), forceSetPrice, updateHoldOrder.getSymbol(), updateHoldOrder.getOperateNo());
-                }
-            }
-     */
-
-    /*
-// 多单开仓价
-            BigDecimal moreOpenPrice = BigDecimal.ZERO;
-            // 多单张数
-            int moreCnt = 0;
-            // 空单开仓价
-            BigDecimal lessOpenPrice = BigDecimal.ZERO;
-            // 空单张数
-            int lessCnt = 0;
-            // 已实现盈亏
-            BigDecimal rewardAmount = BigDecimal.ZERO;
-            for (ContractHoldOrderEntity holdOrderEntity : holdOrderEntities) {
-                if (holdOrderEntity.getOpeningType() == ContractHoldOrderEntity.OPENING_TYPE_MORE) {
-                    moreOpenPrice = holdOrderEntity.getOpeningPrice();
-                    moreCnt = holdOrderEntity.getSymbolCntSale();
-                } else {
-                    lessOpenPrice = holdOrderEntity.getOpeningPrice();
-                    lessCnt = holdOrderEntity.getSymbolCntSale();
-                }
-
-                rewardAmount = rewardAmount.add(holdOrderEntity.getRewardAmount() == null ? BigDecimal.ZERO : holdOrderEntity.getRewardAmount());
-                log.info("rewardAmount : {}", rewardAmount);
-            }
-
-            // 多单张数*多单开仓价-空单张数*空单开仓价
-            BigDecimal allOrderPrice = moreOpenPrice.multiply(BigDecimal.valueOf(moreCnt)).subtract(lessOpenPrice.multiply(BigDecimal.valueOf(lessCnt)));
-            log.info("allOrderPrice : {}", allOrderPrice);
-            // 除数 -- 面值*(多单张数*多单开仓价-空单张数*空单开仓价)-余额-已实现盈亏
-            BigDecimal divisor = lotNumber.multiply(allOrderPrice).subtract(walletContract.getAvailableBalance()).subtract(rewardAmount);
-            log.info("divisor : {}", divisor);
-
-            // 面值*(多单张数-空单张数)
-            BigDecimal dividendOne = lotNumber.multiply(BigDecimal.valueOf(moreCnt + lessCnt));
-            log.info("dividendOne : {}", dividendOne);
-            // (维持保证金率+TAKER手续费)*面值*(开多张数+开空张数)
-            BigDecimal dividendTwo = tradeSettingEntity.getFeeRatio().multiply(lotNumber).multiply(BigDecimal.valueOf(moreCnt + lessCnt));
-            log.info("dividendTwo : {}", dividendTwo);
-
-            BigDecimal forceSetPrice = divisor.divide(dividendOne.subtract(dividendTwo), 8, BigDecimal.ROUND_DOWN);
-            log.info("forceSetPrice : {}", forceSetPrice);
-
-            for (ContractHoldOrderEntity updateHoldOrder : holdOrderEntities) {
-                updateHoldOrder.setForceClosingPrice(forceSetPrice);
-                updateHoldOrder.setIsCanClosing(ContractHoldOrderEntity.ORDER_CAN_CLOSING_Y);
-                holdOrderDao.updateById(updateHoldOrder);
-
-                if (forceSetPrice.compareTo(BigDecimal.ZERO) >= 0) {
-                    sendOrderBombMsg(updateHoldOrder.getId(), updateHoldOrder.getOpeningType(), forceSetPrice, updateHoldOrder.getSymbol(), updateHoldOrder.getOperateNo());
-                }
-            }
-
-
-     */
+        return rewardRatio;
+    }
 
     private static void sendOrderBombMsg(Long id, int type, BigDecimal forceClosingPrice, String symbol, int operateNo, Long memberId) {
         OrderModel model = null;

--
Gitblit v1.9.1