From a9e9e1207ce417388b1d8f03df5833cea24939b7 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Thu, 08 Aug 2024 01:10:12 +0800
Subject: [PATCH] 1

---
 websocketSerivce/src/main/java/org/example/websocket/server/WsServer.java |   45 +++++++++++++++++++++++++++++++++++++--------
 1 files changed, 37 insertions(+), 8 deletions(-)

diff --git a/websocketSerivce/src/main/java/org/example/websocket/server/WsServer.java b/websocketSerivce/src/main/java/org/example/websocket/server/WsServer.java
index 80ccfe9..6aa9363 100644
--- a/websocketSerivce/src/main/java/org/example/websocket/server/WsServer.java
+++ b/websocketSerivce/src/main/java/org/example/websocket/server/WsServer.java
@@ -214,11 +214,9 @@
                     .collect(Collectors.toList());
         }
         //价差
-        if(wsBo.getSpread() > 0){
-            redisValueMap = redisValueMap.stream()
-                    .filter(data -> Double.parseDouble(data.getSpread()) >= wsBo.getSpread())
-                    .collect(Collectors.toList());
-        }
+        redisValueMap = redisValueMap.stream()
+                .filter(data -> Double.parseDouble(data.getSpread()) >= wsBo.getSpread())
+                .collect(Collectors.toList());
         //最低金额
         if(null !=  wsBo.getMinAmount()){
             redisValueMap = redisValueMap.stream()
@@ -241,8 +239,9 @@
                     .collect(Collectors.toList());
         }
         //自选标记
-        if(null != wsBo.getIsMarker()){
-            List<String> list = Arrays.asList(wsBo.getIsMarker().split(","));
+        String mark = RedisUtil.get(wsBo.getUserId() + "_mark");
+        if(StringUtils.isNoneEmpty(mark)){
+            List<String> list = Arrays.asList(mark.split(","));
             redisValueMap.stream()
                     .filter(data -> list.contains(data.getBuyAndSell()))
                     .forEach(data -> data.setMarker(true));
@@ -271,9 +270,39 @@
         Collections.sort(marketDataList, new Comparator<MarketDataOut>() {
             @Override
             public int compare(MarketDataOut a, MarketDataOut b) {
+                // 将买入总价和卖出总价转换为 BigDecimal
+                BigDecimal buyTotalPriceA = new BigDecimal(a.getBuyTotalPrice());
+                BigDecimal buyTotalPriceB = new BigDecimal(b.getBuyTotalPrice());
+                BigDecimal sellTotalPriceA = new BigDecimal(a.getSellTotalPrice());
+                BigDecimal sellTotalPriceB = new BigDecimal(b.getSellTotalPrice());
                 BigDecimal spreadA = new BigDecimal(a.getSpread());
                 BigDecimal spreadB = new BigDecimal(b.getSpread());
-                return spreadB.compareTo(spreadA);
+
+                // 检查 a 和 b 的买入总价或卖出总价是否大于 1000
+                boolean aBuyOrSellAbove1000 = buyTotalPriceA.compareTo(new BigDecimal("1000")) > 0 || sellTotalPriceA.compareTo(new BigDecimal("1000")) > 0;
+                boolean bBuyOrSellAbove1000 = buyTotalPriceB.compareTo(new BigDecimal("1000")) > 0 || sellTotalPriceB.compareTo(new BigDecimal("1000")) > 0;
+
+                if (aBuyOrSellAbove1000 && !bBuyOrSellAbove1000) {
+                    return -1; // a 应排在 b 前面
+                } else if (!aBuyOrSellAbove1000 && bBuyOrSellAbove1000) {
+                    return 1; // b 应排在 a 前面
+                } else {
+                    // a 和 b 都大于 1000 或都小于等于 1000
+                    // 先按照 spread 的降序排列
+                    int spreadComparison = spreadB.compareTo(spreadA);
+                    if (spreadComparison != 0) {
+                        return spreadComparison; // 先按照 spread 排序
+                    } else {
+                        // 如果 spread 相同,再按照买入总价或卖出总价的升序排列
+                        if (aBuyOrSellAbove1000) {
+                            // 对于买入总价或卖出总价大于 1000 的记录
+                            return buyTotalPriceA.compareTo(buyTotalPriceB); // 按照买入总价升序排序
+                        } else {
+                            // 对于买入总价或卖出总价小于等于 1000 的记录
+                            return buyTotalPriceA.compareTo(buyTotalPriceB); // 可以选择卖出总价或买入总价的升序排序
+                        }
+                    }
+                }
             }
         });
     }

--
Gitblit v1.9.3