From d23645e976981bc9b670eea1d469fe8a36be309c Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Wed, 17 Apr 2024 17:19:53 +0800
Subject: [PATCH] 55测试环境

---
 src/main/java/com/xcong/excoin/modules/contract/service/impl/OrderWebsocketServiceImpl.java |  106 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 106 insertions(+), 0 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 d6c5c3c..b7304af 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,7 +1,10 @@
 package com.xcong.excoin.modules.contract.service.impl;
 
 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;
 import com.xcong.excoin.common.enumerates.CoinTypeEnum;
 import com.xcong.excoin.modules.coin.dao.MemberAccountFlowEntityDao;
@@ -16,6 +19,16 @@
 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.contract.service.RabbitOrderService;
+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;
 import com.xcong.excoin.modules.member.entity.AgentReturnEntity;
@@ -27,6 +40,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;
@@ -59,6 +73,16 @@
     MemberWalletContractService memberWalletContractService;
 
     @Resource
+    FollowFollowerProfitDao followFollowerProfitDao;
+
+    @Resource
+    FollowFollowerSettingDao followFollowerSettingDao;
+    @Resource
+    FollowFollowerOrderRelationDao followFollowerOrderRelationDao;
+    @Resource
+    FollowTraderInfoDao followTraderInfoDao;
+
+    @Resource
     CacheSettingUtils cacheSettingUtils;
 
     @Resource
@@ -80,7 +104,11 @@
     private ContractEntrustOrderDao contractEntrustOrderDao;
     @Resource
     private RedisUtils redisUtils;
+    @Resource
+    private FollowOrderOperationService followOrderOperationService;
 
+    @Resource
+    private FollowProducer followProducer;
 
     public void dealOrderFromMq(List<OrderModel> list, Integer type) {
         if (CollectionUtils.isNotEmpty(list)) {
@@ -148,6 +176,21 @@
         if (CollUtil.isNotEmpty(entrustOrders)) {
             for (ContractEntrustOrderEntity entrustOrder : entrustOrders) {
                 contractEntrustOrderDao.deleteById(entrustOrder.getId());
+            }
+        }
+    }
+
+    private void closingFollowerOrders(MemberEntity memberEntity, ContractHoldOrderEntity holdOrderEntity, ContractOrderEntity contractOrderEntity) {
+        // 判断当前持仓是否为跟单订单
+        if (ContractOrderEntity.CONTRACTTYPE_DOCUMENTARY == holdOrderEntity.getContractType()) {
+            FollowTraderInfoEntity traderInfoEntity = followTraderInfoDao.selectTraderInfoByOrderId(holdOrderEntity.getId());
+            updateFollowOrderRelation(holdOrderEntity.getId(), contractOrderEntity.getId());
+
+            // 若为交易员,则平仓跟随者订单
+            if (MemberEntity.IS_TRADER_Y.equals(memberEntity.getIsTrader())) {
+                followOrderOperationService.closingFollowOrders(holdOrderEntity.getOrderNo());
+            } else {
+                LogRecordUtils.insertFollowerNotice(memberEntity.getId(), NoticeConstant.CLOSE_ORDER_TITLE, StrUtil.format(NoticeConstant.CLOSE_ORDER_CONTENT, contractOrderEntity.getSymbol(), contractOrderEntity.getClosingPrice().setScale(2, BigDecimal.ROUND_HALF_UP).toString(), contractOrderEntity.getRewardAmount().setScale(2, BigDecimal.ROUND_HALF_UP).toString(), traderInfoEntity.getNickname()));
             }
         }
     }
@@ -246,6 +289,8 @@
                         if (ContractEntrustOrderEntity.POSITION_TYPE_ALL == order.getPositionType()) {
                             ThreadPoolUtils.sendWholePrice(memberId);
                             deleteEntrustCloseOrder(order.getOrderNo());
+                        } else {
+                            closingFollowerOrders(memberEntity, order, contractOrderEntity);
                         }
                     }
                 }
@@ -347,6 +392,8 @@
                         if (ContractEntrustOrderEntity.POSITION_TYPE_ALL == order.getPositionType()) {
                             ThreadPoolUtils.sendWholePrice(memberId);
                             deleteEntrustCloseOrder(order.getOrderNo());
+                        } else {
+                            closingFollowerOrders(memberEntity, order, contractOrderEntity);
                         }
                     }
                 }
@@ -446,6 +493,8 @@
                         if (ContractEntrustOrderEntity.POSITION_TYPE_ALL == order.getPositionType()) {
                             ThreadPoolUtils.sendWholePrice(memberId);
                             deleteEntrustCloseOrder(order.getOrderNo());
+                        } else {
+                            closingFollowerOrders(memberEntity, order, contractOrderEntity);
                         }
                     }
                 }
@@ -544,6 +593,8 @@
                         if (ContractEntrustOrderEntity.POSITION_TYPE_ALL == order.getPositionType()) {
                             ThreadPoolUtils.sendWholePrice(memberId);
                             deleteEntrustCloseOrder(order.getOrderNo());
+                        } else {
+                            closingFollowerOrders(memberEntity, order, contractOrderEntity);
                         }
                     }
                 }
@@ -602,6 +653,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);
@@ -624,6 +685,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);
 
                 // 需要一个历史插入
@@ -652,6 +714,12 @@
 
                 //返佣
                 ThreadPoolUtils.calReturnMoney(memberEntity.getId(), openFeePrice, contractOrderEntity, AgentReturnEntity.ORDER_TYPE_OPEN);
+
+                // 若该用户为交易员且开启带单模式,则发送带单异步
+                if (isOpenFollow) {
+                    contractHoldOrderService.sendFollowOrder(tradeInfo, contractHoldOrderEntity);
+                }
+
             }
         }
     }
@@ -794,6 +862,7 @@
                     continue;
                 }
 
+                // 仅逐仓有用
                 if (ContractEntrustOrderEntity.POSITION_TYPE_ADD == coinsOrder.getPositionType()) {
                     bombOrder(coinsOrder);
                 } else {
@@ -880,6 +949,28 @@
                 record.setSymbol(coinsOrder.getSymbol());
                 record.setPrice(coinsOrder.getBondAmount());
                 memberAccountFlowEntityDao.insert(record);
+
+                MemberEntity memberEntity = memberService.getById(memId);
+                // 如果订单为交易员的订单,爆仓了,则旗下的所有跟单都进行平仓
+                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())) {
+                    //更新他的个人跟随者收益记录
+                    QueryWrapper<FollowFollowerSettingEntity> queryWrapper = new QueryWrapper<>();
+                    queryWrapper.eq("member_id",coinsOrder.getMemberId());
+                    FollowFollowerSettingEntity followFollowerSettingEntity = followFollowerSettingDao.selectOne(queryWrapper);
+//                    if(ObjectUtil.isNotEmpty(followFollowerSettingEntity)){
+//                        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");
 
@@ -911,6 +1002,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