Split out current owner for parity with Fiber

Author: sebmarkbage

packages/react-server/src/ReactFlightServer.js

@@ -87,11 +87,9 @@ import {
   getThenableStateAfterSuspending,
   resetHooksForRequest,
 } from './ReactFlightHooks';
-import {
-  DefaultAsyncDispatcher,
-  currentOwner,
-  setCurrentOwner,
-} from './flight/ReactFlightAsyncDispatcher';
+import {DefaultAsyncDispatcher} from './flight/ReactFlightAsyncDispatcher';
+
+import {currentOwner, setCurrentOwner} from './flight/ReactFlightCurrentOwner';
 
 import {
   getIteratorFn,

packages/react-server/src/flight/ReactFlightAsyncDispatcher.js

@@ -15,6 +15,8 @@ import {resolveRequest, getCache} from '../ReactFlightServer';
 
 import {disableStringRefs} from 'shared/ReactFeatureFlags';
 
+import {currentOwner} from './ReactFlightCurrentOwner';
+
 function resolveCache(): Map<Function, mixed> {
   const request = resolveRequest();
   if (request) {
@@ -36,8 +38,6 @@ export const DefaultAsyncDispatcher: AsyncDispatcher = ({
   },
 }: any);
 
-export let currentOwner: ReactComponentInfo | null = null;
-
 if (__DEV__) {
   DefaultAsyncDispatcher.getOwner = (): null | ReactComponentInfo => {
     return currentOwner;
@@ -48,7 +48,3 @@ if (__DEV__) {
     return null;
   };
 }
-
-export function setCurrentOwner(componentInfo: null | ReactComponentInfo) {
-  currentOwner = componentInfo;
-}

packages/react-server/src/flight/ReactFlightCurrentOwner.js

@@ -0,0 +1,16 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @flow
+ */
+
+import type {ReactComponentInfo} from 'shared/ReactTypes';
+
+export let currentOwner: ReactComponentInfo | null = null;
+
+export function setCurrentOwner(componentInfo: null | ReactComponentInfo) {
+  currentOwner = componentInfo;
+}