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/BitgetStock.java | 118 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 118 insertions(+), 0 deletions(-)
diff --git a/websocketSerivce/src/main/java/org/example/task/BitgetStock.java b/websocketSerivce/src/main/java/org/example/task/BitgetStock.java
index 6e2da5f..fba2768 100644
--- a/websocketSerivce/src/main/java/org/example/task/BitgetStock.java
+++ b/websocketSerivce/src/main/java/org/example/task/BitgetStock.java
@@ -1,9 +1,127 @@
package org.example.task;
+import com.alibaba.fastjson.JSON;
+import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
+import com.fasterxml.jackson.core.type.TypeReference;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.google.common.collect.Maps;
+import com.google.common.reflect.TypeToken;
+import com.google.gson.Gson;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang.StringUtils;
+import org.apache.http.HttpEntity;
+import org.apache.http.HttpResponse;
+import org.apache.http.client.HttpClient;
+import org.apache.http.client.methods.HttpGet;
+import org.apache.http.impl.client.HttpClients;
+import org.apache.http.util.EntityUtils;
+import org.example.common.*;
+import org.example.pojo.Currency;
+import org.example.pojo.bo.CurrencyBitgetBo;
+import org.example.server.impl.CurrencySerivceImpl;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.net.HttpURLConnection;
+import java.net.URL;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.concurrent.locks.Lock;
+import java.util.concurrent.locks.ReentrantLock;
+import java.util.stream.Collectors;
+
/**
* @program: demo
* @description: bitget交易所
* @create: 2024-07-15 17:24
**/
+@Component
+@Slf4j
public class BitgetStock {
+
+ @Autowired
+ private CurrencySerivceImpl currencyService;
+
+ private final Lock syncCurrencyLock = new ReentrantLock();
+
+
+ /**
+ * 同步bitget交易所交易对
+ */
+ @Scheduled(cron = "0 0/45 * * * ?")
+ public void syncCurrency() {
+ // 使用Lock来确保同步
+ syncCurrencyLock.lock();
+ log.info("【同步bitget交易所交易对】---->开始");
+ try {
+ sync();
+ } catch (Exception e) {
+ log.error("【同步bitget交易所交易对】出现异常: " + e.getMessage());
+ e.printStackTrace();
+ } finally {
+ syncCurrencyLock.unlock();
+ log.info("【同步bitget交易所交易对】---->结束");
+ }
+ }
+
+ public void sync() throws IOException {
+ // 调用外部接口获取数据
+ String json = doGet();
+ // 对返回数据格式进行校验
+ if (json != null && !json.isEmpty()) {
+ ObjectMapper objectMapper = new ObjectMapper();
+ Map<String, Object> map = objectMapper.readValue(json, new TypeReference<Map<String, Object>>() {
+ });
+ String symbolsJson = objectMapper.writeValueAsString(map.get("data"));
+ Gson gson = new Gson();
+ List<CurrencyBitgetBo> getList = gson.fromJson(symbolsJson, new TypeToken<List<CurrencyBitgetBo>>() {
+ }.getType());
+
+ getList.parallelStream().forEach(person -> person.setSymbol(StringUtils.remove(person.getSymbol(), "-")));
+
+ // 获取数据库中已有的symbol列表
+ List<Currency> dbList = currencyService.list();
+ Set<String> symbolSet = dbList.stream().map(Currency::getSymbol).collect(Collectors.toSet());
+
+ // 比对接口返回的数据和数据库中已有的数据，找出新增的数据
+ List<Currency> saveList = getList.stream()
+ .filter(currency -> !symbolSet.contains(currency.getSymbol()))
+ .map(currency -> {
+ Currency newCurrency = new Currency();
+ newCurrency.setSymbol(currency.getSymbol());
+ newCurrency.setBaseAsset(currency.getBaseCoin());
+ newCurrency.setQuoteAsset(currency.getQuoteCoin());
+ newCurrency.setSource(currency.getSource());
+ return newCurrency;
+ })
+ .collect(Collectors.toList());
+
+ // 批量保存新增数据到数据库
+ if (CollectionUtils.isNotEmpty(saveList)) {
+ currencyService.saveBatch(saveList);
+ }
+ } else {
+ log.info("同步bitget交易所交易对，外部接口返回数据为空");
+ }
+ }
+
+ public static String doGet() throws IOException {
+ HttpClient httpClient = HttpClients.createDefault();
+ HttpGet request = new HttpGet("https://api.bitget.com/api/v2/spot/public/symbols");
+ HttpResponse response = httpClient.execute(request);
+ try {
+ // 处理响应内容
+ HttpEntity entity = response.getEntity();
+ String responseBody = EntityUtils.toString(entity);
+ return responseBody;
+ } finally {
+ // 确保释放资源
+ EntityUtils.consume(response.getEntity());
+ }
+ }
}
--
Gitblit v1.9.3