Simplify BaseProducer::poll to rely on From impl.

fede1024 · Dec 12, 2024 · ac574f2 · ac574f2
1 parent dcd606b
commit ac574f2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/src/client.rs b/src/client.rs
@@ -300,10 +300,10 @@ impl<C: ClientContext> Client<C> {
         &self.context
     }
 
-    pub(crate) fn poll_event(
+    pub(crate) fn poll_event<T: Into<Timeout>>(
         &self,
         queue: &NativeQueue,
-        timeout: Timeout,
+        timeout: T,
     ) -> EventPollResult<NativeEvent> {
         let event = unsafe { NativeEvent::from_ptr(queue.poll(timeout)) };
         if let Some(ev) = event {

diff --git a/src/producer/base_producer.rs b/src/producer/base_producer.rs
@@ -362,9 +362,7 @@ where
     pub fn poll<T: Into<Timeout>>(&self, timeout: T) {
         let deadline: Deadline = timeout.into().into();
         loop {
-            let event = self
-                .client()
-                .poll_event(&self.queue, Timeout::After(deadline.remaining()));
+            let event = self.client().poll_event(&self.queue, &deadline);
             if let EventPollResult::Event(ev) = event {
                 let evtype = unsafe { rdsys::rd_kafka_event_type(ev.ptr()) };
                 match evtype {