From 9ee9e0d815a1c02e0f37859757aa0322096cb94c Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Thu, 27 Aug 2020 17:22:09 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/modules/contract/service/impl/RabbitOrderServiceImpl.java |  151 +++++++++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 136 insertions(+), 15 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/contract/service/impl/RabbitOrderServiceImpl.java b/src/main/java/com/xcong/excoin/modules/contract/service/impl/RabbitOrderServiceImpl.java
index 726f020..8a1ca9b 100644
--- a/src/main/java/com/xcong/excoin/modules/contract/service/impl/RabbitOrderServiceImpl.java
+++ b/src/main/java/com/xcong/excoin/modules/contract/service/impl/RabbitOrderServiceImpl.java
@@ -6,6 +6,7 @@
 import com.xcong.excoin.common.enumerates.CoinTypeEnum;
 import com.xcong.excoin.common.enumerates.OrderClosingTypeEnum;
 import com.xcong.excoin.common.system.service.CommonService;
+import com.xcong.excoin.modules.contract.dao.ContractEntrustOrderDao;
 import com.xcong.excoin.modules.contract.dao.ContractHoldOrderDao;
 import com.xcong.excoin.modules.contract.dao.ContractOrderDao;
 import com.xcong.excoin.modules.contract.entity.ContractEntrustOrderEntity;
@@ -31,6 +32,7 @@
 import com.xcong.excoin.modules.member.entity.MemberSettingEntity;
 import com.xcong.excoin.modules.member.entity.MemberWalletContractEntity;
 import com.xcong.excoin.modules.platform.entity.PlatformTradeSettingEntity;
+import com.xcong.excoin.rabbit.pricequeue.OrderModel;
 import com.xcong.excoin.utils.*;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
@@ -38,6 +40,7 @@
 
 import javax.annotation.Resource;
 import java.math.BigDecimal;
+import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
 
@@ -51,30 +54,22 @@
 
     @Resource
     private MemberDao memberDao;
-
     @Resource
     private OrderWebsocketServiceImpl orderWebsocketService;
-
     @Resource
     private ContractHoldOrderDao contractHoldOrderDao;
-
     @Resource
     private ContractOrderDao contractOrderDao;
-
     @Resource
     private CommonService commonService;
-
     @Resource
     private MemberWalletContractDao memberWalletContractDao;
-
     @Resource
     private CacheSettingUtils cacheSettingUtils;
-
     @Resource
     private RedisUtils redisUtils;
     @Resource
     private MemberSettingDao memberSettingDao;
-
     @Resource
     private FollowFollowerOrderRelationDao followFollowerOrderRelationDao;
     @Resource
@@ -85,6 +80,8 @@
     private FollowTraderProfitDetailDao followTraderProfitDetailDao;
     @Resource
     private FollowFollowerProfitDao followFollowerProfitDao;
+    @Resource
+    private ContractEntrustOrderDao contractEntrustOrderDao;
 
     @Transactional(rollbackFor = Exception.class)
     @Override
@@ -93,6 +90,7 @@
             try {
                 if (ids.size() == 1) {
                     ContractHoldOrderEntity holdOrderEntity = contractHoldOrderDao.selectById(ids.get(0));
+
                     // 判断仓位类型是否逐仓
                     if (holdOrderEntity.getPositionType() == ContractEntrustOrderEntity.POSITION_TYPE_ADD) {
                         // 逐仓平仓
@@ -260,7 +258,7 @@
 
         MemberEntity memberEntity = memberDao.selectById(holdOrderEntity.getMemberId());
 
-        MemberWalletContractEntity walletContract = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(holdOrderEntity.getMemberId(), CoinTypeEnum.USDT.name());
+        MemberWalletContractEntity walletContract = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(holdOrderEntity.getMemberId(), CoinTypeConvert.convertContractTypeToCoin(symbol));
         if (walletContract != null) {
             BigDecimal lotNumber = cacheSettingUtils.getSymbolSku(symbol);
             // 盈亏
@@ -296,13 +294,21 @@
                     profitOrLoss = profitOrLoss.multiply(BigDecimal.ONE.add(tradeSettingEntity.getProfitParam()));
                 }
             }
+            log.info("profitOrLoss:{}", profitOrLoss);
 
             // 保证金
             BigDecimal bondAmount = CalculateUtil.getBondAmount(holdOrderEntity.getOpeningPrice(), lotNumber, closeCnt, holdOrderEntity.getLeverRatio());
 
-            // 平仓手续费
-            BigDecimal fee = holdOrderEntity.getOpeningFeeAmount().divide(BigDecimal.valueOf(holdOrderEntity.getSymbolCnt()), 8, BigDecimal.ROUND_DOWN).multiply(BigDecimal.valueOf(closeCnt));
+            log.info("bondAmount:{}", bondAmount);
+            // 平仓手续费  TODO 可能需要修复手续费
+            BigDecimal fee = BigDecimal.ZERO;
 
+            if (holdOrderEntity.getSymbolCntSale() != 0) {
+                fee = holdOrderEntity.getOpeningFeeAmount().divide(BigDecimal.valueOf(holdOrderEntity.getSymbolCnt()), 8, BigDecimal.ROUND_DOWN).multiply(BigDecimal.valueOf(closeCnt));
+            } else {
+                fee = holdOrderEntity.getOpeningFeeAmount();
+            }
+            log.info("fee:{}", fee);
             ContractOrderEntity contractOrderEntity = ContractHoldOrderEntityMapper.INSTANCE.holdOrderToOrder(holdOrderEntity);
             contractOrderEntity.setId(null);
             contractOrderEntity.setClosingPrice(newPrice);
@@ -314,12 +320,21 @@
             contractOrderEntity.setClosingFeeAmount(fee);
             contractOrderDao.insert(contractOrderEntity);
 
-            holdOrderEntity.setSymbolCntSale(holdOrderEntity.getSymbolCntSale() - closeCnt);
-            contractHoldOrderDao.updateById(holdOrderEntity);
+            if (holdOrderEntity.getSymbolCntSale() != 0) {
+                holdOrderEntity.setOpeningFeeAmount(holdOrderEntity.getOpeningFeeAmount().subtract(fee));
+                holdOrderEntity.setBondAmount(holdOrderEntity.getBondAmount().subtract(bondAmount));
+                contractHoldOrderDao.updateById(holdOrderEntity);
+            } else {
+                contractHoldOrderDao.deleteById(holdOrderEntity.getId());
+            }
 
-            BigDecimal changeAmount = profitOrLoss.add(bondAmount).subtract(fee);
+            BigDecimal changeAmount = profitOrLoss.add(bondAmount);
+            log.info("changeAmount : {}", changeAmount);
 
-            memberWalletContractDao.increaseWalletContractBalanceById(changeAmount, fee.negate(), null, walletContract.getId());
+            BigDecimal total = profitOrLoss.subtract(fee);
+            log.info("totalMoney : {}", total);
+
+            memberWalletContractDao.increaseWalletContractBalanceById(changeAmount, total, null, walletContract.getId());
 
             ThreadPoolUtils.sendWholeForceClosingPrice(holdOrderEntity.getSymbol(), memberEntity);
             // 计算佣金
@@ -327,4 +342,110 @@
 
         }
     }
+
+    @Override
+    public void entrustCloseOrder(List<OrderModel> list) {
+        if (CollUtil.isNotEmpty(list)) {
+            List<Long> ids = new ArrayList<>();
+            list.forEach(model -> ids.add(model.getOrderId()));
+            List<ContractEntrustOrderEntity> contractEntrustOrderEntities = contractEntrustOrderDao.selectEntrustOrderListByIds(ids);
+
+            if (CollUtil.isNotEmpty(contractEntrustOrderEntities)) {
+                for (ContractEntrustOrderEntity entrustOrderEntity : contractEntrustOrderEntities) {
+                    closeOrder(entrustOrderEntity);
+                }
+            } else {
+                log.info("平仓委托单未找到");
+            }
+        }
+    }
+
+    private void closeOrder(ContractEntrustOrderEntity entrustOrder) {
+        log.info("执行平仓委托");
+
+        Long memberId = entrustOrder.getMemberId();
+        int orderType = entrustOrder.getEntrustType() == ContractEntrustOrderEntity.ENTRUST_TYPE_CLOSE_MORE ? ContractHoldOrderEntity.OPENING_TYPE_MORE : ContractHoldOrderEntity.OPENING_TYPE_LESS;
+        String symbol = entrustOrder.getSymbol();
+
+        MemberEntity memberEntity = memberDao.selectById(memberId);
+        MemberWalletContractEntity walletContract = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberId, CoinTypeConvert.convertContractTypeToCoin(symbol));
+
+        BigDecimal lotNumber = cacheSettingUtils.getSymbolSku(symbol);
+
+        BigDecimal newPrice = entrustOrder.getEntrustPrice();
+        int closeCnt = entrustOrder.getSymbolCnt();
+
+        ContractHoldOrderEntity holdOrderEntity = contractHoldOrderDao.selectWholeHoldOrderByOrderType(memberId, orderType, symbol);
+        BigDecimal profitOrLoss;
+        int closingType;
+        // 平多
+        if (entrustOrder.getEntrustType() == ContractEntrustOrderEntity.ENTRUST_TYPE_CLOSE_MORE) {
+            profitOrLoss = newPrice.subtract(holdOrderEntity.getOpeningPrice()).multiply(lotNumber).multiply(new BigDecimal(closeCnt));
+            closingType = OrderClosingTypeEnum.CLOSE_MORE.getValue();
+
+            // 平空
+        } else {
+            profitOrLoss = newPrice.subtract(holdOrderEntity.getOpeningPrice()).multiply(lotNumber).multiply(new BigDecimal(closeCnt));
+            closingType = OrderClosingTypeEnum.CLOSE_LESS.getValue();
+        }
+
+        if (memberEntity.getIsProfit() == MemberEntity.IS_PROFIT_Y) {
+            PlatformTradeSettingEntity tradeSettingEntity = cacheSettingUtils.getTradeSetting();
+            if (profitOrLoss.compareTo(BigDecimal.ZERO) > -1) {
+                profitOrLoss = profitOrLoss.multiply(BigDecimal.ONE.subtract(tradeSettingEntity.getProfitParam()));
+            } else {
+                profitOrLoss = profitOrLoss.multiply(BigDecimal.ONE.add(tradeSettingEntity.getProfitParam()));
+            }
+        }
+        log.info("profitOrLoss:{}", profitOrLoss);
+
+        // 保证金
+        BigDecimal bondAmount = CalculateUtil.getBondAmount(holdOrderEntity.getOpeningPrice(), lotNumber, closeCnt, holdOrderEntity.getLeverRatio());
+
+        log.info("bondAmount:{}", bondAmount);
+        // 平仓手续费  TODO 可能需要修复手续费
+        BigDecimal fee = BigDecimal.ZERO;
+
+        if (holdOrderEntity.getSymbolCntSale() != 0) {
+            fee = holdOrderEntity.getOpeningFeeAmount().divide(BigDecimal.valueOf(holdOrderEntity.getSymbolCnt()), 8, BigDecimal.ROUND_DOWN).multiply(BigDecimal.valueOf(closeCnt));
+        } else {
+            fee = holdOrderEntity.getOpeningFeeAmount();
+        }
+
+        log.info("fee:{}", fee);
+        ContractOrderEntity contractOrderEntity = ContractHoldOrderEntityMapper.INSTANCE.holdOrderToOrder(holdOrderEntity);
+        contractOrderEntity.setId(null);
+        contractOrderEntity.setEntrustOpeningPrice(newPrice);
+        contractOrderEntity.setEntrustTime(new Date());
+        contractOrderEntity.setClosingPrice(newPrice);
+        contractOrderEntity.setOrderType(orderType);
+        contractOrderEntity.setClosingType(closingType);
+        contractOrderEntity.setSymbolCnt(closeCnt);
+        contractOrderEntity.setRewardAmount(profitOrLoss);
+        contractOrderEntity.setBondAmount(bondAmount);
+        contractOrderEntity.setClosingFeeAmount(fee);
+        contractOrderDao.insert(contractOrderEntity);
+
+        if (holdOrderEntity.getSymbolCntSale() - closeCnt != 0) {
+            holdOrderEntity.setSymbolCntSale(holdOrderEntity.getSymbolCntSale() - closeCnt);
+            holdOrderEntity.setOpeningFeeAmount(holdOrderEntity.getOpeningFeeAmount().subtract(fee));
+            holdOrderEntity.setBondAmount(holdOrderEntity.getBondAmount().subtract(bondAmount));
+            contractHoldOrderDao.updateById(holdOrderEntity);
+        } else {
+            contractHoldOrderDao.deleteById(holdOrderEntity.getId());
+        }
+
+        BigDecimal changeAmount = profitOrLoss.add(bondAmount);
+        log.info("changeAmount : {}", changeAmount);
+
+        BigDecimal total = profitOrLoss.subtract(fee);
+        log.info("totalMoney : {}", total);
+
+        memberWalletContractDao.increaseWalletContractBalanceById(changeAmount, total, null, walletContract.getId());
+
+        contractEntrustOrderDao.deleteById(entrustOrder.getId());
+        ThreadPoolUtils.sendWholeForceClosingPrice(holdOrderEntity.getSymbol(), memberEntity);
+        // 计算佣金
+        ThreadPoolUtils.calReturnMoney(memberEntity.getId(), fee, contractOrderEntity, AgentReturnEntity.ORDER_TYPE_CLOSE);
+    }
 }

--
Gitblit v1.9.1