From 2b57b0706eb75d67f7e2bea4ab02460a798ec78b Mon Sep 17 00:00:00 2001
From: d1slike <merser1q@gmail.com>
Date: Tue, 13 Feb 2024 20:18:48 +0300
Subject: [PATCH] Set must reconnect = false only in error case

---
 client.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/client.go b/client.go
index b22fdcd..8ba6d23 100644
--- a/client.go
+++ b/client.go
@@ -89,7 +89,9 @@ func (s *Client) Run(ctx context.Context) error {
 	case <-ctx.Done():
 		return ctx.Err()
 	case err := <-s.firstOccurredError:
-		s.mustReconnect.Store(false)
+		if err != nil {
+			s.mustReconnect.Store(false)
+		}
 		return err
 	}
 }