xiaoyong931011
2021-01-28 968f77fbe04b387274522ad5e682bacdbbcbe9bc
src/main/java/com/xcong/excoin/rabbit/pricequeue/WebsocketPriceService.java
@@ -1,6 +1,7 @@
package com.xcong.excoin.rabbit.pricequeue;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.map.MapUtil;
import cn.hutool.core.util.StrUtil;
import com.alibaba.fastjson.JSONObject;
import com.xcong.excoin.common.contants.AppContants;
@@ -11,6 +12,8 @@
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.pricequeue.whole.WholeDataQueue;
import com.xcong.excoin.rabbit.pricequeue.whole.WholePriceDataModel;
import com.xcong.excoin.rabbit.producer.OrderProducer;
import com.xcong.excoin.utils.*;
import lombok.extern.slf4j.Slf4j;
@@ -278,6 +281,47 @@
        }
    }
    public void wholeBomb() {
        Map<String, WholePriceDataModel> dataModelMap = WholeDataQueue.MAP;
        if (CollUtil.isEmpty(dataModelMap)) {
            return;
        }
        for (Map.Entry<String, WholePriceDataModel> entry : dataModelMap.entrySet()) {
            WholePriceDataModel wholePriceData = entry.getValue();
            List<ContractHoldOrderEntity> list = wholePriceData.getList();
            if (CollUtil.isNotEmpty(list)) {
                BigDecimal totalProfitOrLoss = BigDecimal.ZERO;
                Map<String, BigDecimal> prices = new HashMap<>();
                for (ContractHoldOrderEntity holdOrderEntity : list) {
                    BigDecimal newPrice = (BigDecimal) redisUtils.get(CoinTypeConvert.convertToKey(holdOrderEntity.getSymbol()));
                    BigDecimal rewardRatio = null;
                    if (ContractHoldOrderEntity.OPENING_TYPE_MORE == holdOrderEntity.getOpeningType()) {
                        // (最新价-开仓价)*规格*张数
                        rewardRatio = newPrice.subtract(holdOrderEntity.getOpeningPrice()).multiply(holdOrderEntity.getSymbolSku()).multiply(new BigDecimal(holdOrderEntity.getSymbolCntSale()));
                        // 开空
                    } else {
                        // (开仓价-最新价)*规格*张数
                        rewardRatio = holdOrderEntity.getOpeningPrice().subtract(newPrice).multiply(holdOrderEntity.getSymbolSku()).multiply(new BigDecimal(holdOrderEntity.getSymbolCntSale()));
                    }
                    totalProfitOrLoss = totalProfitOrLoss.add(rewardRatio);
                    prices.put(holdOrderEntity.getSymbol(), newPrice);
                }
                BigDecimal holdBond = wholePriceData.getHoldBond();
                BigDecimal balance = wholePriceData.getBalance();
                if (balance.add(totalProfitOrLoss).compareTo(holdBond) > 0) {
                    continue;
                }
                wholePriceData.setPrices(prices);
                orderProducer.sendWholeBomb(JSONObject.toJSONString(wholePriceData));
            }
        }
    }
    public void wholeBomb(String symbol, String price) {
        List<Long> memberIds = contractHoldOrderDao.selectMemberHasWholeOrder();
@@ -302,8 +346,9 @@
                    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) {
                    BigDecimal target = wallet.getTotalBalance().multiply(BigDecimal.valueOf(0.01));
//                    log.info("sub : {}, target : {}", sub, target);
                    if (sub.compareTo(target) <= 0) {
                        List<OrderModel> list = new ArrayList<>();
                        OrderModel orderModel = new OrderModel(null, 0, price, symbol, memberId);
                        list.add(orderModel);
@@ -311,7 +356,8 @@
                        String key = AppContants.WHOLE_BOMB_PREFIX + memberId;
                        Map<Object, Object> value = redisUtils.hmget(key);
                        if (value == null) {
                        if (MapUtil.isEmpty(value)) {
                            log.info("priceMap -- {}", priceMap);
                            orderProducer.sendWholeBomb(content);
                            contractHoldOrderDao.updateMemberAllHoldOrderClosingStatus(memberId);