mm: Close book feeds

Book feeds were not being closed, leading to warning logs.
martonp · Aug 11, 2024 · 72df9f8 · 72df9f8
1 parent 1efedb4
commit 72df9f8
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/client/mm/exchange_adaptor.go b/client/mm/exchange_adaptor.go
@@ -2297,6 +2297,7 @@ func (u *unifiedExchangeAdaptor) cancelAllOrders(ctx context.Context) {
 		u.tryCancelOrders(ctx, nil, true)
 		return
 	}
+	defer bookFeed.Close()
 
 	mktCfg, err := u.clientCore.ExchangeMarket(u.host, u.baseID, u.quoteID)
 	if err != nil {

diff --git a/client/mm/mm_arb_market_maker.go b/client/mm/mm_arb_market_maker.go
@@ -448,6 +448,7 @@ func (a *arbMarketMaker) botLoop(ctx context.Context) (*sync.WaitGroup, error) {
 					a.rebalance(epoch.Current)
 				}
 			case <-ctx.Done():
+				bookFeed.Close()
 				return
 			}
 		}

diff --git a/client/mm/mm_basic.go b/client/mm/mm_basic.go
@@ -395,6 +395,7 @@ func (m *basicMarketMaker) botLoop(ctx context.Context) (*sync.WaitGroup, error)
 					m.rebalance(epoch.Current)
 				}
 			case <-ctx.Done():
+				bookFeed.Close()
 				return
 			}
 		}

diff --git a/client/mm/mm_simple_arb.go b/client/mm/mm_simple_arb.go
@@ -465,6 +465,7 @@ func (a *simpleArbMarketMaker) botLoop(ctx context.Context) (*sync.WaitGroup, er
 					a.rebalance(epoch.Current)
 				}
 			case <-ctx.Done():
+				bookFeed.Close()
 				return
 			}
 		}