Track owner on AsyncSequence nodes

Author: sebmarkbage

packages/react-server/src/ReactFlightAsyncSequence.js

@@ -7,12 +7,15 @@
  * @flow
  */
 
+import type {ReactComponentInfo} from 'shared/ReactTypes';
+
 export const IO_NODE = 0;
 export const PROMISE_NODE = 1;
 export const AWAIT_NODE = 2;
 
 export type IONode = {
   tag: 0,
+  owner: null | ReactComponentInfo,
   stack: Error, // callsite that spawned the I/O
   start: number, // start time when the first part of the I/O sequence started
   end: number, // we typically don't use this. only when there's no promise intermediate.
@@ -22,6 +25,7 @@ export type IONode = {
 
 export type PromiseNode = {
   tag: 1,
+  owner: null | ReactComponentInfo,
   stack: Error, // callsite that created the Promise
   start: number, // start time when the Promise was created
   end: number, // end time when the Promise was resolved.
@@ -31,6 +35,7 @@ export type PromiseNode = {
 
 export type AwaitNode = {
   tag: 2,
+  owner: null | ReactComponentInfo,
   stack: Error, // callsite that awaited (using await, .then(), Promise.all(), ...)
   start: -1.1, // not used. We use the timing of the awaited promise.
   end: -1.1, // not used.

packages/react-server/src/ReactFlightServer.js

@@ -3627,7 +3627,7 @@ function serializeIONode(
       name = name.slice(7);
     }
   }
-  const owner = null; // TODO
+  const owner = ioNode.owner;
   // Ensure the owner is already outlined.
   if (owner != null) {
     outlineComponentInfo(request, owner);

packages/react-server/src/ReactFlightServerConfigDebugNode.js

@@ -15,6 +15,7 @@ import type {
 } from './ReactFlightAsyncSequence';
 
 import {IO_NODE, PROMISE_NODE, AWAIT_NODE} from './ReactFlightAsyncSequence';
+import {resolveOwner} from './flight/ReactFlightCurrentOwner';
 import {createHook, executionAsyncId} from 'async_hooks';
 import {enableAsyncDebugInfo} from 'shared/ReactFeatureFlags';
 
@@ -46,6 +47,7 @@ export function initAsyncDebugInfo(): void {
             // so that we can later pick the best stack trace in user space.
             node = ({
               tag: AWAIT_NODE,
+              owner: resolveOwner(),
               stack: new Error(),
               start: -1.1,
               end: -1.1,
@@ -55,6 +57,7 @@ export function initAsyncDebugInfo(): void {
           } else {
             node = ({
               tag: PROMISE_NODE,
+              owner: resolveOwner(),
               stack: new Error(),
               start: performance.now(),
               end: -1.1, // Set when we resolve.
@@ -74,6 +77,7 @@ export function initAsyncDebugInfo(): void {
             // We have begun a new I/O sequence.
             node = ({
               tag: IO_NODE,
+              owner: resolveOwner(),
               stack: new Error(), // This is only used if no native promises are used.
               start: performance.now(),
               end: -1.1, // Only set when pinged.
@@ -84,6 +88,7 @@ export function initAsyncDebugInfo(): void {
             // We have begun a new I/O sequence after the await.
             node = ({
               tag: IO_NODE,
+              owner: resolveOwner(),
               stack: new Error(),
               start: performance.now(),
               end: -1.1, // Only set when pinged.