From 8f967361e88ba8947f641633710399aa4910667a Mon Sep 17 00:00:00 2001
From: nquidox <nquidox@gmail.com>
Date: Thu, 2 Apr 2026 16:09:17 +0300
Subject: [PATCH] connection check

---
 handler.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/handler.go b/handler.go
index e54146b..61dc5dc 100644
--- a/handler.go
+++ b/handler.go
@@ -99,7 +99,13 @@ func (c *Client) connectAndSignal(addr string, timeout time.Duration) error {
 		if res.err != nil {
 			return res.err
 		}
-		c.connection = res.conn
+
+		c.changeConnection(res.conn)
+		if err := c.init(res.conn); err != nil {
+			res.conn.Close()
+			return fmt.Errorf("init failed: %w", err)
+		}
+
 		close(c.connected)
 		return nil
 	}