modify

Helius

2020-09-08 1fc98d66589965b5b10e09334f715b28ed5ab455

 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 cn.hutool.core.util.StrUtil;
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.RedisUtils;
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 javax.annotation.Resource;
import java.math.BigDecimal;
import java.util.*;
import java.util.concurrent.PriorityBlockingQueue;

@@ -22,7 +31,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 +92,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 {
@@ -264,4 +278,50 @@
        }
    }


    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);

                        String key = AppContants.WHOLE_BOMB_PREFIX + memberId;
                        String value = redisUtils.getString(key);
                        if (StrUtil.isBlank(value)) {
                            orderProducer.sendWholeBomb(content);
                            contractHoldOrderDao.updateMemberAllHoldOrderClosingStatus(memberId);

                            redisUtils.set(key, memberId);
                        }
                    }
                }
            }
        }
    }
}