package com.yami.trading.api.controller;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.yami.trading.bean.data.domain.Kline;
|
import com.yami.trading.bean.data.domain.Realtime;
|
import com.yami.trading.bean.data.dto.BeforeClose;
|
import com.yami.trading.bean.ico.domain.Ico;
|
import com.yami.trading.bean.item.domain.Item;
|
import com.yami.trading.common.domain.Result;
|
import com.yami.trading.common.exception.YamiShopBindException;
|
import com.yami.trading.common.util.DateUtils;
|
import com.yami.trading.common.web.ResultObject;
|
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.KlineService;
|
import com.yami.trading.service.data.DataService;
|
import com.yami.trading.service.etf.MarketService;
|
import com.yami.trading.service.ico.IcoService;
|
import com.yami.trading.service.item.ItemService;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Qualifier;
|
import org.springframework.web.bind.annotation.*;
|
|
import java.math.BigDecimal;
|
import java.math.RoundingMode;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
/**
|
* K线图
|
*/
|
@RestController
|
@CrossOrigin
|
@Api(tags = "K线图实时行情")
|
public class KlineController {
|
|
|
public static final String HOBI = "/api/hobi!";
|
private static Logger logger = LoggerFactory.getLogger(KlineController.class);
|
@Autowired
|
private KlineService klineService;
|
@Autowired
|
private ItemService itemService;
|
@Autowired
|
@Qualifier("dataService")
|
private DataService dataService;
|
@Autowired
|
private IcoService icoService;
|
|
@Autowired
|
private DataDBService dataDBService;
|
|
@ApiOperation(value = "行情")
|
@GetMapping(HOBI + "getKline.action")
|
public Result<List<Map<String, Object>>> getKline(@RequestParam String symbol, @RequestParam String line) {
|
ResultObject resultObject = new ResultObject();
|
// 1min, 5min, 15min, 30min, 60min, 4hour, 1day, 1mon, 1week
|
try {
|
if ("1quarter".equalsIgnoreCase(line)) {
|
line = Kline.PERIOD_QUARTER;
|
}
|
if ("1year".equalsIgnoreCase(line)) {
|
line = Kline.PERIOD_YEAR;
|
}
|
// 数据处理
|
List<Kline> data = this.dataService.kline(symbol, line);
|
if ("1day".equals(line) || "5day".equals(line) || "1mon".equals(line) || "1week".equals(line) || "quarter".equalsIgnoreCase(line)
|
|| "year".equalsIgnoreCase(line)) {
|
for (Kline datum : data) {
|
datum.setCurrentTime(
|
DateUtils.timeStamp2Date(String.valueOf(datum.getTs()), "yyyy-MM-dd"));
|
}
|
} else if ("1min".equals(line)) {
|
for (Kline datum : data) {
|
datum.setCurrentTime(DateUtils.timeStamp2Date(String.valueOf(datum.getTs()), "HH:mm"));
|
}
|
} else {
|
for (Kline datum : data) {
|
datum.setCurrentTime(
|
DateUtils.timeStamp2Date(String.valueOf(datum.getTs()), "MM-dd HH:mm"));
|
}
|
}
|
return Result.succeed(this.build(data, line, symbol));
|
} catch (Exception e) {
|
logger.error("getKline error", e);
|
throw new YamiShopBindException("Failed to load chart data");
|
}
|
}
|
|
|
private List<Map<String, Object>> build(List<Kline> data, String line, String symbol) {
|
Collections.sort(data);
|
int len = data.size();
|
for (int i = 1; i < len; i++) {
|
data.get(i).setOpen(data.get(i - 1).getClose());
|
}
|
|
|
Set<Long> tsSet = new HashSet<Long>();
|
List<Map<String, Object>> list = new ArrayList<Map<String, Object>>();
|
Item bySymbol = itemService.findBySymbol(symbol);
|
|
for (int i = 0; i < data.size(); i++) {
|
Kline kline = data.get(i);
|
Long ts = kline.getTs();
|
if (tsSet.contains(ts)) {
|
continue;
|
} else {
|
tsSet.add(ts);
|
}
|
String fake = bySymbol.getFake();
|
// if("1".equalsIgnoreCase(fake)){
|
// klineService.smoothlyKline(kline, 0.99);
|
// }
|
int decimal = bySymbol.getDecimals();
|
BigDecimal low = kline.getLow();
|
BigDecimal high = kline.getHigh();
|
BigDecimal open = kline.getOpen();
|
BigDecimal close = kline.getClose();
|
if (low == null || low.compareTo(BigDecimal.ZERO) == 0) {
|
continue;
|
}
|
if (high == null || high.compareTo(BigDecimal.ZERO) == 0) {
|
continue;
|
}
|
if (close == null || close.compareTo(BigDecimal.ZERO) == 0) {
|
continue;
|
}
|
if (open == null || open.compareTo(BigDecimal.ZERO) == 0) {
|
continue;
|
}
|
Map<String, Object> map = new HashMap<String, Object>();
|
map.put("line", line);
|
map.put("symbol", kline.getSymbol());
|
map.put("timestamp", ts);
|
map.put("decimals", decimal);
|
map.put("ts", ts);
|
map.put("current_time", kline.getCurrentTime());
|
map.put("open", open.setScale(decimal, RoundingMode.HALF_UP));
|
map.put("close", close.setScale(decimal, RoundingMode.HALF_UP));
|
map.put("high", high.setScale(decimal, RoundingMode.HALF_UP));
|
map.put("low", low.setScale(decimal, RoundingMode.HALF_UP));
|
map.put("volume", kline.getVolume());
|
|
if (i == data.size() - 1 && isRollingIntradayLine(line)) {
|
Realtime realtime = DataCache.getLatestRealTime(symbol);
|
BigDecimal latestClose = close;
|
if (realtime != null && realtime.getClose() != null) {
|
latestClose = realtime.getClose();
|
map.put("close", latestClose.setScale(decimal, RoundingMode.HALF_UP));
|
}
|
BeforeClose beforeClose = dataDBService.getBeforeClose(kline.getSymbol(), line, ts, realtime);
|
BigDecimal periodHigh = high;
|
BigDecimal periodLow = low;
|
if (beforeClose != null) {
|
if (beforeClose.getMaxClose() != null && beforeClose.getMaxClose().compareTo(BigDecimal.ZERO) > 0) {
|
periodHigh = beforeClose.getMaxClose();
|
}
|
if (beforeClose.getMinClose() != null && beforeClose.getMinClose().compareTo(BigDecimal.ZERO) > 0) {
|
periodLow = beforeClose.getMinClose();
|
}
|
}
|
periodHigh = periodHigh.max(latestClose);
|
periodLow = periodLow.min(latestClose);
|
map.put("high", periodHigh.setScale(decimal, RoundingMode.HALF_UP));
|
map.put("low", periodLow.setScale(decimal, RoundingMode.HALF_UP));
|
}
|
list.add(map);
|
}
|
return list;
|
}
|
|
/**
|
* 5/15/30/60 分钟未收盘 K 线需用当前周期内的最高/最低价刷新
|
*/
|
private boolean isRollingIntradayLine(String line) {
|
return Kline.PERIOD_5MIN.equalsIgnoreCase(line)
|
|| Kline.PERIOD_15MIN.equalsIgnoreCase(line)
|
|| Kline.PERIOD_30MIN.equalsIgnoreCase(line)
|
|| Kline.PERIOD_60MIN.equalsIgnoreCase(line);
|
}
|
|
}
|
