From cd95c9114fcbb3fc65c666b8b86c378f9a96d55b Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Tue, 12 Nov 2024 23:08:02 +0800
Subject: [PATCH] 1
---
websocketSerivce/src/main/java/org/example/server/impl/CurrencySerivceImpl.java | 146 +++++++++++++++++++++++++++++++-----------------
1 files changed, 94 insertions(+), 52 deletions(-)
diff --git a/websocketSerivce/src/main/java/org/example/server/impl/CurrencySerivceImpl.java b/websocketSerivce/src/main/java/org/example/server/impl/CurrencySerivceImpl.java
index bdf64ef..a2d7e16 100644
--- a/websocketSerivce/src/main/java/org/example/server/impl/CurrencySerivceImpl.java
+++ b/websocketSerivce/src/main/java/org/example/server/impl/CurrencySerivceImpl.java
@@ -1,5 +1,6 @@
package org.example.server.impl;
-
+import org.apache.commons.lang3.StringUtils;
+import org.example.websocket.server.WsServer;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
@@ -9,7 +10,6 @@
import com.google.gson.GsonBuilder;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
-import org.apache.commons.lang.StringUtils;
import org.example.dao.CurrencyMapper;
import org.example.pojo.Currency;
import org.example.pojo.Market;
@@ -19,7 +19,6 @@
import org.example.pojo.bo.MarketBo;
import org.example.server.CurrencySerivce;
import org.example.util.RedisUtil;
-import org.example.websocket.server.WsServer;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.yaml.snakeyaml.error.Mark;
@@ -59,18 +58,27 @@
private WsServer wsServer;
private void extracted() {
- ExecutorService executor = Executors.newFixedThreadPool(4); // 适当增加线程数以匹配处理的交易所数量
+ ExecutorService executor = Executors.newFixedThreadPool(20); // 适当增加线程数以匹配处理的交易所数量
// 异步处理每个交易所的数据
- CompletableFuture<Void> mexcFuture = CompletableFuture.runAsync(() -> processMexc(), executor);
- CompletableFuture<Void> gateFuture = CompletableFuture.runAsync(() -> processMarketData(RedisUtil.keys("gate"), gateList, "gate"), executor);
- CompletableFuture<Void> bitgetFuture = CompletableFuture.runAsync(() -> processMarketData(RedisUtil.keys("bitget"), bitgetList, "bitget"), executor);
- CompletableFuture<Void> kucoinFuture = CompletableFuture.runAsync(() -> processMarketData(RedisUtil.keys("kucoin"), kucoinList, "kucoin"), executor);
+ CompletableFuture<Void> mexcFuture = checkAndProcess("mexc", mexcList, executor);
+ CompletableFuture<Void> gateFuture = checkAndProcess("gate", gateList, executor);
+ CompletableFuture<Void> bitgetFuture = checkAndProcess("bitget", bitgetList, executor);
+ CompletableFuture<Void> kucoinFuture = checkAndProcess("kucoin", kucoinList, executor);
// 等待所有任务完成
CompletableFuture.allOf(mexcFuture, gateFuture, bitgetFuture, kucoinFuture).join();
executor.shutdown(); // 关闭线程池
+ }
+
+ private CompletableFuture<Void> checkAndProcess(String exchangeName, List<MarketBo> marketList, ExecutorService executor) {
+ return CompletableFuture.runAsync(() -> {
+ Set<String> keys = RedisUtil.keys(exchangeName);
+ if (keys != null && !keys.isEmpty()) {
+ processMarketData(keys, marketList, exchangeName);
+ }
+ }, executor);
}
private void processMexc() {
@@ -95,7 +103,7 @@
if (bidsObj instanceof List && !((List<?>) bidsObj).isEmpty()) {
List<?> bidsList = (List<?>) bidsObj;
- Map<?, ?> bidsMap = (Map<?, ?>) bidsList.get(bidsList.size() - 1);
+ Map<?, ?> bidsMap = (Map<?, ?>) bidsList.get(0);
BidsBo bidsBo = new BidsBo();
bidsBo.setP(new BigDecimal(bidsMap.get("p").toString()));
bidsBo.setV(new BigDecimal(bidsMap.get("v").toString()));
@@ -109,36 +117,46 @@
}
private void processMarketData(Set<String> set, List<MarketBo> list, String exchangeName) {
- for (String key : set) {
- String v = RedisUtil.get(key);
- Map<String, Object> redisValueMap = gson.fromJson(v, new TypeToken<Map<String, Object>>() {}.getType());
- Object asksObj = redisValueMap.get("asks");
- Object bidsObj = redisValueMap.get("bids");
+ ArrayList<String> kyes = new ArrayList<>(set);
+ List<String> mget = RedisUtil.mget(kyes);
+
+ Map<String, String> resultMap = new HashMap<>();
+ for (int i = 0; i < kyes.size(); i++) {
+ resultMap.put(kyes.get(i), mget.get(i));
+ }
+
+ for (Map.Entry<String, String> entry : resultMap.entrySet()) {
+ String v = entry.getValue();
+ String key = entry.getKey();
+
MarketBo marketBo = new MarketBo();
- if (asksObj instanceof List && !((List<?>) asksObj).isEmpty()) {
- List<?> asksList = (List<?>) asksObj;
- String[][] dataArray = gson.fromJson(gson.toJson(asksList), String[][].class);
- String[] asksData = dataArray[0];
- AsksBo asksBo = new AsksBo();
- asksBo.setP(new BigDecimal(asksData[0]));
- asksBo.setV(new BigDecimal(asksData[1]));
- marketBo.setAsks(asksBo);
+ Map<String, Map<String,String>> redisValueMap = gson.fromJson(v, new TypeToken<Map<String, Map<String,String>>>() {}.getType());
+ Map<String,String> asksObj = redisValueMap.get("asks");
+ Map<String,String> bidsObj = redisValueMap.get("bids");
+
+ try {
+ if(StringUtils.isNotEmpty(asksObj.get("p")) && StringUtils.isNotEmpty(asksObj.get("v")) &&
+ StringUtils.isNotEmpty(bidsObj.get("p")) && StringUtils.isNotEmpty(bidsObj.get("v"))) {
+ AsksBo asksBo = new AsksBo();
+ asksBo.setP(new BigDecimal(asksObj.get("p").toString()));
+ asksBo.setV(new BigDecimal(asksObj.get("v").toString()));
+ marketBo.setAsks(asksBo);
+
+ BidsBo bidsBo = new BidsBo();
+ bidsBo.setP(new BigDecimal(bidsObj.get("p").toString()));
+ bidsBo.setV(new BigDecimal(bidsObj.get("v").toString()));
+ marketBo.setBids(bidsBo);
+
+
+ marketBo.setKey(key.replaceAll(exchangeName, ""));
+ marketBo.setExchange(exchangeName);
+ list.add(marketBo);
+ }
+ }catch (Exception e){
+ e.printStackTrace();
}
- if (bidsObj instanceof List && !((List<?>) bidsObj).isEmpty()) {
- List<?> bidsList = (List<?>) bidsObj;
- String[][] dataArray = gson.fromJson(gson.toJson(bidsList), String[][].class);
- String[] bidsData = dataArray[bidsList.size() - 1];
- BidsBo bidsBo = new BidsBo();
- bidsBo.setP(new BigDecimal(bidsData[0]));
- bidsBo.setV(new BigDecimal(bidsData[1]));
- marketBo.setBids(bidsBo);
- }
-
- marketBo.setKey(key.replaceAll(exchangeName, ""));
- marketBo.setExchange(exchangeName);
- list.add(marketBo);
}
}
@@ -188,9 +206,14 @@
if (markets1.getBids() == null) continue;
for (int j = 0; j < exchanges.length; j++) {
- if (i == j) continue;
+
MarketBo markets2 = exchangeMap.get(exchanges[j]);
+
+ if(markets1.getExchange().equals(markets2.getExchange())){
+ continue;
+ }
+
if (markets2.getAsks() == null) continue;
CompletableFuture<Void> future = CompletableFuture.runAsync(() -> {
@@ -218,25 +241,41 @@
allFutures.join();
executor.shutdown(); // 关闭线程池
-
+ mexcList.clear();
+ gateList.clear();
+ bitgetList.clear();
+ kucoinList.clear();
pushWs(marketDataOuts); // 推送结果
}
private static void assembleMarketDataOut(String coinName, MarketBo markets1, MarketBo markets2, BigDecimal profitPercentage, BigDecimal buyPrice, BigDecimal sellPrice, List<MarketDataOut> marketDataOuts, String formattedDateTime) {
- MarketDataOut marketDataOut = new MarketDataOut();
- marketDataOut.setBaseAsset(coinName.replaceAll("USDT","").toLowerCase()); // 设置基础资产
- marketDataOut.setBuyingPlatform(markets1.getExchange()); // 设置买入平台
- marketDataOut.setSellPlatform(markets2.getExchange()); // 设置卖出平台
- marketDataOut.setSpread(profitPercentage.toString()); // 设置利润百分比
- marketDataOut.setBuyPrice(buyPrice.toString()); // 设置买入价格
- marketDataOut.setSellPrice(sellPrice.toString()); // 设置卖出价格
- marketDataOut.setBuyNumber(markets1.getBids().getV().toString()); // 设置买入数量
- marketDataOut.setSellNumber(markets2.getAsks().getV().toString()); // 设置卖出数量
- marketDataOut.setBuyTotalPrice(markets1.getBids().getP().multiply(markets1.getBids().getV()).toString()); // 设置买入总价
- marketDataOut.setSellTotalPrice(markets2.getAsks().getP().multiply(markets2.getAsks().getV()).toString()); // 设置卖出总价
- marketDataOut.setServceTime(formattedDateTime); // 设置服务时间
- marketDataOuts.add(marketDataOut); // 添加到输出列表
+ if(coinName.contains(String.valueOf("USDT"))){
+ MarketDataOut marketDataOut = new MarketDataOut();
+ marketDataOut.setBaseAsset(coinName.replaceAll("USDT","").toLowerCase().toUpperCase()); // 设置基础资产
+ marketDataOut.setBuyingPlatform(capitalizeFirstLetter(markets1.getExchange())); // 设置买入平台，首字母大写
+ marketDataOut.setSellPlatform(capitalizeFirstLetter(markets2.getExchange())); // 设置卖出平台，首字母大写
+ marketDataOut.setSpread(profitPercentage.setScale(4, RoundingMode.DOWN).toPlainString()); // 设置利润百分比
+ marketDataOut.setBuyPrice(buyPrice.toPlainString()); // 设置买入价格
+ marketDataOut.setSellPrice(sellPrice.toPlainString()); // 设置卖出价格
+ marketDataOut.setBuyNumber(markets1.getBids().getV().setScale(4, RoundingMode.HALF_UP).toPlainString()); // 设置买入数量
+ marketDataOut.setSellNumber(markets2.getAsks().getV().setScale(4, RoundingMode.HALF_UP).toPlainString()); // 设置卖出数量
+ marketDataOut.setBuyTotalPrice((markets1.getBids().getP().multiply(markets1.getBids().getV())).setScale(0, RoundingMode.HALF_UP).toPlainString()); // 设置买入总价
+ marketDataOut.setSellTotalPrice((markets2.getAsks().getP().multiply(markets2.getAsks().getV())).setScale(0,RoundingMode.HALF_UP).toPlainString()); // 设置卖出总价
+ marketDataOut.setServceTime(formattedDateTime); // 设置服务时间
+ marketDataOut.setBuyAndSell(marketDataOut.getBaseAsset()+marketDataOut.getBuyingPlatform()+marketDataOut.getSellPlatform());
+ marketDataOuts.add(marketDataOut); // 添加到输出列表
+ }else{
+ System.out.println(coinName);
+ }
}
+
+ public static String capitalizeFirstLetter(String word) {
+ if (word == null || word.isEmpty()) {
+ return word;
+ }
+ return Character.toUpperCase(word.charAt(0)) + word.substring(1);
+ }
+
private void pushWs(List<MarketDataOut> marketDataOuts) {
// String key = "MARKET_Date";
@@ -247,7 +286,7 @@
// RedisUtil.set(key,json);
try {
// 全体发送
- wsServer.sendMessageToAll("11111111111111111111");
+ wsServer.sendMessageToAll(json);
} catch (Exception e) {
e.printStackTrace();
}
@@ -255,7 +294,10 @@
public void quotationCalculation(){
extracted();
- findProfitablePairs(mexcList, gateList, bitgetList, kucoinList);
+ // 检查列表是否为空并调用 findProfitablePairs 方法
+ if (!mexcList.isEmpty() || !gateList.isEmpty() || !bitgetList.isEmpty() || !kucoinList.isEmpty()) {
+ findProfitablePairs(mexcList, gateList, bitgetList, kucoinList);
+ }
}
public void scheduler(){
--
Gitblit v1.9.3