Allow for multiple graphql endpoints in the same application with schema segregation

packages/apollo/tests/code-first-multigraph/app.module.ts

@@ -0,0 +1,8 @@
+import { Module } from '@nestjs/common';
+import { PostModule } from './post/post.module';
+import { UserModule } from './user/user.module';
+
+@Module({
+ imports: [PostModule, UserModule],
+})
+export class ApplicationModule {}

packages/apollo/tests/code-first-multigraph/gateway.module.ts

@@ -0,0 +1,18 @@
+import { Module } from '@nestjs/common';
+import { GqlModuleOptions, GraphQLModule } from '@nestjs/graphql';
+import { ApolloGatewayDriver, ApolloGatewayDriverConfig } from '../../lib';
+
+@Module({
+ imports: [
+ GraphQLModule.forRoot<ApolloGatewayDriverConfig & GqlModuleOptions>({
+ driver: ApolloGatewayDriver,
+ gateway: {
+ serviceList: [
+ { name: 'user', url: 'http://localhost:3000/user/graphql' },
+ { name: 'post', url: 'http://localhost:3000/post/graphql' },
+ ],
+ },
+ }),
+ ],
+})
+export class GatewayModule {}

packages/apollo/tests/code-first-multigraph/main.ts

@@ -0,0 +1,20 @@
+import { ValidationPipe } from '@nestjs/common';
+import { NestFactory } from '@nestjs/core';
+import { ApplicationModule } from './app.module';
+import { GatewayModule } from './gateway.module';
+
+/**
+ * This use case comes from the need for an application
+ * to be designed in a way which would allow future context
+ * separation from a deployment point of view.
+ */
+export async function bootstrap() {
+ const app = await NestFactory.create(ApplicationModule);
+ app.useGlobalPipes(new ValidationPipe());
+ await app.listen(3000);
+
+ const gateway = await NestFactory.create(GatewayModule);
+ await gateway.listen(3001);
+}
+
+bootstrap();

packages/apollo/tests/code-first-multigraph/post/external-user/user.resolver.ts

@@ -0,0 +1,14 @@
+import { Resolver, ResolveField, Parent } from '@nestjs/graphql';
+
+import { Post, Publication } from '../post.type';
+import { User } from './user.type';
+
+@Resolver(() => User)
+export class UserResolver {
+ @ResolveField('posts', () => [Post])
+ userPosts(@Parent() user: User): Post[] {
+ return [
+ { id: 10, authorId: user.id, status: Publication.DRAFT, title: 'test' },
+ ];
+ }
+}

packages/apollo/tests/code-first-multigraph/post/external-user/user.type.ts

@@ -0,0 +1,10 @@
+import { ObjectType, Directive, Field, ID } from '@nestjs/graphql';
+
+@ObjectType()
+@Directive('@extends')
+@Directive('@key(fields: "id")')
+export class User {
+ @Field(() => ID)
+ @Directive('@external')
+ id: number;
+}

packages/apollo/tests/code-first-multigraph/post/post.module.ts

@@ -0,0 +1,22 @@
+import { Module } from '@nestjs/common';
+import { GraphQLModule } from '@nestjs/graphql';
+
+import { ApolloDriverConfig } from '../../../lib';
+import { ApolloFederationDriver } from '../../../lib/drivers';
+
+import { UserResolver } from './external-user/user.resolver';
+import { PostResolver } from './post.resolver';
+
+@Module({
+ imports: [
+ GraphQLModule.forRoot<ApolloDriverConfig>({
+ driver: ApolloFederationDriver,
+ debug: false,
+ autoSchemaFile: true,
+ path: '/post/graphql',
+ include: [PostModule],
+ }),
+ ],
+ providers: [PostResolver, UserResolver],
+})
+export class PostModule {}

packages/apollo/tests/code-first-multigraph/post/post.resolver.ts

@@ -0,0 +1,48 @@
+import {
+ Resolver,
+ Query,
+ ResolveField,
+ Parent,
+ ID,
+ Args,
+} from '@nestjs/graphql';
+
+import { User } from './external-user/user.type';
+import { Post, Publication } from './post.type';
+
+@Resolver(() => Post)
+export class PostResolver {
+ store = new Set<Post>([
+ {
+ id: 1,
+ authorId: 1,
+ title: 'How to Jedi II',
+ status: Publication.PUBLISHED,
+ },
+ {
+ id: 2,
+ authorId: 2,
+ title: 'Why lightsabers are unreliable',
+ status: Publication.DRAFT,
+ },
+ ]);
+
+ @Query(() => [Post])
+ posts() {
+ return [...this.store];
+ }
+
+ @Query(() => [Post])
+ post(@Args('id', { type: () => ID }) id: number): Post | undefined {
+ for (const post of this.store) {
+ if (post.id.toString() === id.toString()) {
+ return post;
+ }
+ }
+ }
+
+ @ResolveField(() => User)
+ user(@Parent() post: Post) {
+ return { __typename: 'User', id: post.authorId };
+ }
+}

packages/apollo/tests/code-first-multigraph/post/post.type.ts

@@ -0,0 +1,30 @@
+import {
+ registerEnumType,
+ ObjectType,
+ Directive,
+ Field,
+ ID,
+} from '@nestjs/graphql';
+
+export enum Publication {
+ PUBLISHED,
+ DRAFT,
+}
+
+registerEnumType(Publication, { name: 'Publication' });
+
+@ObjectType()
+@Directive('@key(fields: "id")')
+export class Post {
+ @Field(() => ID)
+ id: number;
+
+ @Field()
+ title: string;
+
+ @Field(() => Publication)
+ status: Publication;
+
+ @Field(() => ID)
+ authorId: number;
+}

packages/apollo/tests/code-first-multigraph/user/user.module.ts

@@ -0,0 +1,21 @@
+import { Module } from '@nestjs/common';
+import { GraphQLModule } from '@nestjs/graphql';
+
+import { ApolloDriverConfig } from '../../../lib';
+import { ApolloFederationDriver } from '../../../lib/drivers';
+
+import { UserResolver } from './user.resolver';
+
+@Module({
+ imports: [
+ GraphQLModule.forRoot<ApolloDriverConfig>({
+ driver: ApolloFederationDriver,
+ debug: false,
+ autoSchemaFile: true,
+ path: '/user/graphql',
+ include: [UserModule],
+ }),
+ ],
+ providers: [UserResolver],
+})
+export class UserModule {}

packages/apollo/tests/code-first-multigraph/user/user.resolver.ts

@@ -0,0 +1,50 @@
+import {
+ Field,
+ ObjectType,
+ Resolver,
+ Query,
+ ID,
+ Directive,
+ Args,
+ ResolveReference,
+} from '@nestjs/graphql';
+
+@ObjectType()
+@Directive('@key(fields: "id")')
+class User {
+ @Field(() => ID)
+ id: number;
+
+ @Field()
+ name: string;
+}
+
+@Resolver(() => User)
+export class UserResolver {
+ private readonly store = new Set<User>([
+ { id: 1, name: 'Luke' },
+ { id: 2, name: 'Han' },
+ ]);
+
+ @Query(() => [User])
+ users(): User[] {
+ return [...this.store];
+ }
+
+ @Query(() => User)
+ user(@Args('id', { type: () => ID }) id: number): User | undefined {
+ for (const user of this.store) {
+ if (user.id.toString() === id.toString()) {
+ return user;
+ }
+ }
+ }
+
+ @ResolveReference()
+ resolveReference(reference: {
+ __typename: string;
+ id: number;
+ }): User | undefined {
+ return this.user(reference.id);
+ }
+}