Helius
2020-09-07 20e7ef5cbd5630d9d8b319856cad8e64acbce2c3
src/main/java/com/xcong/excoin/rabbit/pricequeue/WebsocketPriceService.java
@@ -3,14 +3,25 @@
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.CacheSettingUtils;
import com.xcong.excoin.utils.CalculateUtil;
import com.xcong.excoin.utils.RedisUtils;
import com.xcong.excoin.utils.SpringContextHolder;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections.CollectionUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
import java.math.BigDecimal;
import java.util.*;
import java.util.concurrent.PriorityBlockingQueue;
@@ -22,7 +33,12 @@
    OrderProducer orderProducer;
    @Resource
    private RedisUtils redisUtils;
    @Resource
    private ContractHoldOrderDao contractHoldOrderDao;
    @Resource
    private MemberDao memberDao;
    @Resource
    private MemberWalletContractDao memberWalletContractDao;
    /**
     * @param symbol
     * @param price
@@ -78,7 +94,7 @@
    }
    private void addExecType(OrderModel model) {
        List<Object> orderTypes = redisUtils.lGet(AppContants.RABBIT_TYPE + model.getOrderId(), 0 , -1);
        List<Object> orderTypes = redisUtils.lGet(AppContants.RABBIT_TYPE + model.getOrderId(), 0, -1);
        if (CollUtil.isNotEmpty(orderTypes)) {
            orderTypes.add(model.getType());
        } else {
@@ -87,6 +103,7 @@
        }
        redisUtils.lSet(AppContants.RABBIT_TYPE + model.getOrderId(), orderTypes, 10);
        redisUtils.lSet(AppContants.MEMBER_TYPE + model.getMemberId(), orderTypes, 5);
    }
    // 处理消息 正序的 包括
@@ -204,6 +221,13 @@
            // 2:开多6:爆仓平多
            // 10:止盈平空11:止损平多
            for (OrderModel model : orderModelList) {
                /*
                   问题: 1、逐仓: 当行情大时,若设置的止损点与爆仓过于接近,则可能会出现直接爆仓,而不止损的情况
                        2、全仓: 止盈价/止损价 设置的与委托平仓价相同,需优先处理止盈/止损
                   解决: 将订单ID作为Key, 该订单执行的队列类型集合作为value, 用于在执行爆仓、委托平仓时,是否存在止盈/止损,若存在则不执行该爆仓和委托平仓
                 */
                addExecType(model);
                // 开空止盈
                List<OrderModel> kkzyList = new ArrayList<OrderModel>();
                // 开多止损
@@ -256,4 +280,37 @@
        }
    }
    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;
                    for (ContractHoldOrderEntity holdOrderEntity : holdOrderEntities) {
                        BigDecimal lotNumber = cacheSettingUtils.getSymbolSku(holdOrderEntity.getSymbol());
                        BigDecimal profitOrLess = CalculateUtil.calOrderProfitOrLess(holdOrderEntity.getOpeningType(), new BigDecimal(price), holdOrderEntity.getOpeningPrice(), lotNumber, holdOrderEntity.getSymbolCnt(), memberEntity.getIsProfit());
                        totalProfitOrLess = totalProfitOrLess.add(profitOrLess);
                    }
                    MemberWalletContractEntity wallet = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberId, CoinTypeEnum.USDT.name());
                    BigDecimal sub = wallet.getTotalBalance().subtract(totalProfitOrLess);
                    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);
                    }
                }
            }
        }
    }
}