nestjs-graphql-typeorm-dataloader implements the middleware and the decorators needed to enable graphql data-loader in the entity or DTO level. So instead of defining the data loader in every resolve field, it is easier to just define it over a more generic field and let it handle which type of relations does it have to resolve and load the data properly. This also eliminates the setup for how to load the data, for which key to load the data for every field resolver.
- Registering a
nestjsinterceptororgraphqlpluginenables to set up the connection to give it a unique id to containerize the request better. This enables the data to be loaded for multiple resolve fields. - Entities decorated with the
nestjsgraphqlextensionsfield contains the metadata on how to solve this exact field. field-middlewarethat is initiated explicitly for thatfieldsolves the field fetching according to the direction of theextensionmetadata and solves the given relation by the database manager.
This plugin requires an interceptor to keep track of the request ids in a container environment, which could be done by nest itself but the second thing is you inject the connection of typeorm to field-middleware which nestjs does not enable to do. This will write a unique key to every request in the context directly. If you are going to use typeorm explicitly which this plugin is intended for you have to pass in the connection, since dependency injection is not available to fetch in the field-middleware.
You can choose between one of two options.
Import the interceptor and use it globally or module-scoped in your project.
import { getConnection } from 'typeorm'
import { DataLoaderInterceptor } from '@webundsoehne/nestjs-graphql-typeorm-dataloader'
@Module({
imports: [
{
provide: APP_INTERCEPTOR,
useFactory: (): DataLoaderInterceptor => new DataLoaderInterceptor({ typeormGetConnection: getConnection })
}
]
})
export class ServerModule {}Initially designed this plugin to use the apollo plugin method, then converted it to an interceptor. Even though they basically do the same thing the apollo-server-plugin is also there. This can be only global scoped since it is injected into the apollo-server itself.
import { getConnection } from 'typeorm'
import { ApolloServerDataLoaderPlugin } from '@webundsoehne/nestjs-graphql-typeorm-dataloader'
@Module({
imports: [
GraphQLModule.forRoot({
// ...
buildSchemaOptions: {
plugins: [new ApolloServerDataLoaderPlugin({ typeormGetConnection: getConnection })]
}
})
]
})
export class ServerModule {}Field middleware can either be injected to Field, FieldResolver, or globally.
Unfortunately nest.js does not allow to tamper with the GraphQL set up, so I could not overwrite the middleware field while you are decorating the field with extension so this stayed as a manual process.
This is due to graphql resolvers and field-resolvers inside the nest.js only registered once, therefore you can not lazily register metadata afterward, this behavior as I understand it can be seen in field decorator and field resolver decorator and following the behavior to add metadata for resolvers.
While you will see this making sense in the upcoming examples, it should just be done as follows.
@Field(() => [DocumentHistoryEntity], { nullable: true, middleware: [TypeormLoaderMiddleware] })To inject this middleware for each field, which will cause a little overhead but not much since it is pretty basic to check the context to have matching keys can be done as follows. But for more specific control over the fields, you can always use the injecting to a specific field approach.
import { getConnection } from 'typeorm'
import { ApolloServerDataLoaderPlugin, TypeormLoaderMiddleware } from '@webundsoehne/nestjs-graphql-typeorm-dataloader'
@Module({
providers: [
imports: [
GraphQLModule.forRoot({
// ...
buildSchemaOptions: {
fieldMiddleware: [TypeormLoaderMiddleware]
}
})
]
]
})
export class ServerModule {}Entities or DTOs should be decorated with directions on how to resolve a relation.
The only critical thing here is getting the relation ids of the relation. Since typeorm already exposes fetching relation ids, another field can be decorated with RelationId and since the parent document will be injected into the function in TypeormLoaderExtension as an argument, this relation ids will be resolved and typeorm metadata will indicate the relation type and it will use the appropriate dataLoader with the given field. You can omit this field's serialization by marking it without a field decorator.
You can either use these decorators in the entity, DTO, or resolver. But the intention is to keep this in the DTOs or entities to define resolving them generically. You can also further process the output result, the GraphQL way.
If you define the resolver and extensions at the entity or DTO level, you do not need to define any field resolvers for a given field, and it will be resolved automatically.
Please do not forget to set the middleware per field if you did not set it globally, else it won't work at all.
Imagine a relationship where every company has a corporation. So it is an incoming relationship from the other side where we inherit the foreign key.
import { TypeormLoaderExtension, TypeormLoaderMiddleware } from '@webundsoehne/nestjs-graphql-typeorm-dataloader'
// ...
@ObjectType()
@Entity('company')
export class CompanyEntity extends BaseEntityWithPrimary<CompanyEntity> {
// ...
@Field(() => UUID)
@Column({ name: 'corporation_id', type: 'uuid' })
@IsUUID()
corporationId: string
// relations-incoming
@Field(() => CorporationEntity, { middleware: [TypeormLoaderMiddleware] })
@ManyToOne(() => CorporationEntity, (corporation) => corporation.companies, {
onUpdate: 'CASCADE',
onDelete: 'RESTRICT'
})
@JoinColumn({ name: 'corporation_id' })
@TypeormLoaderExtension((company: CompanyEntity) => company.corporationId)
corporation: CorporationEntity
// ...
}If you want to directly utilize a key from the other side of the relationship, you can set the option as follows. So it is an outgoing relationship where the other side inherits the foreign key.
Imagine that the user can have multiple user-companies, where the other side has the companyId as a foreign key already.
This only works with oneToOne and oneToMany.
import { TypeormLoaderExtension, TypeormLoaderMiddleware } from '@webundsoehne/nestjs-graphql-typeorm-dataloader'
// ...
@ObjectType()
@Entity('company')
export class CompanyEntity extends BaseEntityWithPrimary<CompanyEntity> {
// ...
// relations-outgoing
@Field(() => [UserCompanyEntity], { nullable: true })
@OneToMany(() => UserCompanyEntity, (userCompany) => userCompany.company, {
nullable: true,
onDelete: 'CASCADE'
})
@TypeormLoaderExtension((userCompany: UserCompanyEntity) => userCompany.companyId, { selfKey: true })
userCompanies?: UserCompanyEntity[]
// ...
}If the other option does not work out, or you don't have the foreign key joined to the column, you can use the relation id directly.
import { TypeormLoaderExtension, TypeormLoaderMiddleware } from '@webundsoehne/nestjs-graphql-typeorm-dataloader'
import { RelationId } from 'typeorm'
// ...
@ObjectType()
@Entity('company')
export class CompanyEntity extends BaseEntityWithPrimary<CompanyEntity> {
// ...
@RelationId((company: CompanyEntity) => company.userCompanies)
userCompanyIds: string[]
// relations-outgoing
@Field(() => [UserCompanyEntity], { nullable: true })
@OneToMany(() => UserCompanyEntity, (userCompany) => userCompany.company, {
nullable: true,
onDelete: 'CASCADE'
})
@TypeormLoaderExtension((user: UserEntity) => user.userCompanyIds)
userCompanies?: UserCompanyEntity[]
// ...
}You can also define your own data loader, but this time it should be in the resolver itself.
import { CustomLoaderExtension, CustomLoaderMiddleware } from '@webundsoehne/nestjs-graphql-typeorm-dataloader'
@Resolver(() => UserEntity)
export class UserResolver {
@ResolveField('documents', () => UserEntity, {
nullable: true,
middleware: [ CustomLoaderMiddleware ]
})
@CustomLoaderExtension(async (ids, { context }) => {
const documents = await this.documentRepository.find({
where: { user: { id: In(ids) } }
})
const documentById = groupBy(documents, 'userId')
return ids.map((id) => documentById[id] ?? [])
})
public resolveDocuments(@Parent() user: UserEntity): (dataloader: DataLoader<number, Photo[]>) => DocumentEntity[] {
return (dataloader: DataLoader<string, DocumentEntity[]>) =>
dataloader.load(user.id)
}
}
}Since this will resolve value and use the next function to forward it, you can later process the data utilizing a field-resolver.
Based on type-graphql-dataloader.