producer: use ErrClientClosed properly if waiting to produce

Previously we would just return the client context, but we may as well properly promise ErrClientClosed.
twmb · Jul 8, 2021 · 8f1e732 · 8f1e732
1 parent ee918d9
commit 8f1e732
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/kgo/producer.go b/pkg/kgo/producer.go
@@ -291,7 +291,7 @@ func (cl *Client) Produce(
 		select {
 		case <-p.waitBuffer:
 		case <-cl.ctx.Done():
-			drainBuffered(cl.ctx.Err())
+			drainBuffered(ErrClientClosed)
 			return
 		case <-ctx.Done():
 			drainBuffered(ctx.Err())