package com.yami.trading.api.controller;
|
|
import com.yami.trading.bean.data.domain.Kline;
|
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.web.ResultObject;
|
import com.yami.trading.huobi.constants.KlinePeriodEnum;
|
import com.yami.trading.huobi.data.TimeZoneConverterService;
|
import com.yami.trading.service.data.DataService;
|
import com.yami.trading.service.item.ItemService;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Qualifier;
|
import org.springframework.web.bind.annotation.CrossOrigin;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.RequestParam;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import java.math.BigDecimal;
|
import java.math.RoundingMode;
|
import java.util.*;
|
|
/**
|
* K线图
|
*/
|
@RestController
|
@CrossOrigin
|
@Api(tags = "K线图实时行情")
|
@Slf4j
|
public class KlineController {
|
@Autowired
|
private ItemService itemService;
|
@Autowired
|
@Qualifier("dataService")
|
private DataService dataService;
|
|
@Autowired
|
private TimeZoneConverterService timeZoneConverterService;
|
|
@ApiOperation(value = "行情")
|
@GetMapping("/api/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
|
Item bySymbol = itemService.findBySymbol(symbol);
|
if (bySymbol == null) {
|
throw new YamiShopBindException("symbol not exist");
|
}
|
String timeZone = timeZoneConverterService.getTimeZoneByItemCloseType(bySymbol.getOpenCloseType());
|
try {
|
if ("1quarter".equalsIgnoreCase(line)) {
|
line = Kline.PERIOD_QUARTER;
|
}
|
if ("1year".equalsIgnoreCase(line)) {
|
line = Kline.PERIOD_YEAR;
|
}
|
// 数据处理，取多少记录合适？ TODO
|
int size = 300;
|
KlinePeriodEnum klinePeriodType = KlinePeriodEnum.lineOf(line);
|
if (klinePeriodType != null) {
|
size = klinePeriodType.getMaxSize(); // 先返回 maxSize
|
}
|
List<Kline> data = this.dataService.kline(symbol, line);
|
//log.info("------> KlineController.getKline 获取到的 kline 集合长度为:{}", data.size());
|
int len = data.size();
|
if ("1day".equals(line)
|
|| "5day".equals(line)
|
|| "1mon".equals(line)
|
|| "1week".equals(line)
|
|| "quarter".equalsIgnoreCase(line)
|
|| "year".equalsIgnoreCase(line)) {
|
for (int i = 0; i < len; i++) {
|
Kline datum = data.get(i);
|
// // caster 2023-11-09 调整，基于盘口配置的时区，修改展示时间
|
// Date showTime = DateTimeTools.transferToShowTime(datum.getTs());
|
// datum.setCurrentTime(DateUtil.format(showTime, "yyyy-MM-dd"));
|
|
String dateStr = timeZoneConverterService.convertTimeZone(datum.getTs(), timeZone);
|
datum.setCurrentTime(dateStr.split(" ")[0]);
|
}
|
} else if ("1min".equals(line)) {
|
for (int i = 0; i < len; i++) {
|
Kline datum = data.get(i);
|
|
// // caster 2023-11-09 调整，基于盘口配置的时区，修改展示时间
|
// Date showTime = DateTimeTools.transferToShowTime(datum.getTs());
|
// datum.setCurrentTime(DateUtil.format(showTime, "HH:mm"));
|
String dateStr = timeZoneConverterService.convertTimeZone(datum.getTs(), timeZone);
|
datum.setCurrentTime(dateStr.split(" ")[1]);
|
}
|
} else {
|
for (int i = 0; i < len; i++) {
|
Kline datum = data.get(i);
|
|
// // caster 2023-11-09 调整，基于盘口配置的时区，修改展示时间
|
// Date showTime = DateTimeTools.transferToShowTime(datum.getTs());
|
// datum.setCurrentTime(DateUtil.format(showTime, "MM-dd HH:mm"));
|
String dateStr = timeZoneConverterService.convertTimeZone(datum.getTs(), timeZone);
|
String substring = dateStr.substring(dateStr.indexOf("-")+1);
|
datum.setCurrentTime(substring);
|
}
|
}
|
|
return Result.succeed(this.build(data, line, symbol));
|
} catch (Exception e) {
|
log.error("getKline error", e);
|
throw new YamiShopBindException("k线图获取失败");
|
}
|
}
|
|
private List<Map<String, Object>> build(List<Kline> data, String line, String symbol) {
|
//log.info("------> KlineContrller.build comein ...");
|
Set<Long> tsSet = new HashSet<>();
|
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);
|
}
|
|
int decimal = bySymbol.getDecimals();
|
double low = kline.getLow();
|
double high = kline.getHigh();
|
double open = kline.getOpen();
|
double close = kline.getClose();
|
if ( low == 0) {
|
continue;
|
}
|
if (high == 0) {
|
continue;
|
}
|
if (close == 0) {
|
continue;
|
}
|
if (open == 0) {
|
continue;
|
}
|
Map<String, Object> map = new HashMap<>();
|
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", BigDecimal.valueOf(open).setScale(decimal, RoundingMode.HALF_UP));
|
map.put("close", BigDecimal.valueOf(close).setScale(decimal, RoundingMode.HALF_UP));
|
map.put("high", BigDecimal.valueOf(high).setScale(decimal, RoundingMode.HALF_UP));
|
map.put("low", BigDecimal.valueOf(low).setScale(decimal, RoundingMode.HALF_UP));
|
map.put("volume", kline.getVolume());
|
list.add(map);
|
}
|
return list;
|
}
|
|
}
|
