From afc924b0d88ef320ad09060ae49804556efd8eb6 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Fri, 11 Sep 2020 10:08:25 +0800
Subject: [PATCH] Merge branch 'master' into whole_new

---
 src/main/java/com/xcong/excoin/modules/contract/service/impl/ContractHoldOrderServiceImpl.java |  402 ++++++++++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 337 insertions(+), 65 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 c70dd46..1a0d271 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
@@ -7,6 +7,7 @@
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.xcong.excoin.common.LoginUserUtils;
+import com.xcong.excoin.common.contants.AppContants;
 import com.xcong.excoin.common.enumerates.CoinTypeEnum;
 import com.xcong.excoin.common.enumerates.RabbitPriceTypeEnum;
 import com.xcong.excoin.common.response.Result;
@@ -28,11 +29,9 @@
 import com.xcong.excoin.modules.documentary.entity.FollowTraderInfoEntity;
 import com.xcong.excoin.modules.member.dao.MemberDao;
 import com.xcong.excoin.modules.member.dao.MemberLevelRateDao;
+import com.xcong.excoin.modules.member.dao.MemberSettingDao;
 import com.xcong.excoin.modules.member.dao.MemberWalletContractDao;
-import com.xcong.excoin.modules.member.entity.AgentReturnEntity;
-import com.xcong.excoin.modules.member.entity.MemberEntity;
-import com.xcong.excoin.modules.member.entity.MemberLevelRateEntity;
-import com.xcong.excoin.modules.member.entity.MemberWalletContractEntity;
+import com.xcong.excoin.modules.member.entity.*;
 import com.xcong.excoin.modules.platform.dao.TradeSettingDao;
 import com.xcong.excoin.modules.platform.entity.PlatformTradeSettingEntity;
 import com.xcong.excoin.rabbit.producer.OrderProducer;
@@ -45,6 +44,7 @@
 import sun.rmi.runtime.Log;
 
 import javax.annotation.Resource;
+import javax.validation.constraints.NotNull;
 import java.math.BigDecimal;
 import java.math.RoundingMode;
 import java.util.*;
@@ -86,7 +86,9 @@
 
     @Resource
     private MemberDao memberDao;
-    
+    @Resource
+    private MemberSettingDao memberSettingDao;
+
     @Resource
     private FollowTraderInfoDao followTraderInfoDao;
     @Resource
@@ -97,47 +99,55 @@
     public Result submitOrder(SubmitOrderDto submitOrderDto) {
         MemberEntity memberEntity = LoginUserUtils.getAppLoginUser();
 
+        // 判断当前对应的持仓/委托
+        if (memberEntity.getContractPositionType() == ContractEntrustOrderEntity.POSITION_TYPE_ADD) {
+            List<ContractHoldOrderEntity> holdList = contractHoldOrderDao.selectMemberHoldOrderByPositionType(ContractEntrustOrderEntity.POSITION_TYPE_ALL, memberEntity.getId());
+            List<ContractEntrustOrderEntity> entrustList = contractEntrustOrderDao.selectMemberEntrustOrderByPositionType(ContractEntrustOrderEntity.POSITION_TYPE_ALL, memberEntity.getId());
+            if (CollUtil.isNotEmpty(holdList) || CollUtil.isNotEmpty(entrustList)) {
+                return Result.fail("存在全仓订单,无法下单");
+            }
+
+            // 逐仓逻辑
+            return doPositionTypeForAdd(submitOrderDto, memberEntity);
+        } else {
+            List<ContractHoldOrderEntity> holdList = contractHoldOrderDao.selectMemberHoldOrderByPositionType(ContractEntrustOrderEntity.POSITION_TYPE_ADD, memberEntity.getId());
+            List<ContractEntrustOrderEntity> entrustList = contractEntrustOrderDao.selectMemberEntrustOrderByPositionType(ContractEntrustOrderEntity.POSITION_TYPE_ADD, memberEntity.getId());
+            if (CollUtil.isNotEmpty(holdList) || CollUtil.isNotEmpty(entrustList)) {
+                return Result.fail("存在逐仓订单,无法下单");
+            }
+
+            // 全仓逻辑
+            return doPositionTypeForWhole(submitOrderDto, memberEntity);
+        }
+
+    }
+
+    /**
+     * 提交订单 -- 全仓逻辑
+     *
+     * @return
+     */
+    public Result doPositionTypeForWhole(SubmitOrderDto submitOrderDto, MemberEntity memberEntity) {
+        log.info("全仓逻辑");
         // 获取最新价
         BigDecimal newPrice = new BigDecimal(redisUtils.getString(CoinTypeConvert.convertToKey(submitOrderDto.getSymbol())));
-
         MemberWalletContractEntity walletContract = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeEnum.USDT.name());
-
         PlatformTradeSettingEntity tradeSettingEntity = cacheSettingUtils.getTradeSetting();
 
-        Long id = memberEntity.getId();
-        MemberEntity selectById = memberDao.selectById(id);
-        BigDecimal spread = selectById.getSpread();
+        MemberSettingEntity memberSetting = memberSettingDao.selectMemberSettingByMemberId(memberEntity.getId());
+        BigDecimal spread = memberSetting.getSpread();
         // 规格
         BigDecimal lotNumber = cacheSettingUtils.getSymbolSku(submitOrderDto.getSymbol());
 
         // 开仓价
-        BigDecimal openingPrice = BigDecimal.ZERO;
+        BigDecimal openingPrice = CalculateUtil.getOpeningPrice(submitOrderDto.getOrderType(), newPrice, spread);
 
-        // 开多
-        if (submitOrderDto.getOrderType() == ContractHoldOrderEntity.OPENING_TYPE_MORE) {
-            // 市场价*(1 + (点差/10000))
-            openingPrice = newPrice.multiply(BigDecimal.ONE.add(spread.divide(new BigDecimal(10000)))).setScale(8, BigDecimal.ROUND_DOWN);
-
-            // 开空
-        } else if (submitOrderDto.getOrderType() == ContractHoldOrderEntity.OPENING_TYPE_LESS) {
-            // 市场价*(1 - (点差/10000))
-            openingPrice = newPrice.multiply(BigDecimal.ONE.subtract(spread.divide(new BigDecimal(10000)))).setScale(8, BigDecimal.ROUND_DOWN);
-        } else {
-            return Result.fail("未知类型");
-        }
-
-        log.info("开仓价:{}", openingPrice);
         // 开仓手续费 建仓价*规格*手数*手续费率
-        BigDecimal openFeePrice = openingPrice.multiply(lotNumber)
-                .multiply(new BigDecimal(submitOrderDto.getSymbolCnt()))
-                .multiply(tradeSettingEntity.getFeeRatio().divide(new BigDecimal(100)))
-                .setScale(8, BigDecimal.ROUND_DOWN);
+        BigDecimal openFeePrice = CalculateUtil.getOpenFeePrice(openingPrice, lotNumber, submitOrderDto.getSymbolCnt(), tradeSettingEntity.getFeeRatio());
         log.info("开仓手续费:{}", openFeePrice);
 
         // 保证金 建仓价*规格*手数*(1/杠杆倍率)
-        BigDecimal bondAmount = openingPrice.multiply(lotNumber).multiply(new BigDecimal(submitOrderDto.getSymbolCnt()))
-                .multiply(BigDecimal.ONE.divide(new BigDecimal(submitOrderDto.getLeverRatio())))
-                .setScale(8, BigDecimal.ROUND_DOWN);
+        BigDecimal bondAmount = CalculateUtil.getBondAmount(openingPrice, lotNumber, submitOrderDto.getSymbolCnt(), submitOrderDto.getLeverRatio());
         log.info("保证金:{}", bondAmount);
 
         // 预付款为 --> 保证金+开仓手续费+平仓手续费 (开仓平仓手续费相同)
@@ -145,7 +155,144 @@
         log.info("预付款:{}", prePaymentAmount);
 
         if (prePaymentAmount.compareTo(walletContract.getAvailableBalance()) > -1) {
-            return Result.fail("可用金额不足");
+            // 可用余额不足
+            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.getSymbolCntSale()))).divide(BigDecimal.valueOf(submitOrderDto.getSymbolCnt() + wholeHoldOrder.getSymbolCntSale()), 8, BigDecimal.ROUND_DOWN);
+            log.info("计算后开仓价:{}", newOpenPrice);
+            BigDecimal fee = openFeePrice.add(wholeHoldOrder.getOpeningFeeAmount());
+            log.info("手续费相加:{}", fee);
+
+            ContractOrderEntity contractOrderEntity = ContractHoldOrderEntityMapper.INSTANCE.holdOrderToOrder(wholeHoldOrder);
+            contractOrderEntity.setBondAmount(bondAmount.add(openFeePrice));
+            contractOrderEntity.setOpeningPrice(openingPrice);
+            contractOrderEntity.setSymbolCnt(submitOrderDto.getSymbolCnt());
+            contractOrderEntity.setOpeningTime(new Date());
+            contractOrderEntity.setMarkPrice(newPrice);
+            contractOrderEntity.setOpeningFeeAmount(openFeePrice);
+            contractOrderDao.insert(contractOrderEntity);
+
+            wholeHoldOrder.setOpeningFeeAmount(fee);
+            wholeHoldOrder.setOpeningPrice(newOpenPrice);
+            wholeHoldOrder.setBondAmount(wholeHoldOrder.getBondAmount().add(bondAmount).add(openFeePrice));
+            wholeHoldOrder.setSymbolCnt(wholeHoldOrder.getSymbolCnt() + submitOrderDto.getSymbolCnt());
+            wholeHoldOrder.setSymbolCntSale(wholeHoldOrder.getSymbolCntSale() + submitOrderDto.getSymbolCnt());
+            wholeHoldOrder.setPrePaymentAmount(prePaymentAmount.add(wholeHoldOrder.getPrePaymentAmount()));
+            wholeHoldOrder.setOperateNo(wholeHoldOrder.getOperateNo());
+            int i = contractHoldOrderDao.updateById(wholeHoldOrder);
+            if (i > 0) {
+                memberWalletContractDao.increaseWalletContractBalanceById(prePaymentAmount.negate(), openFeePrice.negate(), null, walletContract.getId());
+
+                // 发送预估强平价
+                ThreadPoolUtils.sendWholeForceClosingPrice(submitOrderDto.getSymbol(), memberEntity);
+                // 计算佣金
+                ThreadPoolUtils.calReturnMoney(memberEntity.getId(), openFeePrice, contractOrderEntity, AgentReturnEntity.ORDER_TYPE_OPEN);
+
+                // 插入财务流水
+                if (submitOrderDto.getOrderType() == ContractHoldOrderEntity.OPENING_TYPE_MORE) {
+                    LogRecordUtils.insertMemberAccountFlow(memberEntity.getId(), prePaymentAmount, walletContract.getAvailableBalance().subtract(prePaymentAmount), submitOrderDto.getSymbol(), "买涨持仓", "买涨:" + submitOrderDto.getSymbol());
+                } else {
+                    LogRecordUtils.insertMemberAccountFlow(memberEntity.getId(), prePaymentAmount, walletContract.getAvailableBalance().subtract(prePaymentAmount), submitOrderDto.getSymbol(), "买跌持仓", "买跌:" + submitOrderDto.getSymbol());
+                }
+
+                // 提交成功
+                return Result.ok(MessageSourceUtils.getString("member_service_0024"));
+            }
+            // 提交失败
+            return Result.fail(MessageSourceUtils.getString("member_service_0067"));
+        }
+
+        ContractHoldOrderEntity holdOrderEntity = new ContractHoldOrderEntity();
+        holdOrderEntity.setMemberId(memberEntity.getId());
+        holdOrderEntity.setOrderNo(commonService.generateOrderNo(memberEntity.getId()));
+        holdOrderEntity.setPositionType(memberEntity.getContractPositionType());
+        holdOrderEntity.setTradeType(ContractHoldOrderEntity.TRADE_TYPE_MARK);
+        holdOrderEntity.setSymbol(submitOrderDto.getSymbol());
+        holdOrderEntity.setSymbolCnt(submitOrderDto.getSymbolCnt());
+        holdOrderEntity.setSymbolCntSale(submitOrderDto.getSymbolCnt());
+        holdOrderEntity.setSymbolSku(lotNumber);
+        holdOrderEntity.setLeverRatio(submitOrderDto.getLeverRatio());
+        holdOrderEntity.setOpeningFeeAmount(openFeePrice);
+        holdOrderEntity.setOpeningPrice(openingPrice);
+        holdOrderEntity.setOpeningType(submitOrderDto.getOrderType());
+        holdOrderEntity.setMarkPrice(newPrice);
+        holdOrderEntity.setIsCanClosing(ContractHoldOrderEntity.ORDER_CAN_CLOSING_N);
+        holdOrderEntity.setPrePaymentAmount(prePaymentAmount);
+        holdOrderEntity.setBondAmount(bondAmount.add(openFeePrice));
+        holdOrderEntity.setOperateNo(1);
+        // 设置合约类型
+        holdOrderEntity.setContractType(ContractOrderEntity.CONTRACTTYPE_NORMAL);
+
+        ContractOrderEntity contractOrderEntity = ContractHoldOrderEntityMapper.INSTANCE.holdOrderToOrder(holdOrderEntity);
+        contractOrderEntity.setOpeningTime(new Date());
+        contractHoldOrderDao.insert(holdOrderEntity);
+        int i = contractOrderDao.insert(contractOrderEntity);
+
+        if (i > 0) {
+            memberWalletContractDao.increaseWalletContractBalanceById(prePaymentAmount.negate(), openFeePrice.negate(), null, walletContract.getId());
+
+            // 发送预估强平价
+            ThreadPoolUtils.sendWholeForceClosingPrice(submitOrderDto.getSymbol(), memberEntity);
+            // 计算佣金
+            ThreadPoolUtils.calReturnMoney(memberEntity.getId(), contractOrderEntity.getOpeningFeeAmount(), contractOrderEntity, AgentReturnEntity.ORDER_TYPE_OPEN);
+
+            // 插入财务流水
+            if (submitOrderDto.getOrderType() == ContractHoldOrderEntity.OPENING_TYPE_MORE) {
+                LogRecordUtils.insertMemberAccountFlow(memberEntity.getId(), prePaymentAmount, walletContract.getAvailableBalance().subtract(prePaymentAmount), submitOrderDto.getSymbol(), "买涨持仓", "买涨:" + submitOrderDto.getSymbol());
+            } else {
+                LogRecordUtils.insertMemberAccountFlow(memberEntity.getId(), prePaymentAmount, walletContract.getAvailableBalance().subtract(prePaymentAmount), submitOrderDto.getSymbol(), "买跌持仓", "买跌:" + submitOrderDto.getSymbol());
+            }
+
+            // 提交成功
+            return Result.ok(MessageSourceUtils.getString("member_service_0024"));
+        }
+        // 提交失败
+        return Result.fail(MessageSourceUtils.getString("member_service_0067"));
+    }
+
+
+    /**
+     * 提交订单 -- 逐仓逻辑
+     *
+     * @return
+     */
+    public Result doPositionTypeForAdd(@NotNull SubmitOrderDto submitOrderDto, @NotNull MemberEntity memberEntity) {
+        log.info("逐仓逻辑");
+        // 获取最新价
+        BigDecimal newPrice = new BigDecimal(redisUtils.getString(CoinTypeConvert.convertToKey(submitOrderDto.getSymbol())));
+        MemberWalletContractEntity walletContract = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeEnum.USDT.name());
+        PlatformTradeSettingEntity tradeSettingEntity = cacheSettingUtils.getTradeSetting();
+        // 规格
+        BigDecimal lotNumber = cacheSettingUtils.getSymbolSku(submitOrderDto.getSymbol());
+
+        MemberSettingEntity memberSetting = memberSettingDao.selectMemberSettingByMemberId(memberEntity.getId());
+        BigDecimal spread = memberSetting.getSpread();
+
+        // 开仓价
+        BigDecimal openingPrice = CalculateUtil.getOpeningPrice(submitOrderDto.getOrderType(), newPrice, spread);
+        log.info("开仓价:{}", openingPrice);
+
+        // 开仓手续费 建仓价*规格*手数*手续费率
+        BigDecimal openFeePrice = CalculateUtil.getOpenFeePrice(openingPrice, lotNumber, submitOrderDto.getSymbolCnt(), tradeSettingEntity.getFeeRatio());
+        log.info("开仓手续费:{}", openFeePrice);
+
+        // 保证金 建仓价*规格*手数*(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"));
         }
 
         // 预估强平价
@@ -165,10 +312,11 @@
         ContractHoldOrderEntity holdOrderEntity = new ContractHoldOrderEntity();
         holdOrderEntity.setMemberId(memberEntity.getId());
         holdOrderEntity.setOrderNo(commonService.generateOrderNo(memberEntity.getId()));
-        holdOrderEntity.setPositionType(ContractEntrustOrderEntity.POSITION_TYPE_ADD);
+        holdOrderEntity.setPositionType(memberEntity.getContractPositionType());
         holdOrderEntity.setTradeType(ContractHoldOrderEntity.TRADE_TYPE_MARK);
         holdOrderEntity.setSymbol(submitOrderDto.getSymbol());
         holdOrderEntity.setSymbolCnt(submitOrderDto.getSymbolCnt());
+        holdOrderEntity.setSymbolCntSale(submitOrderDto.getSymbolCnt());
         holdOrderEntity.setSymbolSku(lotNumber);
         holdOrderEntity.setLeverRatio(submitOrderDto.getLeverRatio());
         holdOrderEntity.setForceClosingPrice(forceClosingPrice);
@@ -192,7 +340,7 @@
             memberWalletContractDao.increaseWalletContractBalanceById(prePaymentAmount.negate(), openFeePrice.negate(), null, walletContract.getId());
 
             // 发送爆仓消息
-            sendOrderBombMsg(holdOrderEntity.getId(), holdOrderEntity.getOpeningType(), forceClosingPrice, holdOrderEntity.getSymbol(), holdOrderEntity.getOperateNo());
+            sendOrderBombMsg(holdOrderEntity.getId(), holdOrderEntity.getOpeningType(), forceClosingPrice, holdOrderEntity.getSymbol(), holdOrderEntity.getOperateNo(), holdOrderEntity.getMemberId());
 
             // 计算佣金
             ThreadPoolUtils.calReturnMoney(memberEntity.getId(), contractOrderEntity.getOpeningFeeAmount(), contractOrderEntity, AgentReturnEntity.ORDER_TYPE_OPEN);
@@ -213,14 +361,52 @@
                 relationEntity.setOrderType(FollowFollowerOrderRelationEntity.ORDER_TYPE_HOLD);
                 relationEntity.setTradeId(tradeInfo.getId());
                 relationEntity.setTradeMemberId(tradeInfo.getMemberId());
-                relationEntity.setTradeOrderId(id);
+                relationEntity.setTradeOrderNo(holdOrderEntity.getOrderNo());
                 followFollowerOrderRelationDao.insert(relationEntity);
 
                 ThreadPoolUtils.sendFollowOrderTask(holdOrderEntity.getId());
             }
-            return Result.ok("提交成功");
+            // 提交成功
+            return Result.ok(MessageSourceUtils.getString("member_service_0024"));
         }
-        return Result.fail("提交失败");
+        // 提交失败
+        return Result.fail(MessageSourceUtils.getString("member_service_0067"));
+    }
+
+
+    /**
+     * 全仓模式--若当前已经存在持仓,则合并当前持仓
+     */
+    public Result mergeWholeHoldOrder(BigDecimal openPrice, BigDecimal openFeePrice, BigDecimal newPrice, SubmitOrderDto submitOrderDto, ContractHoldOrderEntity holdOrderEntity, MemberEntity memberEntity) {
+        BigDecimal totalPrice = openPrice.multiply(BigDecimal.valueOf(submitOrderDto.getSymbolCnt()));
+        // (当前开仓价*张数 + 原开仓价 * 原可平张数) / (张数 + 原可平张数)
+        BigDecimal newOpenPrice = totalPrice.add(holdOrderEntity.getOpeningPrice().multiply(BigDecimal.valueOf(holdOrderEntity.getSymbolCnt()))).divide(BigDecimal.valueOf(submitOrderDto.getSymbolCnt() + holdOrderEntity.getSymbolCntSale()), 8, BigDecimal.ROUND_DOWN);
+        // 规格
+        BigDecimal lotNumber = cacheSettingUtils.getSymbolSku(submitOrderDto.getSymbol());
+        // 保证金
+        BigDecimal bondAmount = CalculateUtil.getBondAmount(newOpenPrice, lotNumber, submitOrderDto.getSymbolCnt(), submitOrderDto.getLeverRatio());
+        // 预估强平价
+        BigDecimal forceSetPrice = CalculateUtil.getForceSetPrice(bondAmount, newOpenPrice, submitOrderDto.getSymbolCnt(), lotNumber, submitOrderDto.getOrderType(), memberEntity);
+
+        ContractOrderEntity contractOrderEntity = ContractHoldOrderEntityMapper.INSTANCE.holdOrderToOrder(holdOrderEntity);
+        contractOrderEntity.setBondAmount(bondAmount.add(openFeePrice));
+        contractOrderEntity.setForceClosingPrice(forceSetPrice);
+        contractOrderEntity.setOpeningPrice(openPrice);
+        contractOrderEntity.setSymbolCnt(submitOrderDto.getSymbolCnt());
+        contractOrderEntity.setOpeningTime(new Date());
+        contractOrderEntity.setMarkPrice(newPrice);
+
+        contractOrderDao.insert(contractOrderEntity);
+
+        holdOrderEntity.setSymbolCntSale(submitOrderDto.getSymbolCnt() + holdOrderEntity.getSymbolCntSale());
+        holdOrderEntity.setSymbolCnt(submitOrderDto.getSymbolCnt() + holdOrderEntity.getSymbolCnt());
+        holdOrderEntity.setForceClosingPrice(forceSetPrice);
+        holdOrderEntity.setBondAmount(bondAmount);
+        holdOrderEntity.setOpeningPrice(newOpenPrice);
+        holdOrderEntity.setOpeningFeeAmount(holdOrderEntity.getOpeningFeeAmount().add(openFeePrice));
+
+        contractHoldOrderDao.updateById(holdOrderEntity);
+        return Result.ok(MessageSourceUtils.getString("member_service_0024"));
     }
 
     @Override
@@ -239,10 +425,10 @@
     }
 
     @Override
-    public Result findHoldOrderList(String symbol) {
+    public Result findHoldOrderList(String symbol, int type) {
         MemberEntity memberEntity = LoginUserUtils.getAppLoginUser();
 
-        List<ContractHoldOrderEntity> list = contractHoldOrderDao.selectHoldOrderListByMemberIdAndSymbol(memberEntity.getId(), symbol);
+        List<ContractHoldOrderEntity> list = contractHoldOrderDao.selectHoldOrderListByMemberIdAndSymbol(memberEntity.getId(), symbol, type);
         MemberWalletContractEntity walletContractEntity = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeEnum.USDT.name());
         if (CollUtil.isNotEmpty(list)) {
             BigDecimal totalProfitOrLoss = BigDecimal.ZERO;
@@ -257,11 +443,11 @@
                 // 开多
                 if (ContractHoldOrderEntity.OPENING_TYPE_MORE == holdOrderEntity.getOpeningType()) {
                     // (最新价-开仓价)*规格*张数
-                    rewardRatio = newPrice.subtract(holdOrderEntity.getOpeningPrice()).multiply(lotNumber).multiply(new BigDecimal(holdOrderEntity.getSymbolCnt()));
+                    rewardRatio = newPrice.subtract(holdOrderEntity.getOpeningPrice()).multiply(lotNumber).multiply(new BigDecimal(holdOrderEntity.getSymbolCntSale()));
                     // 开空
                 } else {
                     // (开仓价-最新价)*规格*张数
-                    rewardRatio = holdOrderEntity.getOpeningPrice().subtract(newPrice).multiply(lotNumber).multiply(new BigDecimal(holdOrderEntity.getSymbolCnt()));
+                    rewardRatio = holdOrderEntity.getOpeningPrice().subtract(newPrice).multiply(lotNumber).multiply(new BigDecimal(holdOrderEntity.getSymbolCntSale()));
                 }
 
                 if (memberEntity.getIsProfit() == MemberEntity.IS_PROFIT_Y) {
@@ -296,6 +482,10 @@
                 holdOrderListVo.setCanAddMaxBond(walletContractEntity.getAvailableBalance());
                 holdOrderListVo.setReturnRate(returnRate);
                 holdOrderListVo.setProfitOrLoss(rewardRatio);
+                if (ContractEntrustOrderEntity.POSITION_TYPE_ALL == memberEntity.getContractPositionType()) {
+                    BigDecimal forcePrice = CalculateUtil.getForceSetPriceForWhole(holdOrderEntity.getSymbol(), memberEntity);
+                    holdOrderListVo.setForceClosingPrice(forcePrice);
+                }
                 resultList.add(holdOrderListVo);
                 totalProfitOrLoss = totalProfitOrLoss.add(rewardRatio);
             }
@@ -313,11 +503,13 @@
         MemberEntity memberEntity = LoginUserUtils.getAppLoginUser();
         ContractHoldOrderEntity holdOrderEntity = contractHoldOrderDao.selectHoldOrderByMemberIdAndId(memberEntity.getId(), id);
         if (holdOrderEntity == null) {
-            return Result.fail("订单不存在");
+            // 订单不存在
+            return Result.fail(MessageSourceUtils.getString("order_service_0027"));
         }
 
         if (ContractHoldOrderEntity.ORDER_CAN_CLOSING_N == holdOrderEntity.getIsCanClosing()) {
-            return Result.fail("订单暂不可平仓");
+            // 该订单暂不能平仓
+            return Result.fail(MessageSourceUtils.getString("order_service_0045"));
         }
 
         contractHoldOrderDao.updateHoldOrderIsCanClosingById(ContractHoldOrderEntity.ORDER_CAN_CLOSING_N, id);
@@ -326,13 +518,56 @@
         ids.add(id);
         producer.sendCloseTrade(JSONObject.toJSONString(ids));
 
-        return Result.ok("平仓成功");
+        // 平仓成功
+        return Result.ok(MessageSourceUtils.getString("order_service_0044"));
+    }
+
+    @Override
+    public Result cancelHoldOrder(WholeCloseOrderDto wholeCloseOrderDto) {
+        MemberEntity memberEntity = LoginUserUtils.getAppLoginUser();
+        ContractHoldOrderEntity holdOrderEntity = contractHoldOrderDao.selectHoldOrderByMemberIdAndId(memberEntity.getId(), wholeCloseOrderDto.getId());
+        if (holdOrderEntity == null) {
+            // 订单不存在
+            return Result.fail(MessageSourceUtils.getString("order_service_0027"));
+        }
+
+        if (ContractHoldOrderEntity.ORDER_CAN_CLOSING_N == holdOrderEntity.getIsCanClosing()) {
+            // 该订单暂不能平仓
+            return Result.fail(MessageSourceUtils.getString("order_service_0045"));
+        }
+
+        if (wholeCloseOrderDto.getCount() > holdOrderEntity.getSymbolCntSale()) {
+            return Result.fail("可平张数不足");
+        }
+
+        log.info("--->{},{}", holdOrderEntity.getSymbolCntSale(), wholeCloseOrderDto.getCount());
+        int sub = holdOrderEntity.getSymbolCntSale() - wholeCloseOrderDto.getCount();
+        log.info("剩余张数:{}", sub);
+        // 判断当前持仓张数是否等于平仓张数,若等于则直接更新该订单为不可平状态,否则更新可平张数
+        if (sub == 0) {
+            holdOrderEntity.setIsCanClosing(ContractHoldOrderEntity.ORDER_CAN_CLOSING_N);
+        }
+
+        // 更新可平张数
+        holdOrderEntity.setSymbolCntSale(sub);
+        contractHoldOrderDao.updateById(holdOrderEntity);
+
+        // 将待平张数放入缓存
+        redisUtils.set(AppContants.CLOSING_ORDER_PREFIX + holdOrderEntity.getId(), wholeCloseOrderDto.getCount(), 30);
+
+        // 发送平仓消息
+        List<Long> ids = new ArrayList<>();
+        ids.add(wholeCloseOrderDto.getId());
+        producer.sendCloseTrade(JSONObject.toJSONString(ids));
+
+        // 平仓成功
+        return Result.ok(MessageSourceUtils.getString("order_service_0044"));
     }
 
     @Override
     public Result cancelHoldOrderBatch(SymbolDto symbolDto) {
         MemberEntity memberEntity = LoginUserUtils.getAppLoginUser();
-        List<ContractHoldOrderEntity> holdOrderEntities = contractHoldOrderDao.selectHoldOrderListByMemberIdAndSymbol(memberEntity.getId(), symbolDto.getSymbol());
+        List<ContractHoldOrderEntity> holdOrderEntities = contractHoldOrderDao.selectHoldOrderListByMemberIdAndSymbol(memberEntity.getId(), symbolDto.getSymbol(), 1);
         if (CollUtil.isEmpty(holdOrderEntities)) {
             return Result.fail("订单不存在");
         }
@@ -425,23 +660,23 @@
             if (ContractHoldOrderEntity.OPENING_TYPE_MORE == holdOrderEntity.getOpeningType()) {
                 // 开多止盈
                 if (stopProfitPrice != null) {
-                    model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.CLOSE_MORE_STOP_PROFIT.getValue(), stopProfitPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol());
+                    model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.CLOSE_MORE_STOP_PROFIT.getValue(), stopProfitPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol(), holdOrderEntity.getMemberId());
                     producer.sendPriceOperate(JSONObject.toJSONString(model));
                 }
                 // 开多止损
                 if (stopLessPrice != null) {
-                    model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.CLOSE_MORE_STOP_LESS.getValue(), stopLessPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol());
+                    model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.CLOSE_MORE_STOP_LESS.getValue(), stopLessPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol(), holdOrderEntity.getMemberId());
                     producer.sendPriceOperate(JSONObject.toJSONString(model));
                 }
             } else {
                 // 开空止盈
                 if (stopProfitPrice != null) {
-                    model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.CLOSE_LESS_STOP_PROFIT.getValue(), stopProfitPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol());
+                    model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.CLOSE_LESS_STOP_PROFIT.getValue(), stopProfitPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol(), holdOrderEntity.getMemberId());
                     producer.sendPriceOperate(JSONObject.toJSONString(model));
                 }
                 // 开空止损
                 if (stopLessPrice != null) {
-                    model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.CLOSE_LESS_STOP_LESS.getValue(), stopLessPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol());
+                    model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.CLOSE_LESS_STOP_LESS.getValue(), stopLessPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol(), holdOrderEntity.getMemberId());
                     producer.sendPriceOperate(JSONObject.toJSONString(model));
                 }
             }
@@ -485,7 +720,7 @@
 
         if (i > 0) {
             // 发送爆仓消息
-            sendOrderBombMsg(holdOrderEntity.getId(), holdOrderEntity.getOpeningType(), forceClosingPrice, holdOrderEntity.getSymbol(), holdOrderEntity.getOperateNo());
+            sendOrderBombMsg(holdOrderEntity.getId(), holdOrderEntity.getOpeningType(), forceClosingPrice, holdOrderEntity.getSymbol(), holdOrderEntity.getOperateNo(), holdOrderEntity.getMemberId());
             return Result.ok("调整成功");
         }
         return Result.fail("调整失败");
@@ -498,11 +733,18 @@
         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());
+        MemberWalletContractEntity walletContractEntity = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeEnum.USDT.name());
+//        if (memberEntity.getContractPositionType().equals(ContractEntrustOrderEntity.POSITION_TYPE_ADD)) {
+//            // 当前合约委托单
+//            entrustOrderEntities = contractEntrustOrderDao.selectEntrustOrderListByMemberId(memberEntity.getId());
+//            // 当前持仓列表
+//            holdOrderEntities = contractHoldOrderDao.selectHoldOrderListByMemberId(memberEntity.getId());
+//        } else {
+//            entrustOrderEntities = contractEntrustOrderDao.selectEntrustOrderListByMemberIdAndSymbol(memberEntity.getId(), symbol);
+//            holdOrderEntities = contractHoldOrderDao.selectHoldOrderListForWholeByMemberIdAndSymbol(memberEntity.getId(), symbol);
+//        }
 
         // 冻结保证金 -- 即委托单中的保证金之和
         BigDecimal frozenBondAmount = BigDecimal.ZERO;
@@ -527,10 +769,10 @@
                 BigDecimal profitOrLess = BigDecimal.ZERO;
                 // 开多
                 if (ContractHoldOrderEntity.OPENING_TYPE_MORE == holdOrderEntity.getOpeningType()) {
-                    profitOrLess = newPrice.subtract(holdOrderEntity.getOpeningPrice()).multiply(new BigDecimal(holdOrderEntity.getSymbolCnt())).multiply(lotNumber);
+                    profitOrLess = newPrice.subtract(holdOrderEntity.getOpeningPrice()).multiply(new BigDecimal(holdOrderEntity.getSymbolCntSale())).multiply(lotNumber);
                     // 开空
                 } else {
-                    profitOrLess = holdOrderEntity.getOpeningPrice().subtract(newPrice).multiply(new BigDecimal(holdOrderEntity.getSymbolCnt())).multiply(lotNumber);
+                    profitOrLess = holdOrderEntity.getOpeningPrice().subtract(newPrice).multiply(new BigDecimal(holdOrderEntity.getSymbolCntSale())).multiply(lotNumber);
                 }
 
                 if (MemberEntity.IS_PROFIT_Y == memberEntity.getIsProfit()) {
@@ -545,18 +787,19 @@
             }
         }
 
-        MemberWalletContractEntity walletContractEntity = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeEnum.USDT.name());
-
         MemberLevelRateEntity rateEntity = memberLevelRateDao.selectLeverRateByMemberIdAndSymbol(memberEntity.getId(), symbol);
 
+        ContractMoneyInfoVo contractMoneyInfoVo = new ContractMoneyInfoVo();
         // 权益
         BigDecimal equity = walletContractEntity.getTotalBalance().add(totalProfitOrLess);
+        if (equity.compareTo(BigDecimal.ZERO) <= 0) {
+            equity = BigDecimal.ZERO;
+        }
 
-        ContractMoneyInfoVo contractMoneyInfoVo = new ContractMoneyInfoVo();
-        contractMoneyInfoVo.setAvailableBalance(walletContractEntity.getAvailableBalance());
         contractMoneyInfoVo.setBeUsedBondAmount(beUsedBondAmount);
         contractMoneyInfoVo.setFrozenBondAmount(frozenBondAmount);
         contractMoneyInfoVo.setEquity(equity);
+        contractMoneyInfoVo.setAvailableBalance(walletContractEntity.getAvailableBalance());
         contractMoneyInfoVo.setFeeRatio(tradeSetting.getFeeRatio());
         contractMoneyInfoVo.setLeverAgeRatio(tradeSetting.getLeverageRatio());
         contractMoneyInfoVo.setNewPrice(newPriceSymbol);
@@ -568,6 +811,11 @@
     @Override
     public Result changeLeverRate(ChangeLeverRateDto changeLeverRateDto) {
         MemberEntity memberEntity = LoginUserUtils.getAppLoginUser();
+
+        List<ContractHoldOrderEntity> holdOrders = contractHoldOrderDao.selectHoldOrderListForWholeByMemberIdAndSymbol(memberEntity.getId(), changeLeverRateDto.getSymbol());
+        if (CollUtil.isNotEmpty(holdOrders)) {
+            return Result.fail("存在持仓, 无法调整杠杆");
+        }
         MemberLevelRateEntity levelRateEntity = memberLevelRateDao.selectLeverRateByMemberIdAndSymbol(memberEntity.getId(), changeLeverRateDto.getSymbol());
         levelRateEntity.setLevelRateUp(changeLeverRateDto.getLeverRate());
         levelRateEntity.setLevelRateDown(changeLeverRateDto.getLeverRate());
@@ -648,7 +896,7 @@
                 } else {
                     BigDecimal available = walletContractEntity.getAvailableBalance();
                     BigDecimal lessAmount = thisTimeHold.subtract(available);
-                    MemberEntity memberEntity = memberDao.selectById(holdOrderEntity.getId());
+                    MemberEntity memberEntity = memberDao.selectById(holdOrderEntity.getMemberId());
                     memberWalletContractDao.increaseWalletContractBalanceById(available.negate(), available.negate(), null, walletContractEntity.getId());
 
                     BigDecimal newBondAmount = holdOrderEntity.getBondAmount().subtract(lessAmount);
@@ -662,21 +910,45 @@
                     contractHoldOrderDao.updateById(holdOrderEntity);
 
                     // 发送爆仓消息
-                    sendOrderBombMsg(holdOrderEntity.getId(), holdOrderEntity.getOpeningType(), newForcePrice, holdOrderEntity.getSymbol(), holdOrderEntity.getOperateNo());
+                    sendOrderBombMsg(holdOrderEntity.getId(), holdOrderEntity.getOpeningType(), newForcePrice, holdOrderEntity.getSymbol(), holdOrderEntity.getOperateNo(), holdOrderEntity.getMemberId());
                 }
             }
         }
     }
 
-    public void sendOrderBombMsg(Long id, int type, BigDecimal forceClosingPrice, String symbol, int operateNo) {
+    public void sendOrderBombMsg(Long id, int type, BigDecimal forceClosingPrice, String symbol, int operateNo, Long memberId) {
         OrderModel model = null;
         // 开多
         if (ContractHoldOrderEntity.OPENING_TYPE_MORE == type) {
-            model = new OrderModel(id, RabbitPriceTypeEnum.CLOSE_MORE_BOMB.getValue(), forceClosingPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), symbol, operateNo);
+            model = new OrderModel(id, RabbitPriceTypeEnum.CLOSE_MORE_BOMB.getValue(), forceClosingPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), symbol, operateNo, memberId);
             // 开空
         } else {
-            model = new OrderModel(id, RabbitPriceTypeEnum.CLOSE_LESS_BOMB.getValue(), forceClosingPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), symbol, operateNo);
+            model = new OrderModel(id, RabbitPriceTypeEnum.CLOSE_LESS_BOMB.getValue(), forceClosingPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), symbol, operateNo, memberId);
         }
         producer.sendPriceOperate(JSONObject.toJSONString(model));
     }
+
+    @Override
+    public Result changePositionType() {
+        MemberEntity member = LoginUserUtils.getAppLoginUser();
+
+        List<ContractHoldOrderEntity> list = contractHoldOrderDao.selectHoldOrderListByMemberId(member.getId());
+        List<ContractEntrustOrderEntity> entrustList = contractEntrustOrderDao.selectEntrustOrderListByMemberId(member.getId());
+
+        if (CollUtil.isNotEmpty(list) || CollUtil.isNotEmpty(entrustList)) {
+            return Result.fail("存在持仓/委托, 无法更改");
+        }
+
+        Integer positionType = member.getContractPositionType() == ContractEntrustOrderEntity.POSITION_TYPE_ADD ? ContractEntrustOrderEntity.POSITION_TYPE_ALL : ContractEntrustOrderEntity.POSITION_TYPE_ADD;
+        MemberEntity updateEntity = new MemberEntity();
+        updateEntity.setContractPositionType(positionType);
+        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"));
+        }
+        return Result.fail(MessageSourceUtils.getString("member_service_0041"));
+    }
 }

--
Gitblit v1.9.1