From d8b9adf6a39ad94f22f473d11bc829a1cea5865f Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 26 Aug 2020 19:43:12 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/modules/contract/service/impl/ContractHoldOrderServiceImpl.java |   98 ++++++++++++++++++++++--------------------------
 1 files changed, 45 insertions(+), 53 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/contract/service/impl/ContractHoldOrderServiceImpl.java b/src/main/java/com/xcong/excoin/modules/contract/service/impl/ContractHoldOrderServiceImpl.java
index cc77bc4..978f42a 100644
--- a/src/main/java/com/xcong/excoin/modules/contract/service/impl/ContractHoldOrderServiceImpl.java
+++ b/src/main/java/com/xcong/excoin/modules/contract/service/impl/ContractHoldOrderServiceImpl.java
@@ -131,7 +131,7 @@
         log.info("全仓逻辑");
         // 获取最新价
         BigDecimal newPrice = new BigDecimal(redisUtils.getString(CoinTypeConvert.convertToKey(submitOrderDto.getSymbol())));
-        MemberWalletContractEntity walletContract = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeEnum.USDT.name());
+        MemberWalletContractEntity walletContract = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeConvert.convertContractTypeToCoin(submitOrderDto.getSymbol()));
         PlatformTradeSettingEntity tradeSettingEntity = cacheSettingUtils.getTradeSetting();
 
         MemberSettingEntity memberSetting = memberSettingDao.selectMemberSettingByMemberId(memberEntity.getId());
@@ -146,29 +146,32 @@
         BigDecimal openFeePrice = CalculateUtil.getOpenFeePrice(openingPrice, lotNumber, submitOrderDto.getSymbolCnt(), tradeSettingEntity.getFeeRatio());
         log.info("开仓手续费:{}", openFeePrice);
 
-        ContractHoldOrderEntity wholeHoldOrder = contractHoldOrderDao.selectWholeHoldOrderByOrderType(memberEntity.getId(), submitOrderDto.getOrderType());
+        // 保证金 建仓价*规格*手数*(1/杠杆倍率)
+        BigDecimal bondAmount = CalculateUtil.getBondAmount(openingPrice, lotNumber, submitOrderDto.getSymbolCnt(), submitOrderDto.getLeverRatio());
+        log.info("保证金:{}", bondAmount);
+
+        // 预付款为 --> 保证金+开仓手续费+平仓手续费 (开仓平仓手续费相同)
+        BigDecimal prePaymentAmount = bondAmount.add(openFeePrice).add(openFeePrice);
+        log.info("预付款:{}", prePaymentAmount);
+
+        if (prePaymentAmount.compareTo(walletContract.getAvailableBalance()) > -1) {
+            // 可用余额不足
+            return Result.fail(MessageSourceUtils.getString("member_service_0085"));
+        }
+
+        ContractHoldOrderEntity wholeHoldOrder = contractHoldOrderDao.selectWholeHoldOrderByOrderType(memberEntity.getId(), submitOrderDto.getOrderType(), submitOrderDto.getSymbol());
         // 判断当前是否存在全仓订单,若存在则直接合并,若不存在则创建订单
         if (wholeHoldOrder != null) {
             BigDecimal totalPrice = openingPrice.multiply(BigDecimal.valueOf(submitOrderDto.getSymbolCnt()));
             // (当前开仓价*张数 + 原开仓价 * 原可平张数) / (张数 + 原可平张数)
-            BigDecimal newOpenPrice = totalPrice.add(wholeHoldOrder.getOpeningPrice().multiply(BigDecimal.valueOf(wholeHoldOrder.getSymbolCnt()))).divide(BigDecimal.valueOf(submitOrderDto.getSymbolCnt() + wholeHoldOrder.getSymbolCntSale()), 8, BigDecimal.ROUND_DOWN);
+            BigDecimal newOpenPrice = totalPrice.add(wholeHoldOrder.getOpeningPrice().multiply(BigDecimal.valueOf(wholeHoldOrder.getSymbolCntSale()))).divide(BigDecimal.valueOf(submitOrderDto.getSymbolCnt() + wholeHoldOrder.getSymbolCntSale()), 8, BigDecimal.ROUND_DOWN);
             log.info("计算后开仓价:{}", newOpenPrice);
-            BigDecimal bondAmount = CalculateUtil.getBondAmount(newOpenPrice, lotNumber, submitOrderDto.getSymbolCnt() + wholeHoldOrder.getSymbolCntSale(), submitOrderDto.getLeverRatio());
-            log.info("新保证金:{}", bondAmount);
             BigDecimal fee = openFeePrice.add(wholeHoldOrder.getOpeningFeeAmount());
             log.info("手续费相加:{}", fee);
-            BigDecimal prePaymentAmount = bondAmount.add(fee).add(fee);
-
-            BigDecimal subBondAmount = bondAmount.subtract(wholeHoldOrder.getBondAmount());
-            log.info("保证金差值:{}", subBondAmount);
-
-//            BigDecimal forceClosingPrice = CalculateUtil.getForceSetPriceForWhole(submitOrderDto.getSymbol(), memberEntity);
-//            log.info("新预估强平价:{}", forceClosingPrice);
 
             ContractOrderEntity contractOrderEntity = ContractHoldOrderEntityMapper.INSTANCE.holdOrderToOrder(wholeHoldOrder);
             contractOrderEntity.setBondAmount(bondAmount.add(openFeePrice));
-//            contractOrderEntity.setForceClosingPrice(forceClosingPrice);
-            contractOrderEntity.setOpeningPrice(newOpenPrice);
+            contractOrderEntity.setOpeningPrice(openingPrice);
             contractOrderEntity.setSymbolCnt(submitOrderDto.getSymbolCnt());
             contractOrderEntity.setOpeningTime(new Date());
             contractOrderEntity.setMarkPrice(newPrice);
@@ -177,18 +180,15 @@
 
             wholeHoldOrder.setOpeningFeeAmount(fee);
             wholeHoldOrder.setOpeningPrice(newOpenPrice);
-            wholeHoldOrder.setBondAmount(bondAmount);
+            wholeHoldOrder.setBondAmount(wholeHoldOrder.getBondAmount().add(bondAmount).add(openFeePrice));
             wholeHoldOrder.setSymbolCnt(wholeHoldOrder.getSymbolCnt() + submitOrderDto.getSymbolCnt());
             wholeHoldOrder.setSymbolCntSale(wholeHoldOrder.getSymbolCntSale() + submitOrderDto.getSymbolCnt());
-//            wholeHoldOrder.setForceClosingPrice(forceClosingPrice);
-            wholeHoldOrder.setPrePaymentAmount(prePaymentAmount);
+            wholeHoldOrder.setPrePaymentAmount(prePaymentAmount.add(wholeHoldOrder.getPrePaymentAmount()));
             wholeHoldOrder.setOperateNo(wholeHoldOrder.getOperateNo() + 1);
             int i = contractHoldOrderDao.updateById(wholeHoldOrder);
             if (i > 0) {
-                memberWalletContractDao.increaseWalletContractBalanceById(subBondAmount.add(openFeePrice).negate(), openFeePrice.negate(), null, walletContract.getId());
+                memberWalletContractDao.increaseWalletContractBalanceById(prePaymentAmount.negate(), openFeePrice.negate(), null, walletContract.getId());
 
-                // 发送爆仓消息
-//                sendOrderBombMsg(wholeHoldOrder.getId(), wholeHoldOrder.getOpeningType(), forceClosingPrice, wholeHoldOrder.getSymbol(), wholeHoldOrder.getOperateNo());
                 // 发送预估强平价
                 ThreadPoolUtils.sendWholeForceClosingPrice(submitOrderDto.getSymbol(), memberEntity);
                 // 计算佣金
@@ -208,22 +208,6 @@
             return Result.fail(MessageSourceUtils.getString("member_service_0067"));
         }
 
-        // 保证金 建仓价*规格*手数*(1/杠杆倍率)
-        BigDecimal bondAmount = CalculateUtil.getBondAmount(openingPrice, lotNumber, submitOrderDto.getSymbolCnt(), submitOrderDto.getLeverRatio());
-        log.info("保证金:{}", bondAmount);
-
-        // 预付款为 --> 保证金+开仓手续费+平仓手续费 (开仓平仓手续费相同)
-        BigDecimal prePaymentAmount = bondAmount.add(openFeePrice).add(openFeePrice);
-        log.info("预付款:{}", prePaymentAmount);
-
-        if (prePaymentAmount.compareTo(walletContract.getAvailableBalance()) > -1) {
-            // 可用余额不足
-            return Result.fail(MessageSourceUtils.getString("member_service_0085"));
-        }
-
-        // 预估强平价
-//        BigDecimal forceClosingPrice = CalculateUtil.getForceSetPriceForWhole(submitOrderDto.getSymbol(), memberEntity);
-
         ContractHoldOrderEntity holdOrderEntity = new ContractHoldOrderEntity();
         holdOrderEntity.setMemberId(memberEntity.getId());
         holdOrderEntity.setOrderNo(commonService.generateOrderNo(memberEntity.getId()));
@@ -234,7 +218,6 @@
         holdOrderEntity.setSymbolCntSale(submitOrderDto.getSymbolCnt());
         holdOrderEntity.setSymbolSku(lotNumber);
         holdOrderEntity.setLeverRatio(submitOrderDto.getLeverRatio());
-//        holdOrderEntity.setForceClosingPrice(forceClosingPrice);
         holdOrderEntity.setOpeningFeeAmount(openFeePrice);
         holdOrderEntity.setOpeningPrice(openingPrice);
         holdOrderEntity.setOpeningType(submitOrderDto.getOrderType());
@@ -254,11 +237,8 @@
         if (i > 0) {
             memberWalletContractDao.increaseWalletContractBalanceById(prePaymentAmount.negate(), openFeePrice.negate(), null, walletContract.getId());
 
-            // 发送爆仓消息
-//            sendOrderBombMsg(holdOrderEntity.getId(), holdOrderEntity.getOpeningType(), forceClosingPrice, holdOrderEntity.getSymbol(), holdOrderEntity.getOperateNo());
             // 发送预估强平价
             ThreadPoolUtils.sendWholeForceClosingPrice(submitOrderDto.getSymbol(), memberEntity);
-
             // 计算佣金
             ThreadPoolUtils.calReturnMoney(memberEntity.getId(), contractOrderEntity.getOpeningFeeAmount(), contractOrderEntity, AgentReturnEntity.ORDER_TYPE_OPEN);
 
@@ -556,14 +536,18 @@
             return Result.fail("可平张数不足");
         }
 
+        log.info("--->{},{}", holdOrderEntity.getSymbolCntSale(), wholeCloseOrderDto.getCount());
+        int sub = holdOrderEntity.getSymbolCntSale() - wholeCloseOrderDto.getCount();
+        log.info("剩余张数:{}", sub);
         // 判断当前持仓张数是否等于平仓张数,若等于则直接更新该订单为不可平状态,否则更新可平张数
-        if (wholeCloseOrderDto.getCount() == holdOrderEntity.getSymbolCntSale()) {
-            contractHoldOrderDao.updateHoldOrderIsCanClosingById(ContractHoldOrderEntity.ORDER_CAN_CLOSING_N, wholeCloseOrderDto.getId());
-        } else {
-            // 更新可平张数
-            holdOrderEntity.setSymbolCntSale(holdOrderEntity.getSymbolCntSale() - wholeCloseOrderDto.getCount());
-            contractHoldOrderDao.updateById(holdOrderEntity);
+        if (sub == 0) {
+            holdOrderEntity.setIsCanClosing(ContractHoldOrderEntity.ORDER_CAN_CLOSING_N);
         }
+
+        // 更新可平张数
+        holdOrderEntity.setSymbolCntSale(sub);
+        holdOrderEntity.setSymbolCnt(sub);
+        contractHoldOrderDao.updateById(holdOrderEntity);
 
         // 将待平张数放入缓存
         redisUtils.set(AppContants.CLOSING_ORDER_PREFIX + holdOrderEntity.getId(), wholeCloseOrderDto.getCount());
@@ -746,11 +730,20 @@
         PlatformTradeSettingEntity tradeSetting = cacheSettingUtils.getTradeSetting();
         BigDecimal newPriceSymbol = new BigDecimal(redisUtils.getString(CoinTypeConvert.convertToKey(symbol)));
 
-        // 当前合约委托单
-        List<ContractEntrustOrderEntity> entrustOrderEntities = contractEntrustOrderDao.selectEntrustOrderListByMemberId(memberEntity.getId());
-
-        // 当前持仓列表
-        List<ContractHoldOrderEntity> holdOrderEntities = contractHoldOrderDao.selectHoldOrderListByMemberId(memberEntity.getId());
+        List<ContractEntrustOrderEntity> entrustOrderEntities;
+        List<ContractHoldOrderEntity> holdOrderEntities;
+        MemberWalletContractEntity walletContractEntity;
+        if (memberEntity.getContractPositionType().equals(ContractEntrustOrderEntity.POSITION_TYPE_ADD)) {
+            // 当前合约委托单
+            entrustOrderEntities = contractEntrustOrderDao.selectEntrustOrderListByMemberId(memberEntity.getId());
+            // 当前持仓列表
+            holdOrderEntities = contractHoldOrderDao.selectHoldOrderListByMemberId(memberEntity.getId());
+            walletContractEntity = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeEnum.USDT.name());
+        } else {
+            entrustOrderEntities = contractEntrustOrderDao.selectEntrustOrderListByMemberIdAndSymbol(memberEntity.getId(), symbol);
+            holdOrderEntities = contractHoldOrderDao.selectHoldOrderListForWholeByMemberIdAndSymbol(memberEntity.getId(), symbol);
+            walletContractEntity = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeConvert.convertContractTypeToCoin(symbol));
+        }
 
         // 冻结保证金 -- 即委托单中的保证金之和
         BigDecimal frozenBondAmount = BigDecimal.ZERO;
@@ -792,8 +785,6 @@
                 totalProfitOrLess = totalProfitOrLess.add(profitOrLess);
             }
         }
-
-        MemberWalletContractEntity walletContractEntity = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeEnum.USDT.name());
 
         MemberLevelRateEntity rateEntity = memberLevelRateDao.selectLeverRateByMemberIdAndSymbol(memberEntity.getId(), symbol);
 
@@ -945,6 +936,7 @@
         updateEntity.setId(member.getId());
         int i = memberDao.updateById(updateEntity);
         if (i > 0) {
+            member.setContractPositionType(positionType);
             LoginUserUtils.resetAppLoginUser(member);
             return Result.ok(MessageSourceUtils.getString("member_service_0040"));
         }

--
Gitblit v1.9.1