Use in console.timeStamp (#54339)

Summary:

# Changelog: [Internal]

Whenever the `console.timeStamp()` is called and we are actually tracing, we will peek at the `ConsoleTaskOrchestrator` stack and check if there is a task.

If so, it means that this call happened inside a task context, so we can request a stack trace and propagate it down to `PerformanceTracer`.

Reviewed By: huntie

Differential Revision: D85481866

Author: hoxyq

packages/react-native/ReactCommon/jsinspector-modern/RuntimeTargetConsole.cpp

@@ -7,6 +7,7 @@
 
 #include "ConsoleTask.h"
 #include "ConsoleTaskContext.h"
+#include "ConsoleTaskOrchestrator.h"
 
 #include <jsinspector-modern/RuntimeTarget.h>
 #include <jsinspector-modern/tracing/PerformanceTracer.h>
@@ -480,8 +481,17 @@ void consoleTimeStamp(
   }
 
   if (performanceTracer.isTracing()) {
+    auto taskContext = ConsoleTaskOrchestrator::getInstance().top();
+
     performanceTracer.reportTimeStamp(
-        label, start, end, trackName, trackGroup, color, std::move(detail));
+        label,
+        start,
+        end,
+        trackName,
+        trackGroup,
+        color,
+        std::move(detail),
+        taskContext ? taskContext->getSerializedStackTraceProvider() : nullptr);
   }
 
   if (ReactPerfettoLogger::isTracing()) {

packages/react-native/ReactCommon/jsinspector-modern/tracing/PerformanceTracer.cpp

@@ -214,7 +214,8 @@ void PerformanceTracer::reportTimeStamp(
     std::optional<std::string> trackName,
     std::optional<std::string> trackGroup,
     std::optional<ConsoleTimeStampColor> color,
-    std::optional<folly::dynamic> detail) {
+    std::optional<folly::dynamic> detail,
+    std::function<std::optional<folly::dynamic>()>&& stackTraceProvider) {
   if (!tracingAtomic_) {
     return;
   }
@@ -233,6 +234,7 @@ void PerformanceTracer::reportTimeStamp(
           .trackGroup = std::move(trackGroup),
           .color = std::move(color),
           .detail = std::move(detail),
+          .stackTraceProvider = std::move(stackTraceProvider),
           .threadId = getCurrentThreadId(),
       });
 }
@@ -695,6 +697,11 @@ void PerformanceTracer::enqueueTraceEventsFromPerformanceTracerEvent(
               }
               data["devtools"] = folly::toJson(devtoolsDetail);
             }
+            if (event.stackTraceProvider) {
+              if (auto maybeStackTrace = event.stackTraceProvider()) {
+                data["rnStackTrace"] = std::move(*maybeStackTrace);
+              }
+            }
 
             events.emplace_back(
                 TraceEvent{

packages/react-native/ReactCommon/jsinspector-modern/tracing/PerformanceTracer.h

@@ -97,7 +97,8 @@ class PerformanceTracer {
       std::optional<std::string> trackName = std::nullopt,
       std::optional<std::string> trackGroup = std::nullopt,
       std::optional<ConsoleTimeStampColor> color = std::nullopt,
-      std::optional<folly::dynamic> detail = std::nullopt);
+      std::optional<folly::dynamic> detail = std::nullopt,
+      std::function<std::optional<folly::dynamic>()> &&stackTraceProvider = nullptr);
 
   /**
    * Record an Event Loop tick, which will be represented as an Event Loop task
@@ -267,6 +268,7 @@ class PerformanceTracer {
     std::optional<std::string> trackGroup;
     std::optional<ConsoleTimeStampColor> color;
     std::optional<folly::dynamic> detail;
+    std::function<std::optional<folly::dynamic>()> stackTraceProvider;
     ThreadId threadId;
     HighResTimeStamp createdAt = HighResTimeStamp::now();
   };