Merge pull request #241 from filimonov/patch-1

fix: reset WriteDeadline
ClickHouse · Jan 26, 2023 · a409ea7 · a409ea7
2 parents 2fad15a + f2e3022
commit a409ea7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/client.go b/client.go
@@ -254,6 +254,8 @@ func (c *Client) flushBuf(ctx context.Context, b *proto.Buffer) error {
 		if err := c.conn.SetWriteDeadline(deadline); err != nil {
 			return errors.Wrap(err, "set write deadline")
 		}
+		// Reset deadline.
+		defer func() { _ = c.conn.SetWriteDeadline(time.Time{}) }()
 	}
 	n, err := c.conn.Write(b.Buf)
 	if err != nil {