handle ctx.Done() in acquire (#1199)

ClickHouse · Feb 23, 2024 · 51cea28 · 51cea28
1 parent 98ddea0
commit 51cea28
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/clickhouse.go b/clickhouse.go
@@ -265,6 +265,8 @@ func (ch *clickhouse) acquire(ctx context.Context) (conn *connect, err error) {
 	select {
 	case <-timer.C:
 		return nil, ErrAcquireConnTimeout
+	case <-ctx.Done():
+		return nil, ctx.Err()
 	case ch.open <- struct{}{}:
 	}
 	select {