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.fasterxml.jackson.databind.ObjectMapper;
|
import com.google.common.reflect.TypeToken;
|
import com.google.gson.Gson;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang.StringUtils;
|
import org.example.common.MarketDataClient;
|
import org.example.pojo.Currency;
|
import org.example.pojo.bo.CurrencyKucoin;
|
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.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: kucoin交易所
|
* @create: 2024-07-15 17:23
|
**/
|
@Component
|
@Slf4j
|
public class KucoinStock {
|
|
@Autowired
|
private CurrencySerivceImpl currencyService;
|
|
private final Lock syncCurrencyLock = new ReentrantLock();
|
|
/**
|
* 同步kucoin交易所交易对
|
*/
|
@Scheduled(cron = "0 0/35 * * * ?")
|
public void syncCurrency() {
|
// 使用Lock来确保同步
|
syncCurrencyLock.lock();
|
log.info("【同步kucoin交易所交易对】---->开始");
|
try {
|
sync();
|
} catch (Exception e) {
|
log.error("【同步kucoin交易所交易对】出现异常: " + e.getMessage());
|
e.printStackTrace();
|
} finally {
|
syncCurrencyLock.unlock();
|
log.info("【同步kucoin交易所交易对】---->结束");
|
}
|
}
|
|
public void sync() throws JsonProcessingException {
|
// 调用外部接口获取数据
|
String json = MarketDataClient.doGet("https://api.kucoin.com/api/v2/symbols");
|
|
// 对返回数据格式进行校验
|
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<CurrencyKucoin> getList = gson.fromJson(symbolsJson, new TypeToken<List<CurrencyKucoin>>() {
|
}.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.getBaseCurrency());
|
newCurrency.setQuoteAsset(currency.getQuoteCurrency());
|
newCurrency.setSource(currency.getSource());
|
return newCurrency;
|
})
|
.collect(Collectors.toList());
|
|
// 批量保存新增数据到数据库
|
if(CollectionUtils.isNotEmpty(saveList)){
|
currencyService.saveBatch(saveList);
|
}
|
} else {
|
log.info("外部接口返回数据为空");
|
}
|
}
|
|
}
|
