(kafka) branch trunk updated: MINOR: Fix stage recording in DistributedHerder::addRequest (#16784)

[cegerton]

This is an automated email from the ASF dual-hosted git repository.

cegerton pushed a commit to branch trunk
in repository https://gitbox.apache.org/repos/asf/kafka.git


The following commit(s) were added to refs/heads/trunk by this push:
     new 3f3e3767976 MINOR: Fix stage recording in 
DistributedHerder::addRequest (#16784)
3f3e3767976 is described below

commit 3f3e3767976159d1291d81699d53618c3cc224eb
Author: Chris Egerton <chr...@aiven.io>
AuthorDate: Mon Aug 5 20:36:39 2024 +0200

    MINOR: Fix stage recording in DistributedHerder::addRequest (#16784)
    
    Reviewers: Greg Harris <greg.har...@aiven.io>
---
 .../org/apache/kafka/connect/runtime/distributed/DistributedHerder.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git 
a/connect/runtime/src/main/java/org/apache/kafka/connect/runtime/distributed/DistributedHerder.java
 
b/connect/runtime/src/main/java/org/apache/kafka/connect/runtime/distributed/DistributedHerder.java
index 174adea8cdd..c05e9ce0b68 100644
--- 
a/connect/runtime/src/main/java/org/apache/kafka/connect/runtime/distributed/DistributedHerder.java
+++ 
b/connect/runtime/src/main/java/org/apache/kafka/connect/runtime/distributed/DistributedHerder.java
@@ -2382,6 +2382,7 @@ public class DistributedHerder extends AbstractHerder 
implements Runnable {
     }
 
     DistributedHerderRequest addRequest(long delayMs, Callable<Void> action, 
Callback<Void> callback) {
+        callback.recordStage(tickThreadStage);
         DistributedHerderRequest req = new 
DistributedHerderRequest(time.milliseconds() + delayMs, 
requestSeqNum.incrementAndGet(), action, callback);
         requests.add(req);
         // We don't need to synchronize here
@@ -2393,7 +2394,6 @@ public class DistributedHerder extends AbstractHerder 
implements Runnable {
         // queue was added
         if (peekWithoutException() == req)
             member.wakeup();
-        callback.recordStage(tickThreadStage);
         return req;
     }