From 44a6418f8a497f57ad59ed7915b7b79b78b9ecd0 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 02 Sep 2020 18:25:42 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/modules/contract/service/impl/OrderWebsocketServiceImpl.java |  236 ++++++++++++++++++++++++++++++++++++++++++----------------
 1 files changed, 169 insertions(+), 67 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/contract/service/impl/OrderWebsocketServiceImpl.java b/src/main/java/com/xcong/excoin/modules/contract/service/impl/OrderWebsocketServiceImpl.java
index 1489125..c6d461b 100644
--- a/src/main/java/com/xcong/excoin/modules/contract/service/impl/OrderWebsocketServiceImpl.java
+++ b/src/main/java/com/xcong/excoin/modules/contract/service/impl/OrderWebsocketServiceImpl.java
@@ -1,8 +1,11 @@
 package com.xcong.excoin.modules.contract.service.impl;
 
+import cn.hutool.core.collection.CollUtil;
 import com.alibaba.fastjson.JSONObject;
+import com.xcong.excoin.common.contants.AppContants;
 import com.xcong.excoin.modules.coin.dao.MemberAccountFlowEntityDao;
 import com.xcong.excoin.modules.coin.entity.MemberAccountFlowEntity;
+import com.xcong.excoin.modules.contract.dao.ContractEntrustOrderDao;
 import com.xcong.excoin.modules.contract.dao.ContractHoldOrderDao;
 import com.xcong.excoin.modules.contract.entity.ContractEntrustOrderEntity;
 import com.xcong.excoin.modules.contract.entity.ContractHoldOrderEntity;
@@ -24,10 +27,7 @@
 import com.xcong.excoin.modules.platform.entity.PlatformTradeSettingEntity;
 import com.xcong.excoin.rabbit.pricequeue.OrderModel;
 import com.xcong.excoin.rabbit.producer.OrderProducer;
-import com.xcong.excoin.utils.CacheSettingUtils;
-import com.xcong.excoin.utils.CalculateUtil;
-import com.xcong.excoin.utils.CoinTypeConvert;
-import com.xcong.excoin.utils.ThreadPoolUtils;
+import com.xcong.excoin.utils.*;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.collections.CollectionUtils;
 import org.springframework.stereotype.Service;
@@ -75,6 +75,10 @@
     private MemberSettingDao memberSettingDao;
     @Resource
     private ContractHoldOrderDao contractHoldOrderDao;
+    @Resource
+    private ContractEntrustOrderDao contractEntrustOrderDao;
+    @Resource
+    private RedisUtils redisUtils;
 
 
     public void dealOrderFromMq(List<OrderModel> list, Integer type) {
@@ -130,6 +134,23 @@
 
         }
     }
+
+
+    /**
+     * 止盈/止损后删除全仓委托平仓单
+     *
+     * @param orderNo
+     */
+    private void deleteEntrustCloseOrder(String orderNo) {
+        // 若平掉所有张数,若存在委托平仓,则删除委托平仓记录
+        List<ContractEntrustOrderEntity> entrustOrders = contractEntrustOrderDao.selectEntrustOrderByOrderNo(orderNo);
+        if (CollUtil.isNotEmpty(entrustOrders)) {
+            for (ContractEntrustOrderEntity entrustOrder : entrustOrders) {
+                contractEntrustOrderDao.deleteById(entrustOrder.getId());
+            }
+        }
+    }
+
 
     /**
      * 开多止盈
@@ -224,6 +245,11 @@
 
                         //返佣
                         ThreadPoolUtils.calReturnMoney(order.getMemberId(), order.getOpeningFeeAmount(), contractOrderEntity, AgentReturnEntity.ORDER_TYPE_CLOSE);
+
+                        // 全仓模式,止盈/止损 删除委托平仓
+                        if (ContractEntrustOrderEntity.POSITION_TYPE_ALL == order.getPositionType()) {
+                            deleteEntrustCloseOrder(order.getOrderNo());
+                        }
                     }
                 }
             }
@@ -234,6 +260,7 @@
     /**
      * 开空止盈
      */
+    @Transactional(rollbackFor = Exception.class)
     public void dealForLessStopPro(List<ContractHoldOrderEntity> orderList, List<OrderModel> list) {
         //List<CoinsCoinsOrder> orderList = orderMapper.selectOrderByBatchNo(batchno);
         //System.out.println("开空止盈订单batchno:" + batchno);
@@ -322,6 +349,11 @@
 
                         //返佣
                         ThreadPoolUtils.calReturnMoney(order.getMemberId(), order.getOpeningFeeAmount(), contractOrderEntity, AgentReturnEntity.ORDER_TYPE_CLOSE);
+
+                        // 全仓模式,止盈/止损 删除委托平仓
+                        if (ContractEntrustOrderEntity.POSITION_TYPE_ALL == order.getPositionType()) {
+                            deleteEntrustCloseOrder(order.getOrderNo());
+                        }
                     }
                 }
             }
@@ -334,6 +366,7 @@
      *
      * @param
      */
+    @Transactional(rollbackFor = Exception.class)
     public void dealForMoreLoss(List<ContractHoldOrderEntity> orderList, List<OrderModel> list) {
         //List<CoinsCoinsOrder> orderList = orderMapper.selectOrderByBatchNo(batchno);
         //System.out.println("开多止损批次号batchno:" + batchno);
@@ -419,6 +452,11 @@
 
                         //返佣
                         ThreadPoolUtils.calReturnMoney(order.getMemberId(), order.getOpeningFeeAmount(), contractOrderEntity, AgentReturnEntity.ORDER_TYPE_CLOSE);
+
+                        // 全仓模式,止盈/止损 删除委托平仓
+                        if (ContractEntrustOrderEntity.POSITION_TYPE_ALL == order.getPositionType()) {
+                            deleteEntrustCloseOrder(order.getOrderNo());
+                        }
                     }
                 }
             }
@@ -430,6 +468,7 @@
      *
      * @param
      */
+    @Transactional(rollbackFor = Exception.class)
     public void dealForLessLoss(List<ContractHoldOrderEntity> orderList, List<OrderModel> list) {
         // List<CoinsCoinsOrder> orderList = orderMapper.selectOrderByBatchNo(batchno);
         //System.out.println("开空止损批次号batchno:" + batchno);
@@ -515,6 +554,11 @@
 
                         //返佣
                         ThreadPoolUtils.calReturnMoney(order.getMemberId(), order.getOpeningFeeAmount(), contractOrderEntity, AgentReturnEntity.ORDER_TYPE_CLOSE);
+
+                        // 全仓模式,止盈/止损 删除委托平仓
+                        if (ContractEntrustOrderEntity.POSITION_TYPE_ALL == order.getPositionType()) {
+                            deleteEntrustCloseOrder(order.getOrderNo());
+                        }
                     }
                 }
             }
@@ -607,11 +651,11 @@
                 // 市价
                 if (coinsCoinsOrder.getEntrustType() == 1) {
                     // 开多
-                    OrderModel model = new OrderModel(contractHoldOrderEntity.getId(), 6, contractHoldOrderEntity.getForceClosingPrice().setScale(8, RoundingMode.HALF_UP).toPlainString(), coinsCoinsOrder.getSymbol(), 1);
+                    OrderModel model = new OrderModel(contractHoldOrderEntity.getId(), 6, contractHoldOrderEntity.getForceClosingPrice().setScale(8, RoundingMode.HALF_UP).toPlainString(), coinsCoinsOrder.getSymbol(), 1, coinsCoinsOrder.getMemberId());
                     producer.sendPriceOperate(JSONObject.toJSONString(model));
                 } else {
                     // 开空
-                    OrderModel model = new OrderModel(contractHoldOrderEntity.getId(), 7, contractHoldOrderEntity.getForceClosingPrice().setScale(8, RoundingMode.HALF_UP).toPlainString(), coinsCoinsOrder.getSymbol(), 1);
+                    OrderModel model = new OrderModel(contractHoldOrderEntity.getId(), 7, contractHoldOrderEntity.getForceClosingPrice().setScale(8, RoundingMode.HALF_UP).toPlainString(), coinsCoinsOrder.getSymbol(), 1, coinsCoinsOrder.getMemberId());
                     producer.sendPriceOperate(JSONObject.toJSONString(model));
                 }
                 // 扣除手续费
@@ -627,6 +671,21 @@
 
     private void wholeEntrustOrder(ContractEntrustOrderEntity entrustOrder) {
         log.info("全仓委托逻辑处理:{}", entrustOrder.getOrderNo());
+
+        int i = 1;
+        while (true) {
+            List<Object> types = redisUtils.lGet(AppContants.MEMBER_TYPE + entrustOrder.getMemberId(), 0, -1);
+            if (CollUtil.isEmpty(types)) {
+                break;
+            }
+            log.info("存在止盈/止损/爆仓/委托平仓,等待 -- {}", i);
+            try {
+                Thread.sleep(1000);
+            } catch (InterruptedException e) {
+                e.printStackTrace();
+            }
+            i++;
+        }
         MemberWalletContractEntity wallet = memberWalletContractService.findWalletContractByMemberIdAndSymbol(entrustOrder.getMemberId(), CoinTypeConvert.convertContractTypeToCoin(entrustOrder.getSymbol()));
 
         PlatformTradeSettingEntity tradeSettingEntity = cacheSettingUtils.getTradeSetting();
@@ -730,6 +789,11 @@
                 modelMap.put(model.getOrderId(), model.getOperateNo());
             }
             for (ContractHoldOrderEntity coinsOrder : orderList) {
+                List<Object> types = redisUtils.lGet(AppContants.RABBIT_TYPE + coinsOrder.getId(), 0, -1);
+                if(types.contains(9) || types.contains(10) || types.contains(11) || types.contains(12)) {
+                    log.info("存在止盈/止损 orderId : {}, types : {}", coinsOrder.getId(), JSONObject.toJSONString(types));
+                    continue;
+                }
                 Long orderId = coinsOrder.getId();
                 Integer operateNo = coinsOrder.getOperateNo();
                 //判断当前订单是否是最新的爆仓价 不相等时直接进入下一个订单
@@ -738,69 +802,17 @@
                     contractHoldOrderService.updateOrderIsCanClosingAndBatchNoById(orderId);
                     continue;
                 }
-                boolean isDone = false;
-                Long memId = coinsOrder.getMemberId();
-                BigDecimal nowPrice = coinsOrder.getForceClosingPrice();
-                // 创建订单(加入历史表的订单)
-                ContractOrderEntity contractOrderEntity = ContractHoldOrderEntityMapper.INSTANCE.holdOrderToOrder(coinsOrder);
 
-                //查询是否满足爆仓条件
-                if (coinsOrder.getOpeningType() == ContractHoldOrderEntity.OPENING_TYPE_MORE) {
-                    //如果是开多,当前价小于预估强平价即为爆仓
-                    // 设置平仓类型 // 爆仓平多
-                    contractOrderEntity.setClosingType(4);
-                    //更新用户钱包数据
-                    isDone = true;
+                if (ContractEntrustOrderEntity.POSITION_TYPE_ADD == coinsOrder.getPositionType()) {
+                    bombOrder(coinsOrder);
                 } else {
-                    //如果是开空,当前价大于预估强平价即为爆仓
-                    contractOrderEntity.setClosingType(5);
-                    //更新主表订单状态位为“已平仓”
-                    isDone = true;
+                    List<ContractHoldOrderEntity> holdOrderEntities = contractHoldOrderDao.selectHoldOrderListForWholeByMemberIdAndSymbol(coinsOrder.getMemberId(), coinsOrder.getSymbol());
+                    if (CollectionUtils.isNotEmpty(holdOrderEntities)) {
+                        for (ContractHoldOrderEntity holdOrderEntity : holdOrderEntities) {
+                            bombOrder(holdOrderEntity);
+                        }
 
-                }
-                if (isDone) {
-                    //删除次仓订单
-                    contractHoldOrderService.removeById(orderId);
-                    // 订单状态转换
-                    if (ContractOrderEntity.ORDER_TYPE_OPEN_MORE == contractOrderEntity.getOrderType()) {
-                        contractOrderEntity.setOrderType(ContractOrderEntity.ORDER_TYPE_CLOSE_MORE);
-                    } else {
-                        contractOrderEntity.setOrderType(ContractOrderEntity.ORDER_TYPE_CLOSE_LESS);
-                    }
-                    //更新主表订单状态位为“已平仓”
-                    contractOrderEntity.setId(null);
-                    contractOrderEntity.setClosingPrice(nowPrice);
-                    contractOrderEntity.setClosingTime(new Date());
-                    contractOrderEntity.setClosingFeeAmount(coinsOrder.getOpeningFeeAmount());
-
-                    if (coinsOrder.getPositionType() == ContractEntrustOrderEntity.POSITION_TYPE_ADD) {
-                        contractOrderEntity.setRewardAmount(coinsOrder.getBondAmount().subtract(contractOrderEntity.getOpeningFeeAmount()).negate());
-                        contractOrderService.save(contractOrderEntity);
-
-                        //更新用户钱包数据
-                        MemberWalletContractEntity usdt = memberWalletContractService.findWalletContractByMemberIdAndSymbol(memId, "USDT");
-
-                        // 减去的时候用负数
-                        BigDecimal totalPrice = coinsOrder.getBondAmount().negate();
-                        memberWalletContractService.increaseWalletContractBalanceById(null, totalPrice, null, usdt.getId());
-
-                        // 流水记录 TODO
-                        MemberAccountFlowEntity record = new MemberAccountFlowEntity();
-                        record.setCreateTime(new Date());
-                        record.setSource("系统自动平仓");
-                        record.setRemark("系统自动平仓");
-                        record.setBalance(usdt.getAvailableBalance());
-                        record.setMemberId(memId);
-                        record.setSymbol(coinsOrder.getSymbol());
-                        record.setPrice(coinsOrder.getBondAmount());
-                        memberAccountFlowEntityDao.insert(record);
-                    } else {
-                        MemberWalletContractEntity wallet = memberWalletContractService.findWalletContractByMemberIdAndSymbol(memId, CoinTypeConvert.convertContractTypeToCoin(coinsOrder.getSymbol()));
-
-                        contractOrderEntity.setRewardAmount(wallet.getTotalBalance().subtract(coinsOrder.getOpeningFeeAmount()).negate());
-                        contractOrderService.save(contractOrderEntity);
-
-                        memberWalletContractService.increaseWalletContractBalanceById(wallet.getAvailableBalance().negate(), wallet.getTotalBalance().negate(), null, wallet.getId());
+                        MemberWalletContractEntity wallet = memberWalletContractService.findWalletContractByMemberIdAndSymbol(coinsOrder.getMemberId(), CoinTypeConvert.convertContractTypeToCoin(coinsOrder.getSymbol()));
 
                         // 流水记录 TODO
                         MemberAccountFlowEntity record = new MemberAccountFlowEntity();
@@ -808,12 +820,102 @@
                         record.setSource("系统自动平仓");
                         record.setRemark("系统自动平仓");
                         record.setBalance(BigDecimal.ZERO);
-                        record.setMemberId(memId);
+                        record.setMemberId(coinsOrder.getMemberId());
                         record.setSymbol(coinsOrder.getSymbol());
                         record.setPrice(wallet.getTotalBalance().subtract(coinsOrder.getOpeningFeeAmount()));
                         memberAccountFlowEntityDao.insert(record);
                     }
                 }
+
+            }
+        }
+    }
+
+    private void bombOrder(ContractHoldOrderEntity coinsOrder) {
+        Long orderId = coinsOrder.getId();
+        boolean isDone = false;
+        Long memId = coinsOrder.getMemberId();
+        BigDecimal nowPrice = coinsOrder.getForceClosingPrice();
+        // 创建订单(加入历史表的订单)
+        ContractOrderEntity contractOrderEntity = ContractHoldOrderEntityMapper.INSTANCE.holdOrderToOrder(coinsOrder);
+
+        //查询是否满足爆仓条件
+        if (coinsOrder.getOpeningType() == ContractHoldOrderEntity.OPENING_TYPE_MORE) {
+            //如果是开多,当前价小于预估强平价即为爆仓
+            // 设置平仓类型 // 爆仓平多
+            contractOrderEntity.setClosingType(4);
+            //更新用户钱包数据
+            isDone = true;
+        } else {
+            //如果是开空,当前价大于预估强平价即为爆仓
+            contractOrderEntity.setClosingType(5);
+            //更新主表订单状态位为“已平仓”
+            isDone = true;
+
+        }
+        if (isDone) {
+            //删除次仓订单
+            contractHoldOrderService.removeById(orderId);
+            // 订单状态转换
+            if (ContractOrderEntity.ORDER_TYPE_OPEN_MORE == contractOrderEntity.getOrderType()) {
+                contractOrderEntity.setOrderType(ContractOrderEntity.ORDER_TYPE_CLOSE_MORE);
+            } else {
+                contractOrderEntity.setOrderType(ContractOrderEntity.ORDER_TYPE_CLOSE_LESS);
+            }
+            //更新主表订单状态位为“已平仓”
+            contractOrderEntity.setId(null);
+            contractOrderEntity.setClosingPrice(nowPrice);
+            contractOrderEntity.setClosingTime(new Date());
+            contractOrderEntity.setClosingFeeAmount(coinsOrder.getOpeningFeeAmount());
+
+            if (coinsOrder.getPositionType() == ContractEntrustOrderEntity.POSITION_TYPE_ADD) {
+                contractOrderEntity.setRewardAmount(coinsOrder.getBondAmount().subtract(contractOrderEntity.getOpeningFeeAmount()).negate());
+                contractOrderService.save(contractOrderEntity);
+
+                //更新用户钱包数据
+                MemberWalletContractEntity usdt = memberWalletContractService.findWalletContractByMemberIdAndSymbol(memId, "USDT");
+
+                // 减去的时候用负数
+                BigDecimal totalPrice = coinsOrder.getBondAmount().negate();
+                memberWalletContractService.increaseWalletContractBalanceById(null, totalPrice, null, usdt.getId());
+
+                // 流水记录 TODO
+                MemberAccountFlowEntity record = new MemberAccountFlowEntity();
+                record.setCreateTime(new Date());
+                record.setSource("系统自动平仓");
+                record.setRemark("系统自动平仓");
+                record.setBalance(usdt.getAvailableBalance());
+                record.setMemberId(memId);
+                record.setSymbol(coinsOrder.getSymbol());
+                record.setPrice(coinsOrder.getBondAmount());
+                memberAccountFlowEntityDao.insert(record);
+            } else {
+                MemberWalletContractEntity wallet = memberWalletContractService.findWalletContractByMemberIdAndSymbol(memId, CoinTypeConvert.convertContractTypeToCoin(coinsOrder.getSymbol()));
+
+//                BigDecimal rewardRatio = BigDecimal.ZERO;
+//                // 开多
+//                if (ContractHoldOrderEntity.OPENING_TYPE_MORE == coinsOrder.getOpeningType()) {
+//                    // (最新价-开仓价)*规格*张数
+//                    rewardRatio = nowPrice.subtract(coinsOrder.getOpeningPrice()).multiply(coinsOrder.getSymbolSku()).multiply(new BigDecimal(coinsOrder.getSymbolCntSale()));
+//                    // 开空
+//                } else {
+//                    // (开仓价-最新价)*规格*张数
+//                    rewardRatio = coinsOrder.getOpeningPrice().subtract(nowPrice).multiply(coinsOrder.getSymbolSku()).multiply(new BigDecimal(coinsOrder.getSymbolCntSale()));
+//                }
+
+                // 委托单总委托金额
+                BigDecimal totalPrePayment = BigDecimal.ZERO;
+                List<ContractEntrustOrderEntity> entrustOrderEntities = contractEntrustOrderDao.selectEntrustOrderListByMemberId(memId);
+                if (CollUtil.isNotEmpty(entrustOrderEntities)) {
+                    for (ContractEntrustOrderEntity entrustOrderEntity : entrustOrderEntities) {
+                        totalPrePayment = totalPrePayment.add(entrustOrderEntity.getEntrustAmount());
+                    }
+                }
+
+                contractOrderEntity.setRewardAmount(wallet.getTotalBalance().subtract(totalPrePayment).subtract(contractOrderEntity.getOpeningFeeAmount()).negate());
+                contractOrderService.save(contractOrderEntity);
+
+                memberWalletContractService.increaseWalletContractBalanceById(wallet.getAvailableBalance().negate(), wallet.getTotalBalance().subtract(totalPrePayment).negate(), null, wallet.getId());
             }
         }
     }

--
Gitblit v1.9.1