feat(gatsby): PQR workers can access inference metadata (#31858)

* test(gatsby): check if worker can access node created in different process

* make timeout longer for workerpool tests

* test fixes from other pr

* wip: saveStateForWorkers / loadStateInWorker

* basic functionality + type fixes

* wip (failing tests)

* revert child

* add redux state func

* revert early return changes

* initial

* update tests and add env var

* updates

* send inference metadata

* ignore all node_modules

* rename GatsbyStateSlices type to GatsbyStateKeys to more accurately describe it

* actually fix @babel/register

* Restore pool.ts

Co-authored-by: Michal Piechowiak <misiek.piechowiak@gmail.com>
Co-authored-by: gatsbybot <mathews.kyle+gatsbybot@gmail.com>

packages/gatsby/src/bootstrap/index.ts

@@ -61,6 +61,10 @@ export async function bootstrap(
 
   await rebuildSchemaWithSitePage(context)
 
+  if (process.env.GATSBY_EXPERIMENTAL_PARALLEL_QUERY_RUNNING) {
+    saveStateForWorkers([`inferenceMetadata`])
+  }
+
   await extractQueries(context)
 
   if (process.env.GATSBY_EXPERIMENTAL_PARALLEL_QUERY_RUNNING) {

packages/gatsby/src/utils/worker/__tests__/share-state.ts

@@ -168,7 +168,7 @@ describe(`worker (share-state)`, () => {
     `)
   })
 
-  it(`can set slices results into state and access it`, async () => {
+  it(`can set slices results into state and access page & static queries`, async () => {
     worker = createTestWorker()
     const staticQueryID = `1`
 
@@ -234,4 +234,47 @@ describe(`worker (share-state)`, () => {
       }
     `)
   })
+
+  it(`can set slices results into state and access inference metadata`, async () => {
+    worker = createTestWorker()
+
+    store.dispatch({
+      type: `BUILD_TYPE_METADATA`,
+      payload: {
+        typeName: `Test`,
+        nodes: [
+          {
+            id: `1`,
+            parent: null,
+            children: [],
+            foo: `bar`,
+            internal: { type: `Test` },
+          },
+        ],
+      },
+    })
+
+    saveStateForWorkers([`inferenceMetadata`])
+
+    await worker.setInferenceMetadata()
+
+    const inf = await worker.getInferenceMetadata(`Test`)
+
+    expect(inf).toMatchInlineSnapshot(`
+      Object {
+        "dirty": true,
+        "fieldMap": Object {
+          "foo": Object {
+            "string": Object {
+              "example": "bar",
+              "first": "1",
+              "total": 1,
+            },
+          },
+        },
+        "ignoredFields": Object {},
+        "total": 1,
+      }
+    `)
+  })
 })

packages/gatsby/src/utils/worker/__tests__/test-helpers/child-for-tests.ts

@@ -5,6 +5,7 @@ import {
   IGatsbyPageComponent,
   IGatsbyStaticQueryComponents,
 } from "../../../../redux/types"
+import { ITypeMetadata } from "../../../../schema/infer/inference-metadata"
 import reporter from "gatsby-cli/lib/reporter"
 import apiRunner from "../../../api-runner-node"
 
@@ -33,6 +34,9 @@ export function getStaticQueryComponent(
 ): IGatsbyStaticQueryComponents | undefined {
   return store.getState().staticQueryComponents.get(id)
 }
+export function getInferenceMetadata(typeName: string): ITypeMetadata {
+  return store.getState().inferenceMetadata.typeMap[typeName]
+}
 
 // test: reporter
 export function log(message: string): boolean {

packages/gatsby/src/utils/worker/child/index.ts

@@ -1,4 +1,4 @@
 // Note: this doesn't check for conflicts between module exports
 export { renderHTMLProd, renderHTMLDev } from "./render-html"
-export { setQueries } from "./schema"
+export { setQueries, setInferenceMetadata } from "./schema"
 export { loadConfigAndPlugins } from "./load-config-and-plugins"

packages/gatsby/src/utils/worker/child/schema.ts

@@ -1,5 +1,8 @@
 import { setState } from "./state"
 
+export function setInferenceMetadata(): void {
+  setState([`inferenceMetadata`])
+}
 export function setQueries(): void {
   setState([`components`, `staticQueryComponents`])
 }