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 |  249 ++++++++++++++++++++++++++++++++++++++-----------
 1 files changed, 193 insertions(+), 56 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 37b45a5..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
@@ -29,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;
@@ -46,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.*;
@@ -87,6 +86,8 @@
 
     @Resource
     private MemberDao memberDao;
+    @Resource
+    private MemberSettingDao memberSettingDao;
 
     @Resource
     private FollowTraderInfoDao followTraderInfoDao;
@@ -105,49 +106,44 @@
             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;
-
-        // 开多
-        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(MessageSourceUtils.getString("unknown_type"));
-        }
-        log.info("开仓价:{}", openingPrice);
+        BigDecimal openingPrice = CalculateUtil.getOpeningPrice(submitOrderDto.getOrderType(), newPrice, spread);
 
         // 开仓手续费 建仓价*规格*手数*手续费率
-        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/杠杆倍率)
@@ -163,27 +159,155 @@
             return Result.fail(MessageSourceUtils.getString("member_service_0085"));
         }
 
-        boolean isOpenFollow = false;
-        FollowTraderInfoEntity tradeInfo = null;
-        if (memberEntity.getContractPositionType() == ContractEntrustOrderEntity.POSITION_TYPE_ALL) {
-            ContractHoldOrderEntity wholeHoldOrder = contractHoldOrderDao.selectWholeHoldOrderByOrderType(memberEntity.getId(), submitOrderDto.getOrderType());
-            // 判断当前是否存在全仓订单,若存在则直接合并,若不存在则创建订单
-            if (wholeHoldOrder != null) {
-                return mergeWholeHoldOrder(openingPrice, openFeePrice, newPrice, submitOrderDto, wholeHoldOrder, memberEntity);
-            }
-        } else {
-            // 仅在逐仓模式中启用带单,判断是否开启了带单
-            if (MemberEntity.IS_TRADER_Y.equals(memberEntity.getIsTrader())) {
-                tradeInfo = followTraderInfoDao.selectTraderInfoByMemberId(memberEntity.getId());
-                if (tradeInfo.getIsOpen().equals(FollowTraderInfoEntity.ISOPEN_Y)) {
-                    isOpenFollow = true;
+        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());
@@ -248,6 +372,7 @@
         // 提交失败
         return Result.fail(MessageSourceUtils.getString("member_service_0067"));
     }
+
 
     /**
      * 全仓模式--若当前已经存在持仓,则合并当前持仓
@@ -411,14 +536,18 @@
             return Result.fail("可平张数不足");
         }
 
+        log.info("--->{},{}", holdOrderEntity.getSymbolCntSale(), wholeCloseOrderDto.getCount());
+        int sub = holdOrderEntity.getSymbolCntSale() - wholeCloseOrderDto.getCount();
+        log.info("剩余张数:{}", sub);
         // 判断当前持仓张数是否等于平仓张数,若等于则直接更新该订单为不可平状态,否则更新可平张数
-        if (wholeCloseOrderDto.getCount() == holdOrderEntity.getSymbolCntSale()) {
-            contractHoldOrderDao.updateHoldOrderIsCanClosingById(ContractHoldOrderEntity.ORDER_CAN_CLOSING_N, wholeCloseOrderDto.getId());
-        } else {
-            // 更新可平张数
-            holdOrderEntity.setSymbolCntSale(holdOrderEntity.getSymbolCntSale() - wholeCloseOrderDto.getCount());
-            contractHoldOrderDao.updateById(holdOrderEntity);
+        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());
@@ -601,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;
@@ -647,8 +785,6 @@
                 totalProfitOrLess = totalProfitOrLess.add(profitOrLess);
             }
         }
-
-        MemberWalletContractEntity walletContractEntity = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeEnum.USDT.name());
 
         MemberLevelRateEntity rateEntity = memberLevelRateDao.selectLeverRateByMemberIdAndSymbol(memberEntity.getId(), symbol);
 
@@ -800,6 +936,7 @@
         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"));
         }

--
Gitblit v1.9.1