fix: fix 877

apache · Jul 8, 2023 · e5a8464 · e5a8464
1 parent 43986b1
commit e5a8464
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/pulsar/producer_partition.go b/pulsar/producer_partition.go
@@ -1095,9 +1095,13 @@ func (p *partitionProducer) Send(ctx context.Context, msg *ProducerMessage) (Mes
 	}, true)
 
 	// wait for send request to finish
-	<-doneCh
-
-	return msgID, err
+	select {
+	case <-ctx.Done():
+		return nil, ctx.Err()
+	case <-doneCh:
+		// send request has been finished
+		return msgID, err
+	}
 }
 
 func (p *partitionProducer) SendAsync(ctx context.Context, msg *ProducerMessage,