bbgo_origin/pkg/exchange/ftx/stream.go

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")
}