From 877945f57ac73d13ba41bf367edb8df6f153b9eb Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Thu, 06 Aug 2020 10:33:31 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/modules/contract/service/impl/ContractHoldOrderServiceImpl.java |   23 +++++++++++++++++++----
 1 files changed, 19 insertions(+), 4 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 95137e1..2ea272f 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
@@ -22,7 +22,9 @@
 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;
@@ -87,6 +89,8 @@
     
     @Resource
     private FollowTraderInfoDao followTraderInfoDao;
+    @Resource
+    private FollowFollowerOrderRelationDao followFollowerOrderRelationDao;
 
     @Transactional(rollbackFor = Exception.class)
     @Override
@@ -150,8 +154,9 @@
 
         // 判断是否开启了带单
         boolean isOpenFollow = false;
+        FollowTraderInfoEntity tradeInfo = null;
         if (MemberEntity.IS_TRADER_Y.equals(memberEntity.getIsTrader())) {
-            FollowTraderInfoEntity tradeInfo = followTraderInfoDao.selectTraderInfoByMemberId(memberEntity.getId());
+            tradeInfo = followTraderInfoDao.selectTraderInfoByMemberId(memberEntity.getId());
             if (tradeInfo.getIsOpen().equals(FollowTraderInfoEntity.ISOPEN_Y)) {
                 isOpenFollow = true;
             }
@@ -201,6 +206,16 @@
 
             // 若该用户为交易员且开启带单模式,则发送带单异步
             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("提交成功");
@@ -224,10 +239,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;
@@ -317,7 +332,7 @@
     @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("订单不存在");
         }

--
Gitblit v1.9.1