[PLAT-15377] Adding a global uncaught exception handler to yugaware

Summary: [PLAT-15377] Adding a global uncaught exception handler to yugaware

Test Plan: build locally, phabricator

Reviewers: sanketh, svarshney, nsingh

Reviewed By: svarshney, nsingh

Subscribers: yugaware

Differential Revision: https://phorge.dev.yugabyte.com/D38344

managed/src/main/java/GlobalExceptionHandler.java

@@ -0,0 +1,20 @@
+// Copyright (c) Yugabyte, Inc.
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class GlobalExceptionHandler implements Thread.UncaughtExceptionHandler {
+
+  private static Logger logger = LoggerFactory.getLogger(GlobalExceptionHandler.class);
+
+  // Added for test
+  public static void setLogger(Logger l) {
+    logger = l;
+  }
+
+  @Override
+  public void uncaughtException(Thread t, Throwable e) {
+    // Log the exception
+    logger.error("Yugaware uncaught exception in thread '{}'", t.getName(), e);
+  }
+}

managed/src/main/java/MainModule.java

@@ -135,6 +135,8 @@ public MainModule(Environment environment, Config config) {
 
   @Override
   public void configure() {
+    // Bind the uncaught exception handler at the application startup
+    Thread.setDefaultUncaughtExceptionHandler(new GlobalExceptionHandler());
     bind(StaticInjectorHolder.class).asEagerSingleton();
     bind(Long.class)
         .annotatedWith(Names.named("AppStartupTimeMs"))

managed/src/main/java/com/yugabyte/yw/commissioner/TaskExecutor.java

@@ -723,10 +723,10 @@ private void waitForSubTasks(boolean abortOnFailure) {
             anyEx = (anyEx != null) ? anyEx : thisEx;
             runnableSubTask.updateTaskDetailsOnError(TaskInfo.State.Aborted, thisEx);
             removeCompletedSubTask(iter, runnableSubTask, anyEx);
-          } catch (Exception e) {
-            anyEx = e;
-            runnableSubTask.updateTaskDetailsOnError(TaskInfo.State.Failure, e);
-            removeCompletedSubTask(iter, runnableSubTask, e);
+          } catch (Throwable th) {
+            anyEx = th;
+            runnableSubTask.updateTaskDetailsOnError(TaskInfo.State.Failure, th);
+            removeCompletedSubTask(iter, runnableSubTask, th);
           }
         }
       }

managed/src/test/java/GlobalExceptionHandlerTest.java

@@ -0,0 +1,28 @@
+// Copyright (c) Yugabyte, Inc.
+
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+
+import org.junit.Test;
+import org.slf4j.Logger;
+
+public class GlobalExceptionHandlerTest {
+
+  @Test
+  public void testUncaughtExceptionLogsError() throws Exception {
+    Logger mockLogger = mock(Logger.class);
+
+    // Create an instance of the handler
+    GlobalExceptionHandler handler = new GlobalExceptionHandler();
+    GlobalExceptionHandler.setLogger(mockLogger);
+
+    Thread thread = new Thread(() -> {});
+    Throwable exception = new RuntimeException("Test exception");
+
+    handler.uncaughtException(thread, exception);
+
+    // Verify that the logger was called with the expected message
+    verify(mockLogger)
+        .error("Yugaware uncaught exception in thread '{}'", thread.getName(), exception);
+  }
+}