74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package market
|
||
|
||
import (
|
||
"encoding/json"
|
||
"errors"
|
||
"strconv"
|
||
|
||
"git.apinb.com/bsm-sdk/core/utils"
|
||
"git.apinb.com/quant/strategy/types"
|
||
)
|
||
|
||
var (
|
||
KlineUrl = "http://market.senlin.ai/ticker/klines"
|
||
)
|
||
|
||
func FetchKlines(symbol, interval string, limit int) ([]*types.KLine, []float64, error) {
|
||
url := KlineUrl + "?symbol=" + symbol + "&interval=" + interval + "&limit=" + strconv.Itoa(limit)
|
||
body, err := utils.HttpGet(url)
|
||
if err != nil {
|
||
return nil, nil, err
|
||
}
|
||
|
||
var data []*types.KLine
|
||
err = json.Unmarshal(body, &data)
|
||
if err != nil {
|
||
return nil, nil, err
|
||
}
|
||
|
||
if len(data) < 1 {
|
||
return nil, nil, errors.New("no data")
|
||
}
|
||
|
||
var closes []float64
|
||
for _, kline := range data {
|
||
closes = append(closes, kline.Close)
|
||
}
|
||
|
||
return data, closes, nil
|
||
}
|
||
|
||
// 获取最后两个K线
|
||
// 返回：上一个K线, 当前K线, 错误
|
||
func GetLastKline(symbol, interval string) (*types.KLine, *types.KLine, error) {
|
||
url := KlineUrl + "?symbol=" + symbol + "&interval=" + interval + "&limit=2"
|
||
body, err := utils.HttpGet(url)
|
||
if err != nil {
|
||
return nil, nil, err
|
||
}
|
||
|
||
var data []*types.KLine
|
||
err = json.Unmarshal(body, &data)
|
||
if err != nil {
|
||
return nil, nil, err
|
||
}
|
||
|
||
if len(data) < 2 {
|
||
return nil, nil, errors.New("no data")
|
||
}
|
||
|
||
data[1].PriceRate = (data[1].Close - data[1].Open) / data[1].Open
|
||
data[0].PriceRate = (data[0].Close - data[0].Open) / data[0].Open
|
||
|
||
return data[0], data[1], nil
|
||
}
|
||
|
||
func UnmarshalKline(body string) (*types.KLine, error) {
|
||
var data types.KLine
|
||
err := json.Unmarshal([]byte(body), &data)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return &data, nil
|
||
}
|