Merge pull request #1225 from Dukezo/master

feat: add option to specify a discriminator value
nestjs · May 20, 2022 · 72ce783 · 72ce783
2 parents 84ad659 + a338213
commit 72ce783
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/lib/interfaces/model-definition.interface.ts b/lib/interfaces/model-definition.interface.ts
@@ -3,6 +3,7 @@ import { Schema } from 'mongoose';
 export type DiscriminatorOptions = {
   name: string;
   schema: Schema;
+  value?: string;
 };
 
 export type ModelDefinition = {

diff --git a/lib/mongoose.providers.ts b/lib/mongoose.providers.ts
@@ -13,7 +13,7 @@ export function createMongooseProviders(
       ...(option.discriminators || []).map((d) => ({
         provide: getModelToken(d.name, connectionName),
         useFactory: (model: Model<Document>) =>
-          model.discriminator(d.name, d.schema),
+          model.discriminator(d.name, d.schema, d.value),
         inject: [getModelToken(option.name, connectionName)],
       })),
       {
@@ -56,7 +56,7 @@ export function createMongooseAsyncProviders(
       ...(option.discriminators || []).map((d) => ({
         provide: getModelToken(d.name, connectionName),
         useFactory: (model: Model<Document>) =>
-          model.discriminator(d.name, d.schema),
+          model.discriminator(d.name, d.schema, d.value),
         inject: [getModelToken(option.name, connectionName)],
       })),
     ];