From a1b05c46a7141a4c217569b89029a3d0b63dc741 Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Fri, 29 Jan 2021 14:51:12 +0800 Subject: [PATCH] modify --- src/main/java/com/xcong/excoin/rabbit/pricequeue/WebsocketPriceService.java | 77 +++++++++++++++++++++++++++++++++----- 1 files changed, 66 insertions(+), 11 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 5d90b5e..1df8876 100644 --- a/src/main/java/com/xcong/excoin/rabbit/pricequeue/WebsocketPriceService.java +++ b/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,12 +12,16 @@ 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.HoldOrderDataModel; +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; import org.apache.commons.collections.CollectionUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Component; +import org.springframework.transaction.annotation.Transactional; import javax.annotation.Resource; import java.math.BigDecimal; @@ -278,6 +283,56 @@ } } + @Transactional(rollbackFor = Exception.class) + 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<HoldOrderDataModel> list = wholePriceData.getList(); + + if (CollUtil.isNotEmpty(list)) { + BigDecimal totalProfitOrLoss = BigDecimal.ZERO; + Map<String, BigDecimal> prices = new HashMap<>(); + for (HoldOrderDataModel holdOrderData : list) { + String price = redisUtils.getString(CoinTypeConvert.convertToKey(holdOrderData.getSymbol())); + BigDecimal newPrice = new BigDecimal(price); +// BigDecimal newPrice = new BigDecimal("29958.46627789"); + + BigDecimal rewardRatio = null; + if (ContractHoldOrderEntity.OPENING_TYPE_MORE == holdOrderData.getOpeningType()) { + // (最新价-开仓价)*规格*张数 + rewardRatio = newPrice.subtract(holdOrderData.getOpeningPrice()).multiply(holdOrderData.getSymbolSku()).multiply(new BigDecimal(holdOrderData.getSymbolCntSale())); + // 开空 + } else { + // (开仓价-最新价)*规格*张数 + rewardRatio = holdOrderData.getOpeningPrice().subtract(newPrice).multiply(holdOrderData.getSymbolSku()).multiply(new BigDecimal(holdOrderData.getSymbolCntSale())); + } + holdOrderData.setRewardAmount(rewardRatio); + holdOrderData.setClosingPrice(newPrice); + totalProfitOrLoss = totalProfitOrLoss.add(rewardRatio).setScale(8, BigDecimal.ROUND_DOWN); + prices.put(holdOrderData.getSymbol(), newPrice); + } + + BigDecimal holdBond = wholePriceData.getHoldBond(); + BigDecimal balance = wholePriceData.getBalance(); + if (balance.add(totalProfitOrLoss).compareTo(holdBond) > 0) { + continue; + } + + dataModelMap.remove(entry.getKey()); + redisUtils.set(AppContants.WHOLE_BOMB_MAP, JSONObject.toJSONString(dataModelMap)); + log.info("全仓爆仓触发:{}", JSONObject.toJSONString(wholePriceData)); + wholePriceData.setPrices(prices); + contractHoldOrderDao.updateMemberAllHoldOrderClosingStatus(wholePriceData.getMemberId()); + orderProducer.sendWholeBomb(JSONObject.toJSONString(wholePriceData)); + } + } + } + public void wholeBomb(String symbol, String price) { List<Long> memberIds = contractHoldOrderDao.selectMemberHasWholeOrder(); @@ -290,13 +345,11 @@ if (CollUtil.isNotEmpty(holdOrderEntities)) { BigDecimal totalProfitOrLess = BigDecimal.ZERO; - String currentPrice = null; + Map<String, Object> priceMap = new HashMap<>(); for (ContractHoldOrderEntity holdOrderEntity : holdOrderEntities) { - if (symbol.equalsIgnoreCase(holdOrderEntity.getSymbol())) { - currentPrice = price; - } else { - currentPrice = redisUtils.getString(CoinTypeConvert.convertToKey(holdOrderEntity.getSymbol())); - } + String currentPrice = redisUtils.getString(CoinTypeConvert.convertToKey(holdOrderEntity.getSymbol())); + priceMap.put(holdOrderEntity.getSymbol(), currentPrice); + 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); @@ -304,20 +357,22 @@ 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); String content = JSONObject.toJSONString(list); String key = AppContants.WHOLE_BOMB_PREFIX + memberId; - String value = redisUtils.getString(key); - if (StrUtil.isBlank(value)) { + Map<Object, Object> value = redisUtils.hmget(key); + if (MapUtil.isEmpty(value)) { + log.info("priceMap -- {}", priceMap); orderProducer.sendWholeBomb(content); contractHoldOrderDao.updateMemberAllHoldOrderClosingStatus(memberId); - redisUtils.set(key, memberId); + redisUtils.hmset(key, priceMap); } } } -- Gitblit v1.9.1