fix connectivity initialization

c9s · Nov 18, 2024 · 9251f12 · 9251f12
1 parent 10c289a
commit 9251f12
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/pkg/bbgo/session.go b/pkg/bbgo/session.go
@@ -134,12 +134,13 @@ type ExchangeSession struct {
 
 func NewExchangeSession(name string, exchange types.Exchange) *ExchangeSession {
 	userDataStream := exchange.NewStream()
-	userDataConnectivity := types.NewConnectivity()
-	userDataConnectivity.Bind(userDataStream)
 
 	marketDataStream := exchange.NewStream()
 	marketDataStream.SetPublicOnly()
 
+	userDataConnectivity := types.NewConnectivity()
+	userDataConnectivity.Bind(userDataStream)
+
 	marketDataConnectivity := types.NewConnectivity()
 	marketDataConnectivity.Bind(marketDataStream)
 
@@ -870,6 +871,14 @@ func (session *ExchangeSession) InitExchange(name string, ex types.Exchange) err
 	session.MarketDataStream = ex.NewStream()
 	session.MarketDataStream.SetPublicOnly()
 
+	session.UserDataConnectivity = types.NewConnectivity()
+	session.UserDataConnectivity.Bind(session.UserDataStream)
+
+	session.MarketDataConnectivity = types.NewConnectivity()
+	session.MarketDataConnectivity.Bind(session.MarketDataStream)
+
+	session.Connectivity = types.NewConnectivityGroup(session.MarketDataConnectivity, session.MarketDataConnectivity)
+
 	// pointer fields
 	session.Subscriptions = make(map[types.Subscription]types.Subscription)
 	session.Account = &types.Account{}