qbtrade/pkg/exchange/bybit/bybitapi/get_k_lines_request.go
2024-06-27 22:42:38 +08:00

108 lines
3.0 KiB
Go

package bybitapi
import (
"encoding/json"
"fmt"
"time"
"github.com/c9s/requestgen"
"git.qtrade.icu/lychiyu/qbtrade/pkg/fixedpoint"
"git.qtrade.icu/lychiyu/qbtrade/pkg/types"
)
//go:generate -command GetRequest requestgen -method GET -responseType .APIResponse -responseDataField Result
//go:generate -command PostRequest requestgen -method POST -responseType .APIResponse -responseDataField Result
type IntervalSign string
const (
IntervalSignDay IntervalSign = "D"
IntervalSignWeek IntervalSign = "W"
IntervalSignMonth IntervalSign = "M"
)
type KLinesResponse struct {
Symbol string `json:"symbol"`
// An string array of individual candle
// Sort in reverse by startTime
List []KLine `json:"list"`
Category Category `json:"category"`
}
type KLine struct {
// list[0]: startTime, Start time of the candle (ms)
StartTime types.MillisecondTimestamp
// list[1]: openPrice
Open fixedpoint.Value
// list[2]: highPrice
High fixedpoint.Value
// list[3]: lowPrice
Low fixedpoint.Value
// list[4]: closePrice
Close fixedpoint.Value
// list[5]: volume, Trade volume. Unit of contract: pieces of contract. Unit of spot: quantity of coins
Volume fixedpoint.Value
// list[6]: turnover, Turnover. Unit of figure: quantity of quota coin
TurnOver fixedpoint.Value
}
const KLinesArrayLen = 7
func (k *KLine) UnmarshalJSON(data []byte) error {
var jsonArr []json.RawMessage
err := json.Unmarshal(data, &jsonArr)
if err != nil {
return fmt.Errorf("failed to unmarshal jsonRawMessage: %v, err: %w", string(data), err)
}
if len(jsonArr) != KLinesArrayLen {
return fmt.Errorf("unexpected K Lines array length: %d, exp: %d", len(jsonArr), KLinesArrayLen)
}
err = json.Unmarshal(jsonArr[0], &k.StartTime)
if err != nil {
return fmt.Errorf("failed to unmarshal resp index 0: %v, err: %w", string(jsonArr[0]), err)
}
values := make([]fixedpoint.Value, len(jsonArr)-1)
for i, jsonRaw := range jsonArr[1:] {
err = json.Unmarshal(jsonRaw, &values[i])
if err != nil {
return fmt.Errorf("failed to unmarshal resp index %d: %v, err: %w", i+1, string(jsonRaw), err)
}
}
k.Open = values[0]
k.High = values[1]
k.Low = values[2]
k.Close = values[3]
k.Volume = values[4]
k.TurnOver = values[5]
return nil
}
//go:generate GetRequest -url "/v5/market/kline" -type GetKLinesRequest -responseDataType .KLinesResponse
type GetKLinesRequest struct {
client requestgen.APIClient
category Category `param:"category,query" validValues:"spot"`
symbol string `param:"symbol,query"`
// Kline interval.
// - 1,3,5,15,30,60,120,240,360,720: minute
// - D: day
// - M: month
// - W: week
interval string `param:"interval,query" validValues:"1,3,5,15,30,60,120,240,360,720,D,W,M"`
startTime *time.Time `param:"start,query,milliseconds"`
endTime *time.Time `param:"end,query,milliseconds"`
// Limit for data size per page. [1, 1000]. Default: 200
limit *uint64 `param:"limit,query"`
}
func (c *RestClient) NewGetKLinesRequest() *GetKLinesRequest {
return &GetKLinesRequest{
client: c,
category: CategorySpot,
}
}