prevent stage information from getting lost in the log

cdapio · Jan 30, 2025 · 9f20a76 · 9f20a76
1 parent 71536cc
commit 9f20a76
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/...bric/src/main/java/io/cdap/cdap/internal/app/runtime/ProgramControllerServiceAdapter.java b/...bric/src/main/java/io/cdap/cdap/internal/app/runtime/ProgramControllerServiceAdapter.java
@@ -16,8 +16,8 @@
 
 package io.cdap.cdap.internal.app.runtime;
 
-import com.google.common.base.Throwables;
 import com.google.common.util.concurrent.Service;
+import io.cdap.cdap.api.exception.WrappedStageException;
 import io.cdap.cdap.app.runtime.ProgramController;
 import io.cdap.cdap.common.conf.Constants;
 import io.cdap.cdap.common.logging.Loggers;
@@ -94,7 +94,7 @@ public void running() {
 
       @Override
       public void failed(Service.State from, Throwable failure) {
-        Throwable rootCause = Throwables.getRootCause(failure);
+        Throwable rootCause = getRootCause(failure);
         LOG.error("{} Program '{}' failed.", getProgramRunId().getType(),
             getProgramRunId().getProgram(), failure);
         USER_LOG.error(
@@ -104,6 +104,18 @@ public void failed(Service.State from, Throwable failure) {
         error(failure);
       }
 
+      private Throwable getRootCause(Throwable failure) {
+        Throwable cause;
+        while ((cause = failure.getCause()) != null) {
+          failure = cause;
+          if (WrappedStageException.class.isAssignableFrom(failure.getClass())) {
+            // to prevent stage information from getting lost from the log.
+            break;
+          }
+        }
+        return failure;
+      }
+
       @Override
       public void terminated(Service.State from) {
         if (from != Service.State.STOPPING) {