From 2383ba4c69409ffe6d0d8fcf4e5102de5dc97085 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Thu, 28 Jan 2021 19:24:02 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/rabbit/pricequeue/WebsocketPriceService.java |   75 ++++++++++++++++++++++++++++++++-----
 1 files changed, 64 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..e0c2ece 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,6 +12,9 @@
 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;
@@ -278,6 +282,55 @@
         }
     }
 
+    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());
+                System.out.println("触发");
+                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 +343,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 +355,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