fix lock

Author: Quinn-With-Two-Ns

temporal-sdk/src/main/java/io/temporal/internal/worker/AsyncPoller.java

@@ -284,6 +284,7 @@ public void run() {
         } catch (PollTaskAsyncAbort ab) {
           abort = true;
         } catch (Throwable e) {
+          log.error("Error in process task loop", e);
           if (e instanceof InterruptedException) {
             // we restore the flag here, so it can be checked and processed (with exit) in finally.
             Thread.currentThread().interrupt();
@@ -370,24 +371,28 @@ void removePoller(String pollerName) {
     void balance(String p) throws InterruptedException {
       while (!shouldTerminate()) {
         balancerLock.lock();
-        // If this poller has no tasks then we can unblock immediately
-        if (taskCounts.get(p).get() == 0) {
-          balancerLock.unlock();
-          return;
-        }
-        // Check if all tasks have at least one poll request
-        boolean allOtherTasksHavePolls = true;
-        for (String task : taskCounts.keySet()) {
-          if (!Objects.equals(task, p) && taskCounts.get(task).get() == 0) {
-            allOtherTasksHavePolls = false;
-            break;
+        try {
+          // If this poller has no tasks then we can unblock immediately
+          if (taskCounts.get(p).get() == 0) {
+            return;
           }
-        }
-        if (!allOtherTasksHavePolls) {
-          balancerCondition.await();
-        } else {
+          // Check if all tasks have at least one poll request
+          boolean allOtherTasksHavePolls = true;
+          for (String task : taskCounts.keySet()) {
+            if (!Objects.equals(task, p) && taskCounts.get(task).get() == 0) {
+              allOtherTasksHavePolls = false;
+              break;
+            }
+          }
+          if (!allOtherTasksHavePolls) {
+            log.info(
+                "Waiting for other tasks to have at least one poll request before proceeding with task");
+            balancerCondition.await();
+          } else {
+            return;
+          }
+        } finally {
           balancerLock.unlock();
-          return;
         }
       }
     }