bbgo_origin/pkg/exchange/ftx/stream.go

75 lines
1.4 KiB
Go
Raw Normal View History

package ftx
import (
"context"
"sync/atomic"
"github.com/c9s/bbgo/pkg/types"
)
type Stream struct {
types.StandardStream
wsService *WebsocketService
// publicOnly must be accessed atomically
publicOnly int32
}
func NewStream(key, secret string) *Stream {
wss := NewWebsocketService(key, secret)
s := &Stream{
StandardStream: types.StandardStream{},
wsService: wss,
}
return s
}
func (s *Stream) Connect(ctx context.Context) error {
return s.wsService.Connect(ctx)
}
func (s *Stream) SetPublicOnly() {
atomic.StoreInt32(&s.publicOnly, 1)
}
func (s *Stream) Close() error {
return s.wsService.Close()
}
func (s *Stream) Subscribe(channel types.Channel, symbol string, options types.SubscribeOptions) {
panic("implement me")
}
func (s *Stream) OnTradeUpdate(cb func(trade types.Trade)) {
panic("implement me")
}
func (s *Stream) OnOrderUpdate(cb func(order types.Order)) {
panic("implement me")
}
func (s *Stream) OnBalanceSnapshot(cb func(balances types.BalanceMap)) {
panic("implement me")
}
func (s *Stream) OnBalanceUpdate(cb func(balances types.BalanceMap)) {
panic("implement me")
}
func (s *Stream) OnKLineClosed(cb func(kline types.KLine)) {
panic("implement me")
}
func (s *Stream) OnKLine(cb func(kline types.KLine)) {
panic("implement me")
}
func (s *Stream) OnBookUpdate(cb func(book types.OrderBook)) {
panic("implement me")
}
func (s *Stream) OnBookSnapshot(cb func(book types.OrderBook)) {
panic("implement me")
}