Use Query Planner from Hive Router

.changeset/@graphql-mesh_fusion-runtime-1629-dependencies.md

@@ -0,0 +1,7 @@
+---
+'@graphql-mesh/fusion-runtime': patch
+---
+
+dependencies updates: 
+
+- Added dependency [`@graphql-hive/router@0.0.0-manual-f9542ee` ↗︎](https://www.npmjs.com/package/@graphql-hive/router/v/0.0.0) (to `dependencies`)

.changeset/sharp-cars-act.md

@@ -0,0 +1,5 @@
+---
+'@graphql-mesh/fusion-runtime': minor
+---
+
+Use Query Planner from Hive Router

.github/workflows/test.yml

@@ -34,6 +34,10 @@ jobs:
           node-version: ${{matrix.node-version}}
       - name: Test
         run: yarn test
+      - name: "[EXPERIMENTAL] Test with Hive Router Query Planner"
+        env:
+          __EXPERIMENTAL__HIVE_ROUTER_QUERY_PLANNER: 1
+        run: yarn test
 
   unit-bun:
     name: Unit / Bun
@@ -51,6 +55,14 @@ jobs:
           timeout_minutes: 10
           max_attempts: 3
           command: yarn test:bun
+      - name: "[EXPERIMENTAL] Test with Hive Router Query Planner"
+        uses: nick-fields/retry@ce71cc2ab81d554ebbe88c79ab5975992d79ba08 # v3
+        env:
+          __EXPERIMENTAL__HIVE_ROUTER_QUERY_PLANNER: 1
+        with:
+          timeout_minutes: 10
+          max_attempts: 3
+          command: yarn test:bun
 
   leaks:
     strategy:
@@ -76,6 +88,14 @@ jobs:
           timeout_minutes: 10
           max_attempts: 3
           command: yarn test:leaks
+      - name: "[EXPERIMENTAL] Run Tests with Hive Router Query Planner"
+        uses: nick-fields/retry@ce71cc2ab81d554ebbe88c79ab5975992d79ba08 # v3
+        env:
+          __EXPERIMENTAL__HIVE_ROUTER_QUERY_PLANNER: 1
+        with:
+          timeout_minutes: 10
+          max_attempts: 3
+          command: yarn test:leaks
 
   e2e:
     runs-on: ${{ matrix.setup.os }}

e2e/federation-example/federation-example.bench.ts

@@ -9,23 +9,31 @@ import { fetch } from '@whatwg-node/fetch';
 import { bench, describe, expect } from 'vitest';
 
 describe('Gateway', async () => {
-  const { gateway, service } = createTenv(__dirname);
+  const { gateway, service: _service } = createTenv(__dirname);
   const example = createExampleSetup(__dirname, 1000);
 
   const supergraph = await example.supergraph();
 
   const gateways: Record<string, Gateway | Service> = {
-    'Apollo Gateway': await service('apollo-gateway', {
+    // TODO: tests are failing for whatever reason
+    // 'Apollo Gateway': await service('apollo-gateway', {
+    //   env: {
+    //     SUPERGRAPH: supergraph,
+    //   },
+    // }),
+    'Hive Gateway w/ Tools': await gateway({
+      supergraph,
       env: {
-        SUPERGRAPH: supergraph,
+        FORK: 1,
+        NODE_ENV: 'production',
       },
     }),
-    'Hive Gateway': await gateway({
+    'Hive Gateway w/ Hive Router Query Planner': await gateway({
       supergraph,
-      args: ['--jit'],
       env: {
-        JIT: 'true',
+        FORK: 1,
         NODE_ENV: 'production',
+        __EXPERIMENTAL__HIVE_ROUTER_QUERY_PLANNER: 1,
       },
     }),
   };

internal/env/src/index.ts

@@ -141,3 +141,11 @@ export function getNodeVer(): {
   const [major, minor, patch] = process.versions.node.split('.').map(Number);
   return { major: major || NaN, minor: minor || NaN, patch: patch || NaN };
 }
+
+/**
+ * Whether the federation query planner is using Hive Router node-addon for query planning.
+ * @experimental
+ */
+export function usingHiveRouterQueryPlanner(): boolean {
+  return getEnvBool('__EXPERIMENTAL__HIVE_ROUTER_QUERY_PLANNER');
+}

packages/federation/tests/federation-compatibility.test.ts

@@ -12,6 +12,7 @@ import {
   MapperKind,
   mapSchema,
 } from '@graphql-tools/utils';
+import { usingHiveRouterQueryPlanner } from '~internal/env';
 import {
   buildSchema,
   getNamedType,
@@ -91,6 +92,8 @@ describe('Federation Compatibility', () => {
           batch: true,
         });
         gatewayRuntime = createGatewayRuntime({
+          logging: false,
+          maskedErrors: false,
           supergraph: supergraphSdl,
           plugins: () => [useCustomFetch(auditRouter.fetch)],
         });
@@ -150,46 +153,46 @@ describe('Federation Compatibility', () => {
         );
       });
       tests.forEach((_, i) => {
-        (supergraphName === 'requires-with-argument-conflict' ? it.todo : it)(
-          `test-query-${i}`,
-          async () => {
-            const test = tests[i];
-            if (!test) {
-              throw new Error(`Test ${i} not found`);
-            }
-            const response = await gatewayRuntime.fetch(
-              'http://localhost/graphql',
-              {
-                method: 'POST',
-                headers: {
-                  'content-type': 'application/json',
-                },
-                body: JSON.stringify({
-                  query: test.query,
-                }),
+        (!usingHiveRouterQueryPlanner() &&
+          supergraphName === 'requires-with-argument-conflict'
+          ? it.todo // fails in stitching
+          : it)(`test-query-${i}`, async () => {
+          const test = tests[i];
+          if (!test) {
+            throw new Error(`Test ${i} not found`);
+          }
+          const response = await gatewayRuntime.fetch(
+            'http://localhost/graphql',
+            {
+              method: 'POST',
+              headers: {
+                'content-type': 'application/json',
               },
-            );
-            const result: ExecutionResult = await response.json();
-            const received = {
-              data: result.data ?? null,
-              errors: !!result.errors?.length,
-            };
+              body: JSON.stringify({
+                query: test.query,
+              }),
+            },
+          );
+          const result: ExecutionResult = await response.json();
+          const received = {
+            data: result.data ?? null,
+            errors: !!result.errors?.length,
+          };
 
-            const expected = {
-              data: test.expected.data ?? null,
-              errors: test.expected.errors ?? false,
-            };
+          const expected = {
+            data: test.expected.data ?? null,
+            errors: test.expected.errors ?? false,
+          };
 
-            try {
-              expect(received).toEqual(expected);
-            } catch (e) {
-              result.errors?.forEach((err) => {
-                console.error(err);
-              });
-              throw e;
-            }
-          },
-        );
+          try {
+            expect(received).toEqual(expected);
+          } catch (e) {
+            result.errors?.forEach((err) => {
+              console.error(err);
+            });
+            throw e;
+          }
+        });
       });
     });
   }

packages/fusion-runtime/package.json

@@ -46,6 +46,7 @@
     "@envelop/core": "^5.3.2",
     "@envelop/instrumentation": "^1.0.0",
     "@graphql-hive/logger": "workspace:^",
+    "@graphql-hive/router": "0.0.0-manual-f9542ee",
     "@graphql-mesh/cross-helpers": "^0.4.10",
     "@graphql-mesh/transport-common": "workspace:^",
     "@graphql-mesh/types": "^0.104.13",