Reconnection will always be needed here!

ThePrimeagen · May 26, 2024 · e8e11b9 · e8e11b9
1 parent 9561470
commit e8e11b9
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/pkg/v2/distributor/downstream.go b/pkg/v2/distributor/downstream.go
@@ -23,13 +23,10 @@ func NewDownstream(addr string) *Downstream {
 func (ds *Downstream) Run() {
 	var err error
 	for {
-		// Reconnect if necessary
-		if ds.conn == nil {
-			slog.Info("Connecting to downstream server", "addr", ds.addr)
-			u := url.URL{Scheme: "ws", Host: ds.addr, Path: "/ws"}
-			ds.conn, _, err = websocket.DefaultDialer.Dial(u.String(), nil)
-			assert.NoError(err, "unable to connect to downstream server "+ds.addr)
-		}
+		slog.Info("Connecting to downstream server", "addr", ds.addr)
+		u := url.URL{Scheme: "ws", Host: ds.addr, Path: "/ws"}
+		ds.conn, _, err = websocket.DefaultDialer.Dial(u.String(), nil)
+		assert.NoError(err, "unable to connect to downstream server "+ds.addr)
 
 		for {
 			// Discard any incoming messages