feat: ✨ rebuild header in gateway app

zhumeisongsong · Dec 7, 2024 · 5c837c5 · 5c837c5
1 parent 7430114
commit 5c837c5
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/apps/gateway/src/app/app.module.ts b/apps/gateway/src/app/app.module.ts
@@ -1,4 +1,4 @@
-import { IntrospectAndCompose } from '@apollo/gateway';
+import { IntrospectAndCompose, RemoteGraphQLDataSource } from '@apollo/gateway';
 import { gatewayConfig, userAppConfig } from '@shared/config';
 import { ApolloGatewayDriver, ApolloGatewayDriverConfig } from '@nestjs/apollo';
 import { Module } from '@nestjs/common';
@@ -31,6 +31,20 @@ import { AppService } from './app.service';
                 },
               ],
             }),
+            buildService: ({ url = '' }) => {
+              return new RemoteGraphQLDataSource({
+                url,
+                willSendRequest({ request, context }) {
+                  // rebuild header
+                  if (Object.keys(context).length > 0 && request.http) {
+                    request.http.headers.set(
+                      'Authorization',
+                      context['req']['headers']['authorization'],
+                    );
+                  }
+                },
+              });
+            },
           },
         };
       },

diff --git a/apps/users/src/app/app.module.ts b/apps/users/src/app/app.module.ts
@@ -37,7 +37,9 @@ import { AppService } from './app.service';
       playground: process.env['NODE_ENV'] !== 'production',
       sortSchema: true,
       plugins: [ApolloServerPluginInlineTrace()],
-      context: ({ req }: { req: Request }) => ({ req }), // Make sure request is passed to the context
+      context: ({ req }: { req: Request }) => {
+        return { headers: req?.headers };
+      },
     }),
     UsersModule,
     AuthModule,