From 1fc98d66589965b5b10e09334f715b28ed5ab455 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Tue, 08 Sep 2020 10:33:47 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/rabbit/pricequeue/WebsocketPriceService.java |   29 ++++++++++++++++++++---------
 1 files changed, 20 insertions(+), 9 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 9652c3f..5d90b5e 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.util.StrUtil;
 import com.alibaba.fastjson.JSONObject;
 import com.xcong.excoin.common.contants.AppContants;
 import com.xcong.excoin.common.enumerates.CoinTypeEnum;
@@ -11,10 +12,7 @@
 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 com.xcong.excoin.utils.*;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.collections.CollectionUtils;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -292,22 +290,35 @@
 
                 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(price), holdOrderEntity.getOpeningPrice(), lotNumber, holdOrderEntity.getSymbolCnt(), memberEntity.getIsProfit());
-
+                        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().subtract(totalProfitOrLess);
+                    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);
-//                        orderProducer.sendWholeBomb(content);
+
+                        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);
+                        }
                     }
                 }
             }

--
Gitblit v1.9.1