Clean up the error

orkes-io · Oct 10, 2022 · 9a50ebd · 9a50ebd
1 parent abbfd0b
commit 9a50ebd
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/main/java/io/orkes/conductor/client/grpc/PooledPoller.java b/src/main/java/io/orkes/conductor/client/grpc/PooledPoller.java
@@ -236,10 +236,12 @@ public void onCompleted() {
 
     private void drain() {
         long didntGetMessageCount = lastAskedForMessageCount.get();
-        log.info("Didn't get {} messages", didntGetMessageCount);
-        for (int i = 0; i < didntGetMessageCount; i++) {
-            this.saveTask(TaskPb.Task.newBuilder().setTaskId("NO_OP").build());
-            semaphore.release();
+        if(didntGetMessageCount > 0) {
+            log.info("Didn't get {} messages from server as expected", didntGetMessageCount);
+            for (int i = 0; i < didntGetMessageCount; i++) {
+                this.saveTask(TaskPb.Task.newBuilder().setTaskId("NO_OP").build());
+                semaphore.release();
+            }
         }
         callAgain.set(true);
     }