lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-26 00:35:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

rebalance: rename methods

Browse Source
This commit is contained in:
なるみ 2022-06-15 21:34:42 +08:00
parent 694c226bc0
commit f19e1fdf87
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/strategy/rebalance/strategy.go
View File

@ -75,14 +75,14 @@ func (s *Strategy) Validate() error {
} }
func (s *Strategy) Subscribe(session *bbgo.ExchangeSession) { func (s *Strategy) Subscribe(session *bbgo.ExchangeSession) {
for _, symbol := range s.getSymbols() { for _, symbol := range s.symbols() {
session.Subscribe(types.KLineChannel, symbol, types.SubscribeOptions{Interval: s.Interval}) session.Subscribe(types.KLineChannel, symbol, types.SubscribeOptions{Interval: s.Interval})
} }
} }
func (s *Strategy) Run(ctx context.Context, orderExecutor bbgo.OrderExecutor, session *bbgo.ExchangeSession) error { func (s *Strategy) Run(ctx context.Context, orderExecutor bbgo.OrderExecutor, session *bbgo.ExchangeSession) error {
s.activeOrderBooks = make(map[string]*bbgo.ActiveOrderBook) s.activeOrderBooks = make(map[string]*bbgo.ActiveOrderBook)
for _, symbol := range s.getSymbols() { for _, symbol := range s.symbols() {
activeOrderBook := bbgo.NewActiveOrderBook(symbol) activeOrderBook := bbgo.NewActiveOrderBook(symbol)
activeOrderBook.BindStream(session.UserDataStream) activeOrderBook.BindStream(session.UserDataStream)
s.activeOrderBooks[symbol] = activeOrderBook s.activeOrderBooks[symbol] = activeOrderBook
@ -106,7 +106,7 @@ func (s *Strategy) rebalance(ctx context.Context, orderExecutor bbgo.OrderExecut
} }
} }
prices, err := s.getPrices(ctx, session) prices, err := s.prices(ctx, session)
if err != nil { if err != nil {
return return
} }
@ -135,7 +135,7 @@ func (s *Strategy) rebalance(ctx context.Context, orderExecutor bbgo.OrderExecut
} }
} }
func (s *Strategy) getPrices(ctx context.Context, session *bbgo.ExchangeSession) (prices types.Float64Slice, err error) { func (s *Strategy) prices(ctx context.Context, session *bbgo.ExchangeSession) (prices types.Float64Slice, err error) {
for _, currency := range s.currencies { for _, currency := range s.currencies {
if currency == s.BaseCurrency { if currency == s.BaseCurrency {
prices = append(prices, 1.0) prices = append(prices, 1.0)
@ -232,7 +232,7 @@ func (s *Strategy) generateSubmitOrders(prices, marketValues types.Float64Slice)
return submitOrders return submitOrders
} }
func (s *Strategy) getSymbols() (symbols []string) { func (s *Strategy) symbols() (symbols []string) {
for _, currency := range s.currencies { for _, currency := range s.currencies {
if currency == s.BaseCurrency { if currency == s.BaseCurrency {
continue continue