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

davidblewett · web-flow · commit 9b44abacae4a · 2024-01-11T17:24:14.000-05:00
Return back to the caller on rebalance events
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 {

Original file line number	Diff line number	Diff line change
`@@ -144,9 +144,15 @@ where`
`144`	`144`	`}`
`145`	`145`	`rdsys::RD_KAFKA_EVENT_REBALANCE => {`
`146`	`146`	`self.handle_rebalance_event(event);`
	`147`	`+ if timeout != Timeout::Never {`
	`148`	`+ return None;`
	`149`	`+ }`
`147`	`150`	`}`
`148`	`151`	`rdsys::RD_KAFKA_EVENT_OFFSET_COMMIT => {`
`149`	`152`	`self.handle_offset_commit_event(event);`
	`153`	`+ if timeout != Timeout::Never {`
	`154`	`+ return None;`
	`155`	`+ }`
`150`	`156`	`}`
`151`	`157`	`_ => {`
`152`	`158`	`let buf = unsafe {`