From 5b674b3a1d845a2857e8c799e43b7279dbaaa7b8 Mon Sep 17 00:00:00 2001
From: Sushant Mahajan <smahajan@confluent.io>
Date: Mon, 17 Jun 2024 21:26:40 +0530
Subject: [PATCH] MINOR: Override doWork for fast RPC handling.

---
 .../server/group/share/PersisterStateManager.java      | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/server-common/src/main/java/org/apache/kafka/server/group/share/PersisterStateManager.java b/server-common/src/main/java/org/apache/kafka/server/group/share/PersisterStateManager.java
index 8d67e8dd2671c..dc1810af1b4f9 100644
--- a/server-common/src/main/java/org/apache/kafka/server/group/share/PersisterStateManager.java
+++ b/server-common/src/main/java/org/apache/kafka/server/group/share/PersisterStateManager.java
@@ -495,5 +495,15 @@ public Collection<RequestAndCompletionHandler> generateRequests() {
     public void enqueue(PersisterStateManagerHandler handler) {
       queue.add(handler);
     }
+
+    @Override
+    public void doWork() {
+      try {
+        TimeUnit.MILLISECONDS.sleep(10);
+        this.pollOnce(100L);
+      } catch (Exception e) {
+        log.error("Timed out", e);
+      }
+    }
   }
 }
\ No newline at end of file