From ebb754075f9fd70e214b2938d8ef90acf8d6f747 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 26 Aug 2020 15:13:33 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/modules/contract/service/impl/ContractEntrustOrderServiceImpl.java |   56 +++++++++++++++++++++++++++++++++++++++-----------------
 1 files changed, 39 insertions(+), 17 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/contract/service/impl/ContractEntrustOrderServiceImpl.java b/src/main/java/com/xcong/excoin/modules/contract/service/impl/ContractEntrustOrderServiceImpl.java
index 97b8a07..ca87cc2 100644
--- a/src/main/java/com/xcong/excoin/modules/contract/service/impl/ContractEntrustOrderServiceImpl.java
+++ b/src/main/java/com/xcong/excoin/modules/contract/service/impl/ContractEntrustOrderServiceImpl.java
@@ -1,5 +1,6 @@
 package com.xcong.excoin.modules.contract.service.impl;
 
+import cn.hutool.core.collection.CollUtil;
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.xcong.excoin.common.LoginUserUtils;
@@ -251,7 +252,12 @@
             return Result.fail(MessageSourceUtils.getString("entrust_order_not_exist"));
         }
 
-        MemberWalletContractEntity walletContractEntity = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), MemberWalletCoinEnum.WALLETCOINCODE.getValue());
+        MemberWalletContractEntity walletContractEntity;
+        if (ContractEntrustOrderEntity.POSITION_TYPE_ADD == entrustOrderEntity.getPositionType()) {
+            walletContractEntity = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), MemberWalletCoinEnum.WALLETCOINCODE.getValue());
+        } else {
+            walletContractEntity = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeConvert.convertContractTypeToCoin(entrustOrderEntity.getSymbol()));
+        }
 
         BigDecimal total = entrustOrderEntity.getEntrustAmount();
         memberWalletContractDao.increaseWalletContractBalanceById(total, null, entrustOrderEntity.getBondAmount().negate(), walletContractEntity.getId());
@@ -288,28 +294,42 @@
             return Result.fail("可平张数不足");
         }
 
+        List<ContractEntrustOrderEntity> entrustOrders = contractEntrustOrderDao.selectEntrustOrderByOrderNo(holdOrderEntity.getOrderNo());
+        if (CollUtil.isNotEmpty(entrustOrders)) {
+            int entrustCnt = 0;
+            for (ContractEntrustOrderEntity entrustOrder : entrustOrders) {
+                entrustCnt += entrustOrder.getSymbolCnt();
+            }
+
+
+            if (submitCloseEntrustDto.getSymbolCnt() > holdOrderEntity.getSymbolCntSale() - entrustCnt) {
+                return Result.fail("当前存在多个委托,无法提交该数量委托");
+            }
+        }
+
+
         // 获取最新价
-        BigDecimal newPrice = new BigDecimal(redisUtils.getString(CoinTypeConvert.convertToKey(submitCloseEntrustDto.getSymbol())));
+        BigDecimal newPrice = new BigDecimal(redisUtils.getString(CoinTypeConvert.convertToKey(holdOrderEntity.getSymbol())));
 
         int closeType;
         if (holdOrderEntity.getOpeningType() == ContractHoldOrderEntity.OPENING_TYPE_MORE) {
             // 开多, 若委托价低于当前价则立即成交
-            if (submitCloseEntrustDto.getEntrustPrice().compareTo(newPrice) <= 0) {
-                WholeCloseOrderDto wholeCloseOrderDto = new WholeCloseOrderDto();
-                wholeCloseOrderDto.setCount(submitCloseEntrustDto.getSymbolCnt());
-                wholeCloseOrderDto.setId(submitCloseEntrustDto.getId());
-                return contractHoldOrderService.cancelHoldOrder(wholeCloseOrderDto);
-            }
+//            if (submitCloseEntrustDto.getEntrustPrice().compareTo(newPrice) <= 0) {
+//                WholeCloseOrderDto wholeCloseOrderDto = new WholeCloseOrderDto();
+//                wholeCloseOrderDto.setCount(submitCloseEntrustDto.getSymbolCnt());
+//                wholeCloseOrderDto.setId(submitCloseEntrustDto.getId());
+//                return contractHoldOrderService.cancelHoldOrder(wholeCloseOrderDto);
+//            }
 
             closeType = ContractOrderEntity.ORDER_TYPE_CLOSE_MORE;
         } else {
             // 开空, 若委托价高于当前价则立即成交
-            if (submitCloseEntrustDto.getEntrustPrice().compareTo(newPrice) >= 0) {
-                WholeCloseOrderDto wholeCloseOrderDto = new WholeCloseOrderDto();
-                wholeCloseOrderDto.setCount(submitCloseEntrustDto.getSymbolCnt());
-                wholeCloseOrderDto.setId(submitCloseEntrustDto.getId());
-                return contractHoldOrderService.cancelHoldOrder(wholeCloseOrderDto);
-            }
+//            if (submitCloseEntrustDto.getEntrustPrice().compareTo(newPrice) >= 0) {
+//                WholeCloseOrderDto wholeCloseOrderDto = new WholeCloseOrderDto();
+//                wholeCloseOrderDto.setCount(submitCloseEntrustDto.getSymbolCnt());
+//                wholeCloseOrderDto.setId(submitCloseEntrustDto.getId());
+//                return contractHoldOrderService.cancelHoldOrder(wholeCloseOrderDto);
+//            }
 
             closeType = ContractOrderEntity.ORDER_TYPE_CLOSE_LESS;
         }
@@ -317,7 +337,7 @@
 
         MemberWalletContractEntity walletContract = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(member.getId(), CoinTypeConvert.convertContractTypeToCoin(holdOrderEntity.getSymbol()));
 
-        BigDecimal lotNumber = cacheSettingUtils.getSymbolSku(submitCloseEntrustDto.getSymbol());
+        BigDecimal lotNumber = cacheSettingUtils.getSymbolSku(holdOrderEntity.getSymbol());
         BigDecimal bondAmount = CalculateUtil.getBondAmount(holdOrderEntity.getOpeningPrice(), lotNumber, submitCloseEntrustDto.getSymbolCnt(), holdOrderEntity.getLeverRatio());
 
         ContractEntrustOrderEntity entrustOrder = new ContractEntrustOrderEntity();
@@ -329,6 +349,8 @@
         entrustOrder.setLeverRatio(holdOrderEntity.getLeverRatio());
         entrustOrder.setSymbolSku(holdOrderEntity.getSymbolSku());
         entrustOrder.setEntrustType(closeType);
+        entrustOrder.setSymbolCnt(holdOrderEntity.getSymbolCnt());
+        entrustOrder.setSymbol(holdOrderEntity.getSymbol());
 
         int i = contractEntrustOrderDao.insert(entrustOrder);
 
@@ -338,10 +360,10 @@
         if (i > 0) {
             // 发送委托消息
             if (holdOrderEntity.getOpeningType() == ContractEntrustOrderEntity.ENTRUST_TYPE_OPEN_MORE) {
-                OrderModel model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.ENTRUST_OPEN_MORE.getValue(), submitCloseEntrustDto.getEntrustPrice().setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol());
+                OrderModel model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.ENTRUST_CLOSE_MORE.getValue(), submitCloseEntrustDto.getEntrustPrice().setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol());
                 producer.sendPriceOperate(JSONObject.toJSONString(model));
             } else {
-                OrderModel model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.ENTRUST_OPEN_LESS.getValue(), submitCloseEntrustDto.getEntrustPrice().setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol());
+                OrderModel model = new OrderModel(holdOrderEntity.getId(), RabbitPriceTypeEnum.ENTRUST_CLOSE_LESS.getValue(), submitCloseEntrustDto.getEntrustPrice().setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol());
                 producer.sendPriceOperate(JSONObject.toJSONString(model));
             }
         }

--
Gitblit v1.9.1