Add Telemetry for OOM Errors, Fixes AB#3124149

changelog.txt

@@ -34,6 +34,7 @@ Version 19.0.0
 - [MINOR] Managed profile Android util method (#2561)
 - [PATCH] Make userHandle response field optional (#2560)
 - [MINOR] Nonce redirect changes (#2552)
+- [PATCH] Fix OOM error handling (#2567)
 
 Version 18.2.2
 ----------

common4j/src/main/com/microsoft/identity/common/java/controllers/ExceptionAdapter.java

@@ -41,6 +41,8 @@
 import com.microsoft.identity.common.java.exception.UserCancelException;
 import com.microsoft.identity.common.java.logging.Logger;
 import com.microsoft.identity.common.java.net.HttpResponse;
+import com.microsoft.identity.common.java.opentelemetry.AttributeName;
+import com.microsoft.identity.common.java.opentelemetry.SpanExtension;
 import com.microsoft.identity.common.java.providers.microsoft.MicrosoftAuthorizationErrorResponse;
 import com.microsoft.identity.common.java.providers.oauth2.AuthorizationErrorResponse;
 import com.microsoft.identity.common.java.providers.oauth2.AuthorizationResult;
@@ -66,6 +68,7 @@
 
 import edu.umd.cs.findbugs.annotations.Nullable;
 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
+import io.opentelemetry.api.trace.Span;
 import lombok.NonNull;
 
 public class ExceptionAdapter {
@@ -436,6 +439,18 @@ public static ClientException clientExceptionFromException(@NonNull final Throwa
 
 
         if (e instanceof OutOfMemoryError) {
+            // Log status code and record exception in span
+            final Span currentSpan = SpanExtension.current();
+
+            // Attach the stack trace, to debug in telemetry later
+            currentSpan.setAttribute(
+                    AttributeName.out_of_memory_exception_stacktrace.name(),
+                    StringUtil.getStacktraceAsStringFromElementArray(e.getStackTrace())
+            );
+
+            com.microsoft.identity.common.java.logging.Logger.warn(
+                    TAG, "Received an out of memory error, stacktrace attached to span.");
+
             return new ClientException(
                     ClientException.OUT_OF_MEMORY,
                     e.getMessage(),

common4j/src/main/com/microsoft/identity/common/java/opentelemetry/AttributeName.java

@@ -327,5 +327,10 @@ public enum AttributeName {
     /**
      * Indicates the exception in generating a keypair.
      */
-    keypair_gen_exception
+    keypair_gen_exception,
+
+    /**
+     * Records the stacktrace for an out-of-memory exception.
+     */
+    out_of_memory_exception_stacktrace
 }

common4j/src/main/com/microsoft/identity/common/java/util/StringUtil.java

@@ -463,4 +463,24 @@ public static void overwriteWithNull(final char[] chars) {
             chars[i] = '\0';
         }
     }
+
+    /**
+     * Converts an array of stack trace elements (one method invocation) into one concatenated string object.
+     *
+     * @param elements list of stacktrace elements
+     * @return concatenated string list
+     */
+    @NonNull
+    public static String getStacktraceAsStringFromElementArray(@NonNull final StackTraceElement[] elements) {
+        final StringBuilder builder = new StringBuilder();
+
+        builder.append(elements[0]);
+
+        for (int i = 1; i < elements.length; i++) {
+            builder.append("\n");
+            builder.append(elements[i]);
+        }
+
+        return builder.toString();
+    }
 }