From 85eb20cf0c8b70b1bf83e952baf13f581841387b Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Wed, 17 Jul 2024 19:26:51 +0800
Subject: [PATCH] 1
---
websocketSerivce/src/main/java/org/example/task/MexcStock.java | 101 +++++++++++++++++++++++++++++++-------------------
1 files changed, 62 insertions(+), 39 deletions(-)
diff --git a/websocketSerivce/src/main/java/org/example/task/MexcStock.java b/websocketSerivce/src/main/java/org/example/task/MexcStock.java
index 850910b..d4351e2 100644
--- a/websocketSerivce/src/main/java/org/example/task/MexcStock.java
+++ b/websocketSerivce/src/main/java/org/example/task/MexcStock.java
@@ -1,75 +1,98 @@
package org.example.task;
+import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
+import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
-import com.google.common.collect.ImmutableMap;
-import com.google.common.collect.Maps;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.google.common.reflect.TypeToken;
import com.google.gson.Gson;
import lombok.extern.slf4j.Slf4j;
-import org.example.common.MarketDataClient;
+import org.example.common.MexcMarketDataClient;
import org.example.pojo.Currency;
import org.example.pojo.ExchangeInfo;
-import org.example.util.JsonUtil;
+import org.example.pojo.bo.CurrencyMexcBo;
+import org.example.server.impl.CurrencySerivceImpl;
+import org.example.util.ConverterUtil;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.Map;
-import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.*;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
-
-import static com.alibaba.druid.sql.ast.SQLPartitionValue.Operator.List;
+import java.util.stream.Collectors;
/**
* @program: demo
* @description: mexc交易所
- * @create: 2024-07-15 17:22
+ * @create: 2024-07-15 17:22
**/
@Component
@Slf4j
public class MexcStock {
- private final AtomicBoolean syncCurrency = new AtomicBoolean(false);
+ @Autowired
+ private CurrencySerivceImpl currencyService;
private final Lock syncCurrencyLock = new ReentrantLock();
/**
* 同步mexc交易所交易对
*/
- @Scheduled(cron = "0 0/5 * * * ?")
+ @Scheduled(cron = "0 0/30 * * * ?")
public void syncCurrency() {
- if (syncCurrency.get()) {
- return;
- }
- if (syncCurrencyLock.tryLock()) {
- System.out.println("【同步mexc交易所交易对】---->开市");
- try {
- syncCurrency.set(true);
- sync();
- } catch (Exception e) {
- System.err.println("【同步mexc交易所交易对】出现异常: " + e.getMessage());
- } finally {
- syncCurrencyLock.unlock();
- syncCurrency.set(false);
- System.out.println("【同步mexc交易所交易对】---->结束");
- }
+ // 使用Lock来确保同步
+ syncCurrencyLock.lock();
+ log.info("【同步mexc交易所交易对】---->开始");
+ try {
+ sync();
+ } catch (Exception e) {
+ log.error("【同步mexc交易所交易对】出现异常: " + e.getMessage());
+ e.printStackTrace();
+ } finally {
+ syncCurrencyLock.unlock();
+ log.info("【同步mexc交易所交易对】---->结束");
}
}
- public void sync() {
- HashMap<String, String> symbolParams = Maps.newHashMap(ImmutableMap.<String, String>builder()
- .build());
- String json = JsonUtil.toJson(exchangeInfo(symbolParams));
- Gson gson = new Gson();
- Map map = gson.fromJson(json, Map.class);
- String symbols = JsonUtil.toJson(map.get("symbols"));
- ArrayList arrayList = gson.fromJson(symbols, ArrayList.class);
+ public void sync() throws JsonProcessingException {
+ Map<String, String> symbolParams = new HashMap<>();
+ ObjectMapper objectMapper = new ObjectMapper();
+ // 调用外部接口获取数据
+ ExchangeInfo exchangeInfo = exchangeInfo(symbolParams);
+ String json = objectMapper.writeValueAsString(exchangeInfo);
+
+ // 对返回数据格式进行校验
+ if (json != null && !json.isEmpty()) {
+ Map<String, Object> map = objectMapper.readValue(json, new TypeReference<Map<String, Object>>() {
+ });
+
+ // 获取接口返回的symbols数据
+ String symbolsJson = objectMapper.writeValueAsString(map.get("symbols"));
+ Gson gson = new Gson();
+ List<CurrencyMexcBo> getList = gson.fromJson(symbolsJson, new TypeToken<List<CurrencyMexcBo>>() {
+ }.getType());
+
+ // 获取数据库中已有的symbol列表
+ List<Currency> dbList = currencyService.list();
+ Set<String> symbolSet = dbList.stream().map(Currency::getSymbol).collect(Collectors.toSet());
+
+ // 比对接口返回的数据和数据库中已有的数据，找出新增的数据
+ List<CurrencyMexcBo> saveList = getList.stream()
+ .filter(CurrencyMexcBo -> !symbolSet.contains(CurrencyMexcBo.getSymbol()))
+ .collect(Collectors.toList());
+ List<Currency> currencies = ConverterUtil.convertToList(saveList, Currency.class);
+ // 批量保存新增数据到数据库
+ if(CollectionUtils.isNotEmpty(currencies)){
+ currencyService.saveBatch(currencies);
+ }
+ } else {
+ log.info("外部接口返回数据为空");
+ }
}
public static ExchangeInfo exchangeInfo(Map<String, String> params) {
- return MarketDataClient.get("/api/v3/exchangeInfo", params, new TypeReference<ExchangeInfo>() {
+ return MexcMarketDataClient.get("/api/v3/exchangeInfo", params, new TypeReference<ExchangeInfo>() {
});
}
-
-}
+}
\ No newline at end of file
--
Gitblit v1.9.3