test(apollo-federation): added a multigraph use-case

Author: Elies Lou

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,15 @@
+import { ValidationPipe } from '@nestjs/common';
+import { NestFactory } from '@nestjs/core';
+import { ApplicationModule } from './app.module';
+import { GatewayModule } from './gateway.module';
+
+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);
+  }
+}