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 |  455 +++++++++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 380 insertions(+), 75 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 d3b472d..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
@@ -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;
@@ -22,13 +23,15 @@
 import com.xcong.excoin.modules.contract.parameter.dto.*;
 import com.xcong.excoin.modules.contract.parameter.vo.*;
 import com.xcong.excoin.modules.contract.service.ContractHoldOrderService;
+import com.xcong.excoin.modules.documentary.dao.FollowFollowerOrderRelationDao;
+import com.xcong.excoin.modules.documentary.dao.FollowTraderInfoDao;
+import com.xcong.excoin.modules.documentary.entity.FollowFollowerOrderRelationEntity;
+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;
@@ -41,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.*;
@@ -82,53 +86,68 @@
 
     @Resource
     private MemberDao memberDao;
+    @Resource
+    private MemberSettingDao memberSettingDao;
+
+    @Resource
+    private FollowTraderInfoDao followTraderInfoDao;
+    @Resource
+    private FollowFollowerOrderRelationDao followFollowerOrderRelationDao;
 
     @Transactional(rollbackFor = Exception.class)
     @Override
     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());
-
+        MemberWalletContractEntity walletContract = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeConvert.convertContractTypeToCoin(submitOrderDto.getSymbol()));
         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);
 
         // 预付款为 --> 保证金+开仓手续费+平仓手续费 (开仓平仓手续费相同)
@@ -136,20 +155,168 @@
         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() + 1);
+            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"));
         }
 
         // 预估强平价
         BigDecimal forceClosingPrice = CalculateUtil.getForceSetPrice(bondAmount, openingPrice, submitOrderDto.getSymbolCnt(), lotNumber, submitOrderDto.getOrderType(), memberEntity);
         log.info("强平价:{}", forceClosingPrice);
 
+        // 判断是否开启了带单
+        boolean isOpenFollow = false;
+        FollowTraderInfoEntity tradeInfo = null;
+        if (MemberEntity.IS_TRADER_Y.equals(memberEntity.getIsTrader())) {
+            tradeInfo = followTraderInfoDao.selectTraderInfoByMemberId(memberEntity.getId());
+            if (tradeInfo.getIsOpen().equals(FollowTraderInfoEntity.ISOPEN_Y)) {
+                isOpenFollow = true;
+            }
+        }
+
         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);
@@ -160,6 +327,9 @@
         holdOrderEntity.setIsCanClosing(ContractHoldOrderEntity.ORDER_CAN_CLOSING_Y);
         holdOrderEntity.setPrePaymentAmount(prePaymentAmount);
         holdOrderEntity.setBondAmount(bondAmount.add(openFeePrice));
+        holdOrderEntity.setOperateNo(1);
+        // 设置合约类型
+        holdOrderEntity.setContractType(isOpenFollow ? ContractOrderEntity.CONTRACTTYPE_DOCUMENTARY : ContractOrderEntity.CONTRACTTYPE_NORMAL);
 
         ContractOrderEntity contractOrderEntity = ContractHoldOrderEntityMapper.INSTANCE.holdOrderToOrder(holdOrderEntity);
         contractOrderEntity.setOpeningTime(new Date());
@@ -168,6 +338,10 @@
 
         if (i > 0) {
             memberWalletContractDao.increaseWalletContractBalanceById(prePaymentAmount.negate(), openFeePrice.negate(), null, walletContract.getId());
+
+            // 发送爆仓消息
+            sendOrderBombMsg(holdOrderEntity.getId(), holdOrderEntity.getOpeningType(), forceClosingPrice, holdOrderEntity.getSymbol(), holdOrderEntity.getOperateNo());
+
             // 计算佣金
             ThreadPoolUtils.calReturnMoney(memberEntity.getId(), contractOrderEntity.getOpeningFeeAmount(), contractOrderEntity, AgentReturnEntity.ORDER_TYPE_OPEN);
 
@@ -177,9 +351,62 @@
             } else {
                 LogRecordUtils.insertMemberAccountFlow(memberEntity.getId(), prePaymentAmount, walletContract.getAvailableBalance().subtract(prePaymentAmount), submitOrderDto.getSymbol(), "买跌持仓", "买跌:" + submitOrderDto.getSymbol());
             }
-            return Result.ok("提交成功");
+
+            // 若该用户为交易员且开启带单模式,则发送带单异步
+            if (isOpenFollow) {
+                FollowFollowerOrderRelationEntity relationEntity = new FollowFollowerOrderRelationEntity();
+                relationEntity.setIsShow(FollowFollowerOrderRelationEntity.IS_SHOW_Y);
+                relationEntity.setMemberId(holdOrderEntity.getMemberId());
+                relationEntity.setOrderId(holdOrderEntity.getId());
+                relationEntity.setOrderType(FollowFollowerOrderRelationEntity.ORDER_TYPE_HOLD);
+                relationEntity.setTradeId(tradeInfo.getId());
+                relationEntity.setTradeMemberId(tradeInfo.getMemberId());
+                relationEntity.setTradeOrderNo(holdOrderEntity.getOrderNo());
+                followFollowerOrderRelationDao.insert(relationEntity);
+
+                ThreadPoolUtils.sendFollowOrderTask(holdOrderEntity.getId());
+            }
+            // 提交成功
+            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
@@ -198,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;
@@ -228,7 +455,7 @@
                     if (rewardRatio.compareTo(BigDecimal.ZERO) > -1) {
                         rewardRatio = rewardRatio.multiply(BigDecimal.ONE.subtract(tradeSettingEntity.getProfitParam()));
                     } else {
-                        rewardRatio = rewardRatio.multiply(BigDecimal.ONE.add(tradeSettingEntity.getProfitParam()));
+//                        rewardRatio = rewardRatio.multiply(BigDecimal.ONE.add(tradeSettingEntity.getProfitParam()));
                     }
                 }
 
@@ -261,7 +488,7 @@
 
             Map<String, Object> result = new HashMap<>();
             result.put("hold", resultList);
-            result.put("totalProfitOrLoss", totalProfitOrLoss);
+            result.put("totalProfitOrLoss", totalProfitOrLoss.setScale(4, BigDecimal.ROUND_DOWN).toPlainString());
             return Result.ok(result);
         }
         return Result.ok("success");
@@ -272,11 +499,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);
@@ -285,13 +514,57 @@
         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);
+        holdOrderEntity.setSymbolCnt(sub);
+        contractHoldOrderDao.updateById(holdOrderEntity);
+
+        // 将待平张数放入缓存
+        redisUtils.set(AppContants.CLOSING_ORDER_PREFIX + holdOrderEntity.getId(), wholeCloseOrderDto.getCount());
+
+        // 发送平仓消息
+        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,14 +698,14 @@
             if (changeBondDto.getAmount().compareTo(walletContract.getAvailableBalance()) > 0) {
                 return Result.fail("可用余额不足");
             }
-            memberWalletContractDao.increaseWalletContractBalanceById(changeBondDto.getAmount().negate(), null, changeBondDto.getAmount(), walletContract.getId());
+            memberWalletContractDao.increaseWalletContractBalanceById(changeBondDto.getAmount().negate(), null, null, walletContract.getId());
             holdOrderEntity.setBondAmount(holdOrderEntity.getBondAmount().add(changeBondDto.getAmount()));
             // 减少保证金
         } else {
             if (holdOrderEntity.getBondAmount().subtract(holdOrderEntity.getPrePaymentAmount()).subtract(changeBondDto.getAmount()).compareTo(BigDecimal.ZERO) < 0) {
                 return Result.fail("超出保证金最大减少金额");
             }
-            memberWalletContractDao.increaseWalletContractBalanceById(changeBondDto.getAmount(), null, changeBondDto.getAmount().negate(), walletContract.getId());
+            memberWalletContractDao.increaseWalletContractBalanceById(changeBondDto.getAmount(), null, null, walletContract.getId());
             holdOrderEntity.setBondAmount(holdOrderEntity.getBondAmount().subtract(changeBondDto.getAmount()));
         }
 
@@ -443,15 +716,8 @@
         int i = contractHoldOrderDao.updateById(holdOrderEntity);
 
         if (i > 0) {
-            OrderModel model = null;
-            // 开多
-            if (ContractHoldOrderEntity.OPENING_TYPE_MORE == holdOrderEntity.getOpeningType()) {
-                model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.CLOSE_MORE_BOMB.getValue(), forceClosingPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol());
-                // 开空
-            } else {
-                model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.CLOSE_LESS_BOMB.getValue(), forceClosingPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol());
-            }
-            producer.sendPriceOperate(JSONObject.toJSONString(model));
+            // 发送爆仓消息
+            sendOrderBombMsg(holdOrderEntity.getId(), holdOrderEntity.getOpeningType(), forceClosingPrice, holdOrderEntity.getSymbol(), holdOrderEntity.getOperateNo());
             return Result.ok("调整成功");
         }
         return Result.fail("调整失败");
@@ -464,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;
@@ -511,8 +786,6 @@
             }
         }
 
-        MemberWalletContractEntity walletContractEntity = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeEnum.USDT.name());
-
         MemberLevelRateEntity rateEntity = memberLevelRateDao.selectLeverRateByMemberIdAndSymbol(memberEntity.getId(), symbol);
 
         // 权益
@@ -554,7 +827,8 @@
 
         HoldOrderDetailVo holdOrderDetailVo = ContractHoldOrderEntityMapper.INSTANCE.holdOrderToOrderDetailVo(holdOrderEntity);
         BigDecimal feeSpread = cacheSettingUtils.getTradeSetting().getFeeSpreadRatio();
-        holdOrderDetailVo.setOpeningFeeAmount(holdOrderDetailVo.getOpeningFeeAmount().multiply(feeSpread).setScale(8, BigDecimal.ROUND_DOWN));
+//        holdOrderDetailVo.setOpeningFeeAmount(holdOrderDetailVo.getOpeningFeeAmount().multiply(feeSpread).setScale(8, BigDecimal.ROUND_DOWN));
+        holdOrderDetailVo.setOpeningFeeAmount(holdOrderEntity.getOpeningFeeAmount(), feeSpread);
         return Result.ok(holdOrderDetailVo);
     }
 
@@ -580,8 +854,10 @@
 
         OrderDetailVo orderDetailVo = ContractOrderEntityMapper.INSTANCE.entityToDetailVo(contractOrderEntity);
         BigDecimal feeSpread = cacheSettingUtils.getTradeSetting().getFeeSpreadRatio();
-        orderDetailVo.setClosingFeeAmount(orderDetailVo.getClosingFeeAmount() == null ? orderDetailVo.getClosingFeeAmount() : orderDetailVo.getClosingFeeAmount().multiply(feeSpread).setScale(8, BigDecimal.ROUND_DOWN));
-        orderDetailVo.setOpeningFeeAmount(orderDetailVo.getOpeningFeeAmount().multiply(feeSpread).setScale(8, BigDecimal.ROUND_DOWN));
+//        orderDetailVo.setClosingFeeAmount(orderDetailVo.getClosingFeeAmount() == null ? orderDetailVo.getClosingFeeAmount() : orderDetailVo.getClosingFeeAmount().multiply(feeSpread).setScale(8, BigDecimal.ROUND_DOWN));
+//        orderDetailVo.setOpeningFeeAmount(orderDetailVo.getOpeningFeeAmount() == null ? orderDetailVo.getOpeningFeeAmount() : orderDetailVo.getOpeningFeeAmount().multiply(feeSpread).setScale(8, BigDecimal.ROUND_DOWN));
+        orderDetailVo.setOpeningFeeAmount(contractOrderEntity.getOpeningFeeAmount(), feeSpread);
+        orderDetailVo.setClosingFeeAmount(contractOrderEntity.getClosingFeeAmount(), feeSpread);
         return Result.ok(orderDetailVo);
     }
 
@@ -624,17 +900,46 @@
                     holdOrderEntity.setOperateNo(holdOrderEntity.getOperateNo() + 1);
                     contractHoldOrderDao.updateById(holdOrderEntity);
 
-                    OrderModel model = null;
-                    // 开多
-                    if (ContractHoldOrderEntity.OPENING_TYPE_MORE == holdOrderEntity.getOpeningType()) {
-                        model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.CLOSE_MORE_BOMB.getValue(), newForcePrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol());
-                        // 开空
-                    } else {
-                        model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.CLOSE_LESS_BOMB.getValue(), newForcePrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol());
-                    }
-                    producer.sendPriceOperate(JSONObject.toJSONString(model));
+                    // 发送爆仓消息
+                    sendOrderBombMsg(holdOrderEntity.getId(), holdOrderEntity.getOpeningType(), newForcePrice, holdOrderEntity.getSymbol(), holdOrderEntity.getOperateNo());
                 }
             }
         }
     }
+
+    public void sendOrderBombMsg(Long id, int type, BigDecimal forceClosingPrice, String symbol, int operateNo) {
+        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);
+            // 开空
+        } else {
+            model = new OrderModel(id, RabbitPriceTypeEnum.CLOSE_LESS_BOMB.getValue(), forceClosingPrice.setScale(8, RoundingMode.HALF_UP).toPlainString(), symbol, operateNo);
+        }
+        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