From fad951964505823720fe5df4e607d2200e59bc39 Mon Sep 17 00:00:00 2001
From: Nikita Merkulov <n.al.merkulov@gmail.com>
Date: Mon, 6 Oct 2025 17:55:22 +0300
Subject: [PATCH] wait untill processImpl completes before closing stream chan

---
 conn_process.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/conn_process.go b/conn_process.go
index 21872f73c7..507aa727a0 100644
--- a/conn_process.go
+++ b/conn_process.go
@@ -1,4 +1,3 @@
-
 package clickhouse
 
 import (
@@ -117,6 +116,11 @@ func (c *connect) process(ctx context.Context, on *onProcess) error {
 	select {
 	case <-ctx.Done():
 		c.cancel()
+		// Wait for goroutine to finish before returning
+		select {
+		case <-errCh:
+		case <-doneCh:
+		}
 		return ctx.Err()
 
 	case err := <-errCh: