package com.yami.trading.huobi.task;
|
|
import cn.hutool.core.collection.CollectionUtil;
|
import com.yami.trading.bean.data.domain.Kline;
|
import com.yami.trading.bean.data.domain.Realtime;
|
import com.yami.trading.bean.item.domain.Item;
|
import com.yami.trading.huobi.data.AdjustmentValueCache;
|
import com.yami.trading.huobi.data.DataCache;
|
import com.yami.trading.huobi.data.internal.DataDBService;
|
import com.yami.trading.huobi.data.internal.KlineConstant;
|
import com.yami.trading.huobi.data.internal.KlineService;
|
import com.yami.trading.huobi.data.internal.KlineTimeObject;
|
import com.yami.trading.service.item.ItemService;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
import java.util.ArrayList;
|
import java.util.Collections;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* Kline组件初始化加载缓存服务
|
*
|
* @author Jerry
|
* @date 2023/5/8
|
*/
|
@Component
|
@Slf4j
|
public class KlineLoadCache {
|
|
@Autowired
|
private KlineContext klineContext;
|
@Autowired
|
private DataDBService dataDBService;
|
@Autowired
|
private ItemService itemService;
|
@Autowired
|
private KlineService klineService;
|
|
public void loadCache() {
|
List<Item> items = new ArrayList<>(itemService.listWithOutCache());
|
items = items.stream().filter(x -> !x.getType().equalsIgnoreCase(Item.US_STOCKS)).collect(Collectors.toList()); //过滤美股
|
log.info("--------------> KlineLoadCache.loadCache, item 集合长度为1:" + items.size());
|
// 加载调整值到内存
|
for (Item item : items) {
|
AdjustmentValueCache.getCurrentValue().put(item.getSymbol(), item.getAdjustmentValue());
|
}
|
|
// 加载最新实时价格数据到内存
|
log.info("--------------> KlineLoadCache.loadCache, item 集合长度为2:" + items.size());
|
for (Item item : items) {
|
if(!item.isActive()){
|
continue;
|
}
|
String symbol = item.getSymbol();
|
Realtime realtime = klineService.findLatestRealtime(symbol);
|
if (null != realtime) {
|
DataCache.putLatestRealTime(symbol, realtime);
|
}
|
}
|
|
try {
|
// 加载最新实时价格数据到内存
|
log.info("--------------> KlineLoadCache.loadCache, item 集合长度为3:" + items.size());
|
for (Item item : items) {
|
if(!item.isActive()){
|
continue;
|
}
|
String symbol = item.getSymbol();
|
List<Realtime> list = dataDBService.listRealTime60s(symbol);
|
if (null == list || list.size() <= 0) {
|
list = new ArrayList<>();
|
}
|
DataCache.putLatestRealTime60s(symbol, list);
|
if (CollectionUtil.isNotEmpty(list)) {
|
DataCache.putLatestRealTime(symbol, list.get(0));
|
}
|
}
|
|
//加载K线数据到内存
|
log.info("--------------> KlineLoadCache.loadCache, item 集合长度为4:" + items.size());
|
long beginTime = System.currentTimeMillis();
|
for (Item item : items) {
|
if(!item.isActive()){
|
continue;
|
}
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_1MIN);
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_5MIN);
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_15MIN);
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_30MIN);
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_60MIN);
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_2HOUR);
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_4HOUR);
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_1DAY);
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_5DAY);
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_1WEEK);
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_1MON);
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_QUARTER);
|
bulidInit(item.getSymbol(), KlineConstant.PERIOD_YEAR);
|
//log.info("系统启动时候，{} k线图初始化成功", item.getSymbol());
|
klineContext.initKlineSuccess(item.getSymbol());
|
}
|
long endTime = System.currentTimeMillis();
|
log.info("------> 加载: {} 项 K 线图耗时:{} ", items.size(), (endTime - beginTime));
|
} catch (Exception e) {
|
log.error("--------------> KlineLoadCache.loadCache, item 集合长度为5:" + items.size() + ", 异常:", e);
|
throw e;
|
}
|
}
|
|
public void bulidInit(String symbol, String line) {
|
List<Kline> list = klineService.find(itemService.findBySymbol(symbol).getSymbol(), line, Integer.MAX_VALUE);
|
KlineTimeObject model = new KlineTimeObject();
|
Collections.sort(list);
|
model.setKline(list);
|
DataCache.putKline(symbol, line, model);
|
}
|
|
public void setDataDBService(DataDBService dataDBService) {
|
this.dataDBService = dataDBService;
|
}
|
|
public void setItemService(ItemService itemService) {
|
this.itemService = itemService;
|
}
|
|
public void setKlineService(KlineService klineService) {
|
this.klineService = klineService;
|
}
|
|
}
|
