From 4bb70ea5d9ccec9f4c36eb14f86653c29f0fdc30 Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Tue, 08 Sep 2020 10:29:28 +0800 Subject: [PATCH] modify --- src/main/java/com/xcong/excoin/rabbit/pricequeue/WebsocketPriceService.java | 119 ++++++++++++++++++++++++++++++++++++++++++++++++++++++----- 1 files changed, 109 insertions(+), 10 deletions(-) diff --git a/src/main/java/com/xcong/excoin/rabbit/pricequeue/WebsocketPriceService.java b/src/main/java/com/xcong/excoin/rabbit/pricequeue/WebsocketPriceService.java index f0e24fc..0d302c2 100644 --- a/src/main/java/com/xcong/excoin/rabbit/pricequeue/WebsocketPriceService.java +++ b/src/main/java/com/xcong/excoin/rabbit/pricequeue/WebsocketPriceService.java @@ -1,16 +1,25 @@ package com.xcong.excoin.rabbit.pricequeue; +import cn.hutool.core.collection.CollUtil; import com.alibaba.fastjson.JSONObject; +import com.xcong.excoin.common.contants.AppContants; +import com.xcong.excoin.common.enumerates.CoinTypeEnum; +import com.xcong.excoin.modules.contract.dao.ContractHoldOrderDao; +import com.xcong.excoin.modules.contract.entity.ContractHoldOrderEntity; +import com.xcong.excoin.modules.member.dao.MemberDao; +import com.xcong.excoin.modules.member.dao.MemberWalletContractDao; +import com.xcong.excoin.modules.member.entity.MemberEntity; +import com.xcong.excoin.modules.member.entity.MemberWalletContractEntity; import com.xcong.excoin.rabbit.producer.OrderProducer; +import com.xcong.excoin.utils.*; import lombok.extern.slf4j.Slf4j; import org.apache.commons.collections.CollectionUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Component; -import java.util.ArrayList; -import java.util.Date; -import java.util.List; -import java.util.Map; +import javax.annotation.Resource; +import java.math.BigDecimal; +import java.util.*; import java.util.concurrent.PriorityBlockingQueue; @Slf4j @@ -19,7 +28,14 @@ @Autowired OrderProducer orderProducer; - + @Resource + private RedisUtils redisUtils; + @Resource + private ContractHoldOrderDao contractHoldOrderDao; + @Resource + private MemberDao memberDao; + @Resource + private MemberWalletContractDao memberWalletContractDao; /** * @param symbol * @param price @@ -74,6 +90,19 @@ } + private void addExecType(OrderModel model) { + List<Object> orderTypes = redisUtils.lGet(AppContants.RABBIT_TYPE + model.getOrderId(), 0, -1); + if (CollUtil.isNotEmpty(orderTypes)) { + orderTypes.add(model.getType()); + } else { + orderTypes = new ArrayList<>(); + orderTypes.add(model.getType()); + } + + redisUtils.lSet(AppContants.RABBIT_TYPE + model.getOrderId(), orderTypes, 10); + redisUtils.lSet(AppContants.MEMBER_TYPE + model.getMemberId(), orderTypes, 5); + } + // 处理消息 正序的 包括 // 1:买入委托2:开多3:开空4:平多5:平空6:爆仓平多7:爆仓平空8:撤单9:止盈平多10:止盈平空11:止损平多12:止损平空 public void dealAscPriceOrderAndSenMq(List<AscBigDecimal> list, String symbol) { @@ -86,21 +115,29 @@ for (AscBigDecimal asc : list) { String key = asc.getValue().toPlainString(); assert orderMap != null; + log.info("----->->{}, --> {}", JSONObject.toJSONString(orderMap), key); if (orderMap.containsKey(key)) { orderModelList.addAll(orderMap.get(key)); orderMap.remove(key); } } + log.info("------>{}", JSONObject.toJSONString(orderModelList)); if (CollectionUtils.isEmpty(orderModelList)) { return; } - System.out.println("本次执行的列表ASC"); - System.out.println(JSONObject.toJSONString(orderModelList)); + log.info("本次执行的列表ASC"); // 根据订单的类型发送消息 // 3:开空 7:爆仓平空 // 9:止盈平多 12:止损平空 for (OrderModel model : orderModelList) { + /* + 问题: 1、逐仓: 当行情大时,若设置的止损点与爆仓过于接近,则可能会出现直接爆仓,而不止损的情况 + 2、全仓: 止盈价/止损价 设置的与委托平仓价相同,需优先处理止盈/止损 + 解决: 将订单ID作为Key, 该订单执行的队列类型集合作为value, 用于在执行爆仓、委托平仓时,是否存在止盈/止损,若存在则不执行该爆仓和委托平仓 + */ + addExecType(model); + // 止损平空 List<OrderModel> kkzsList = new ArrayList<OrderModel>(); // 止盈平多 @@ -109,9 +146,14 @@ List<OrderModel> bcList = new ArrayList<OrderModel>(); // 开空 List<OrderModel> wtkkList = new ArrayList<OrderModel>(); + // 委托平多 + List<OrderModel> wtpdList = new ArrayList<>(); switch (model.getType()) { case 3: wtkkList.add(model); + break; + case 4: + wtpdList.add(model); break; case 7: bcList.add(model); @@ -142,6 +184,9 @@ if (CollectionUtils.isNotEmpty(wtkkList)) { orderProducer.sendLimit(JSONObject.toJSONString(wtkkList)); } + if (CollectionUtils.isNotEmpty(wtpdList)) { + orderProducer.sendLimitClose(JSONObject.toJSONString(wtpdList)); + } } } } @@ -157,6 +202,7 @@ for (DescBigDecimal desc : list) { String key = desc.getValue().toPlainString(); assert orderMap != null; + log.info("----->->{}, --> {}", JSONObject.toJSONString(orderMap), key); if (orderMap.containsKey(key)) { orderModelList.addAll(orderMap.get(key)); orderMap.remove(key); @@ -167,12 +213,18 @@ if (CollectionUtils.isEmpty(orderModelList)) { return; } - System.out.println("本次执行的列表Desc"); - System.out.println(JSONObject.toJSONString(orderModelList)); + log.info("本次执行的列表Desc"); // 根据订单的类型发送消息 // 2:开多6:爆仓平多 // 10:止盈平空11:止损平多 for (OrderModel model : orderModelList) { + /* + 问题: 1、逐仓: 当行情大时,若设置的止损点与爆仓过于接近,则可能会出现直接爆仓,而不止损的情况 + 2、全仓: 止盈价/止损价 设置的与委托平仓价相同,需优先处理止盈/止损 + 解决: 将订单ID作为Key, 该订单执行的队列类型集合作为value, 用于在执行爆仓、委托平仓时,是否存在止盈/止损,若存在则不执行该爆仓和委托平仓 + */ + addExecType(model); + // 开空止盈 List<OrderModel> kkzyList = new ArrayList<OrderModel>(); // 开多止损 @@ -181,9 +233,14 @@ List<OrderModel> bcList = new ArrayList<OrderModel>(); // 开多委托 List<OrderModel> wtkdList = new ArrayList<OrderModel>(); + // 委托平空 + List<OrderModel> wtpkList = new ArrayList<>(); switch (model.getType()) { case 2: wtkdList.add(model); + break; + case 5: + wtpkList.add(model); break; case 6: bcList.add(model); @@ -212,10 +269,52 @@ } if (CollectionUtils.isNotEmpty(wtkdList)) { orderProducer.sendLimit(JSONObject.toJSONString(wtkdList)); - + } + if (CollectionUtils.isNotEmpty(wtpkList)) { + orderProducer.sendLimitClose(JSONObject.toJSONString(wtpkList)); } } } } + + public void wholeBomb(String symbol, String price) { + List<Long> memberIds = contractHoldOrderDao.selectMemberHasWholeOrder(); + CacheSettingUtils cacheSettingUtils = SpringContextHolder.getBean(CacheSettingUtils.class); + + if (CollUtil.isNotEmpty(memberIds)) { + for (Long memberId : memberIds) { + List<ContractHoldOrderEntity> holdOrderEntities = contractHoldOrderDao.selectHoldOrderListByMemberId(memberId); + MemberEntity memberEntity = memberDao.selectById(memberId); + + if (CollUtil.isNotEmpty(holdOrderEntities)) { + BigDecimal totalProfitOrLess = BigDecimal.ZERO; + String currentPrice = null; + for (ContractHoldOrderEntity holdOrderEntity : holdOrderEntities) { + if (symbol.equalsIgnoreCase(holdOrderEntity.getSymbol())) { + currentPrice = price; + } else { + currentPrice = redisUtils.getString(CoinTypeConvert.convertToKey(holdOrderEntity.getSymbol())); + } + BigDecimal lotNumber = cacheSettingUtils.getSymbolSku(holdOrderEntity.getSymbol()); + BigDecimal profitOrLess = CalculateUtil.calOrderProfitOrLess(holdOrderEntity.getOpeningType(), new BigDecimal(currentPrice), holdOrderEntity.getOpeningPrice(), lotNumber, holdOrderEntity.getSymbolCntSale(), memberEntity.getIsProfit()); + totalProfitOrLess = totalProfitOrLess.add(profitOrLess); + } + MemberWalletContractEntity wallet = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberId, CoinTypeEnum.USDT.name()); + + BigDecimal sub = wallet.getTotalBalance().add(totalProfitOrLess); +// log.info("sub : {}, memberId : {}", sub, memberId); + if (sub.compareTo(BigDecimal.ZERO) <= 0) { + List<OrderModel> list = new ArrayList<>(); + OrderModel orderModel = new OrderModel(null, 0, price, symbol, memberId); + list.add(orderModel); + String content = JSONObject.toJSONString(list); + orderProducer.sendWholeBomb(content); + + contractHoldOrderDao.updateMemberAllHoldOrderClosingStatus(memberId); + } + } + } + } + } } -- Gitblit v1.9.1