Merge pull request #644 from fede1024/scanterog/return-none-on-rebalance

Return back to the caller on rebalance events
fede1024 · Jan 11, 2024 · 9b44aba · 9b44aba
2 parents 738590b + d6ec6e1
commit 9b44aba
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/consumer/base_consumer.rs b/src/consumer/base_consumer.rs
@@ -144,9 +144,15 @@ where
                     }
                     rdsys::RD_KAFKA_EVENT_REBALANCE => {
                         self.handle_rebalance_event(event);
+                        if timeout != Timeout::Never {
+                            return None;
+                        }
                     }
                     rdsys::RD_KAFKA_EVENT_OFFSET_COMMIT => {
                         self.handle_offset_commit_event(event);
+                        if timeout != Timeout::Never {
+                            return None;
+                        }
                     }
                     _ => {
                         let buf = unsafe {