vamossagar12 commented on code in PR #15305: URL: https://github.com/apache/kafka/pull/15305#discussion_r1604759527
########## connect/runtime/src/main/java/org/apache/kafka/connect/runtime/distributed/WorkerRebalanceListener.java: ########## @@ -36,4 +37,15 @@ public interface WorkerRebalanceListener { * or tasks might refer to all or some of the connectors and tasks running on the worker. */ void onRevoked(String leader, Collection<String> connectors, Collection<ConnectorTaskId> tasks); + + + /** + * Invoked when a worker experiences a poll timeout expiry. Invoking this method allows getting + * the stage which was currently being executed when the poll timeout happened. The default implementation + * returns null + * @return The current stage being executed. Could be null + */ + default Stage onPollTimeoutExpiry() { Review Comment: I added the default bit to avoid changing the tests. Turns out it should be ok to modify them so I removed the default bit. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: jira-unsubscr...@kafka.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org