From f5f590dbc5ff6ca1e270ffcfb45a80d43b315e69 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Tue, 30 Mar 2021 14:52:18 +0800
Subject: [PATCH] add entrust order follow order

---
 src/main/java/com/xcong/excoin/modules/contract/service/impl/OrderWebsocketServiceImpl.java |   55 +++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 53 insertions(+), 2 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 28d864e..9510fe7 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
@@ -2,6 +2,7 @@
 
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.util.ObjectUtil;
+import cn.hutool.core.util.StrUtil;
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.xcong.excoin.common.contants.AppContants;
@@ -18,9 +19,14 @@
 import com.xcong.excoin.modules.contract.service.ContractEntrustOrderService;
 import com.xcong.excoin.modules.contract.service.ContractHoldOrderService;
 import com.xcong.excoin.modules.contract.service.ContractOrderService;
+import com.xcong.excoin.modules.documentary.common.NoticeConstant;
+import com.xcong.excoin.modules.documentary.dao.FollowFollowerOrderRelationDao;
 import com.xcong.excoin.modules.documentary.dao.FollowFollowerProfitDao;
 import com.xcong.excoin.modules.documentary.dao.FollowFollowerSettingDao;
+import com.xcong.excoin.modules.documentary.dao.FollowTraderInfoDao;
+import com.xcong.excoin.modules.documentary.entity.FollowFollowerOrderRelationEntity;
 import com.xcong.excoin.modules.documentary.entity.FollowFollowerSettingEntity;
+import com.xcong.excoin.modules.documentary.entity.FollowTraderInfoEntity;
 import com.xcong.excoin.modules.documentary.service.FollowOrderOperationService;
 import com.xcong.excoin.modules.member.dao.AgentReturnDao;
 import com.xcong.excoin.modules.member.dao.MemberSettingDao;
@@ -33,6 +39,7 @@
 import com.xcong.excoin.modules.member.service.MemberWalletContractService;
 import com.xcong.excoin.modules.platform.entity.PlatformTradeSettingEntity;
 import com.xcong.excoin.rabbit.pricequeue.OrderModel;
+import com.xcong.excoin.rabbit.producer.FollowProducer;
 import com.xcong.excoin.rabbit.producer.OrderProducer;
 import com.xcong.excoin.utils.*;
 import lombok.extern.slf4j.Slf4j;
@@ -69,6 +76,10 @@
 
     @Resource
     FollowFollowerSettingDao followFollowerSettingDao;
+    @Resource
+    FollowFollowerOrderRelationDao followFollowerOrderRelationDao;
+    @Resource
+    FollowTraderInfoDao followTraderInfoDao;
 
     @Resource
     CacheSettingUtils cacheSettingUtils;
@@ -94,6 +105,9 @@
     private RedisUtils redisUtils;
     @Resource
     private FollowOrderOperationService followOrderOperationService;
+
+    @Resource
+    private FollowProducer followProducer;
 
     public void dealOrderFromMq(List<OrderModel> list, Integer type) {
         if (CollectionUtils.isNotEmpty(list)) {
@@ -615,6 +629,16 @@
                     contractHoldOrderEntity.setOpeningType(ContractHoldOrderEntity.OPENING_TYPE_LESS);
                 }
 
+                // 判断是否开启了带单
+                boolean isOpenFollow = false;
+                FollowTraderInfoEntity tradeInfo = null;
+                if (MemberEntity.IS_TRADER_Y.equals(memberEntity.getIsTrader())) {
+                    tradeInfo = followTraderInfoDao.selectTraderInfoByMemberId(memberEntity.getId());
+                    if (FollowTraderInfoEntity.ISOPEN_Y.equals(tradeInfo.getIsOpen())) {
+                        isOpenFollow = true;
+                    }
+                }
+
                 //持仓单赋值
                 contractHoldOrderEntity.setMemberId(memId);
                 contractHoldOrderEntity.setIsCanClosing(ContractHoldOrderEntity.ORDER_CAN_CLOSING_Y);
@@ -637,6 +661,7 @@
                 contractHoldOrderEntity.setTradeType(ContractHoldOrderEntity.TRADE_TYPE_LIMIT);
                 contractHoldOrderEntity.setOperateNo(1);
                 contractHoldOrderEntity.setSymbolCntSale(contractHoldOrderEntity.getSymbolCnt());
+                contractHoldOrderEntity.setContractType(isOpenFollow ? ContractOrderEntity.CONTRACTTYPE_DOCUMENTARY : ContractOrderEntity.CONTRACTTYPE_NORMAL);
                 contractHoldOrderService.save(contractHoldOrderEntity);
 
                 // 需要一个历史插入
@@ -665,6 +690,12 @@
 
                 //返佣
                 ThreadPoolUtils.calReturnMoney(memberEntity.getId(), openFeePrice, contractOrderEntity, AgentReturnEntity.ORDER_TYPE_OPEN);
+
+                // 若该用户为交易员且开启带单模式,则发送带单异步
+                if (isOpenFollow) {
+                    contractHoldOrderService.sendFollowOrder(tradeInfo, contractHoldOrderEntity);
+                }
+
             }
         }
     }
@@ -900,9 +931,9 @@
                 if (ContractOrderEntity.CONTRACTTYPE_DOCUMENTARY == coinsOrder.getContractType() && MemberEntity.IS_TRADER_Y.equals(memberEntity.getIsTrader())) {
                     followOrderOperationService.closingFollowOrders(coinsOrder.getOrderNo());
                 }
-                // 如果不是交易员,那么需要更新他的个人跟随者收益记录
+                // 如果不是交易员,
                 if (ContractOrderEntity.CONTRACTTYPE_DOCUMENTARY == coinsOrder.getContractType() && !MemberEntity.IS_TRADER_Y.equals(memberEntity.getIsTrader())) {
-                    //获取当前的跟单设置(一次只允许跟单一个人),获取交易员MemberId
+                    //更新他的个人跟随者收益记录
                     QueryWrapper<FollowFollowerSettingEntity> queryWrapper = new QueryWrapper<>();
                     queryWrapper.eq("member_id",coinsOrder.getMemberId());
                     FollowFollowerSettingEntity followFollowerSettingEntity = followFollowerSettingDao.selectOne(queryWrapper);
@@ -910,6 +941,11 @@
                         Long traderMemberId = followFollowerSettingEntity.getTraderMemberId();
                         followFollowerProfitDao.updateFollowerProfitByTradeMemberId(coinsOrder.getBondAmount().negate(), coinsOrder.getBondAmount().negate(), traderMemberId, memberEntity.getId());
                     }
+                    FollowTraderInfoEntity followTraderInfoEntity = followTraderInfoDao.selectById(followFollowerSettingEntity.getTraderId());
+                    //更新跟随者-订单关联表
+                    updateFollowOrderRelation(coinsOrder.getId(), contractOrderEntity.getId());
+                    //增加爆仓消息提醒
+                    LogRecordUtils.insertFollowerNotice(memberEntity.getId(), NoticeConstant.BOMB_ORDER_TITLE, StrUtil.format(NoticeConstant.BOMB_ORDER_CONTENT, contractOrderEntity.getSymbol(),followTraderInfoEntity.getNickname()));
                 }
             } else {
                 MemberWalletContractEntity wallet = memberWalletContractService.findWalletContractByMemberIdAndSymbol(memId, "USDT");
@@ -942,6 +978,21 @@
         }
     }
 
+    /**
+     * 更新跟单订单关系
+     *
+     * @param oldOrderId 当前持仓ID
+     * @param newOrderId 历史订单ID
+     */
+    public void updateFollowOrderRelation(Long oldOrderId, Long newOrderId) {
+        FollowFollowerOrderRelationEntity orderRelationEntity = followFollowerOrderRelationDao.selectNowOneByorderId(oldOrderId);
+        if (orderRelationEntity != null) {
+            orderRelationEntity.setOrderId(newOrderId);
+            orderRelationEntity.setOrderType(FollowFollowerOrderRelationEntity.ORDER_TYPE_HISTORY);
+            followFollowerOrderRelationDao.updateById(orderRelationEntity);
+        }
+    }
+
     public void calYj(Long mid, BigDecimal money, ContractOrderEntity order, int type) {
         PlatformTradeSettingEntity tradeSetting = cacheSettingUtils.getTradeSetting();
         if (money != null) {

--
Gitblit v1.9.1