fix nil pointer panics

centrifugal · Jul 22, 2023 · 04c6b00 · 04c6b00
1 parent f41ab69
commit 04c6b00
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/client/handler.go b/internal/client/handler.go
@@ -101,7 +101,7 @@ func (h *Handler) Setup() error {
 	}
 
 	var connectStreamHandler proxystream.ConnectHandlerFunc
-	if h.proxyMap.StreamProxyMap.ConnectProxy != nil {
+	if streamHandler != nil && h.proxyMap.StreamProxyMap.ConnectProxy != nil {
 		connectStreamHandler = streamHandler.HandleConnect(h.node)
 	}
 
@@ -138,7 +138,7 @@ func (h *Handler) Setup() error {
 	}
 
 	var proxySubscribeStreamHandler proxystream.SubscribeHandlerFunc
-	if len(h.proxyMap.StreamProxyMap.SubscribeProxies) > 0 {
+	if streamHandler != nil && len(h.proxyMap.StreamProxyMap.SubscribeProxies) > 0 {
 		proxySubscribeStreamHandler = streamHandler.HandleSubscribe(h.node)
 	}