mirror of
https://github.com/c9s/bbgo.git
synced 2024-09-21 00:31:10 +00:00
130 lines
3.1 KiB
Go
130 lines
3.1 KiB
Go
package kucoinapi
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
|
|
"github.com/c9s/bbgo/pkg/util"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// ApiClient defines the request builder method and request method for the API service
|
|
type ApiClient interface {
|
|
// NewAuthenticatedRequest builds up the http request for authentication-required endpoints
|
|
NewAuthenticatedRequest(method, refURL string, params url.Values, payload interface{}) (*http.Request, error)
|
|
|
|
// NewRequest builds up the http request for public endpoints
|
|
NewRequest(method, refURL string, params url.Values, body []byte) (*http.Request, error)
|
|
|
|
// SendRequest sends the request object to the api gateway
|
|
SendRequest(req *http.Request) (*util.Response, error)
|
|
}
|
|
|
|
type BulletService struct {
|
|
client *RestClient
|
|
}
|
|
|
|
func (s *BulletService) NewGetPublicBulletRequest() *GetPublicBulletRequest {
|
|
return &GetPublicBulletRequest{client: s.client}
|
|
}
|
|
|
|
func (s *BulletService) NewGetPrivateBulletRequest() *GetPrivateBulletRequest {
|
|
return &GetPrivateBulletRequest{client: s.client}
|
|
}
|
|
|
|
//go:generate requestgen -type GetPublicBulletRequest
|
|
type GetPublicBulletRequest struct {
|
|
client ApiClient
|
|
}
|
|
|
|
type Bullet struct {
|
|
InstanceServers []struct {
|
|
Endpoint string `json:"endpoint"`
|
|
Protocol string `json:"protocol"`
|
|
Encrypt bool `json:"encrypt"`
|
|
PingInterval int `json:"pingInterval"`
|
|
PingTimeout int `json:"pingTimeout"`
|
|
} `json:"instanceServers"`
|
|
Token string `json:"token"`
|
|
}
|
|
|
|
func (b *Bullet) PingInterval() time.Duration {
|
|
return time.Duration(b.InstanceServers[0].PingInterval) * time.Millisecond
|
|
}
|
|
|
|
func (b *Bullet) PingTimeout() time.Duration {
|
|
return time.Duration(b.InstanceServers[0].PingTimeout) * time.Millisecond
|
|
}
|
|
|
|
func (b *Bullet) URL() (*url.URL, error) {
|
|
if len(b.InstanceServers) == 0 {
|
|
return nil, errors.New("InstanceServers is empty")
|
|
}
|
|
|
|
u, err := url.Parse(b.InstanceServers[0].Endpoint)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
params := url.Values{}
|
|
params.Add("token", b.Token)
|
|
|
|
u.RawQuery = params.Encode()
|
|
return u, nil
|
|
}
|
|
|
|
func (r *GetPublicBulletRequest) Do(ctx context.Context) (*Bullet, error) {
|
|
req, err := r.client.NewRequest("POST", "/api/v1/bullet-public", nil, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
response, err := r.client.SendRequest(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var apiResponse struct {
|
|
Code string `json:"code"`
|
|
Message string `json:"msg"`
|
|
Data *Bullet `json:"data"`
|
|
}
|
|
|
|
if err := response.DecodeJSON(&apiResponse); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return apiResponse.Data, nil
|
|
}
|
|
|
|
//go:generate requestgen -type GetPrivateBulletRequest
|
|
type GetPrivateBulletRequest struct {
|
|
client ApiClient
|
|
}
|
|
|
|
func (r *GetPrivateBulletRequest) Do(ctx context.Context) (*Bullet, error) {
|
|
req, err := r.client.NewAuthenticatedRequest("POST", "/api/v1/bullet-private", nil, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
response, err := r.client.SendRequest(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var apiResponse struct {
|
|
Code string `json:"code"`
|
|
Message string `json:"msg"`
|
|
Data *Bullet `json:"data"`
|
|
}
|
|
|
|
if err := response.DecodeJSON(&apiResponse); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return apiResponse.Data, nil
|
|
}
|