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

threadedstream · web-flow · commit 51cea28b9094 · 2024-02-23T10:16:24.000+01:00
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 {

Original file line number	Diff line number	Diff line change
`@@ -265,6 +265,8 @@ func (ch clickhouse) acquire(ctx context.Context) (conn connect, err error) {`
`265`	`265`	`select {`
`266`	`266`	`case <-timer.C:`
`267`	`267`	`return nil, ErrAcquireConnTimeout`
	`268`	`+ case <-ctx.Done():`
	`269`	`+ return nil, ctx.Err()`
`268`	`270`	`case ch.open <- struct{}{}:`
`269`	`271`	`}`
`270`	`272`	`select {`