Don't block forever in poll when flushing.

davidblewett · davidblewett · commit 91bf1a688ee8 · 2024-12-12T11:55:19.000-05:00
diff --git a/src/producer/base_producer.rs b/src/producer/base_producer.rs
@@ -506,7 +506,11 @@ where
             if ret.is_error() {
                 return Err(KafkaError::Flush(ret.into()));
             } else {
-                self.poll(&deadline);
+                if let Deadline::Never = &deadline {
+                    self.poll(Timeout::After(Duration::ZERO));
+                } else {
+                    self.poll(&deadline);
+                }
             }
         }
         Ok(())

Original file line number	Diff line number	Diff line change
`@@ -506,7 +506,11 @@ where`
`506`	`506`	`if ret.is_error() {`
`507`	`507`	`return Err(KafkaError::Flush(ret.into()));`
`508`	`508`	`} else {`
`509`		`- self.poll(&deadline);`
	`509`	`+ if let Deadline::Never = &deadline {`
	`510`	`+ self.poll(Timeout::After(Duration::ZERO));`
	`511`	`+ } else {`
	`512`	`+ self.poll(&deadline);`
	`513`	`+ }`
`510`	`514`	`}`
`511`	`515`	`}`
`512`	`516`	`Ok(())`