Skip to content

Commit 72ce783

Browse files
Merge pull request #1225 from Dukezo/master
feat: add option to specify a discriminator value
2 parents 84ad659 + a338213 commit 72ce783

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

lib/interfaces/model-definition.interface.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ import { Schema } from 'mongoose';
33
export type DiscriminatorOptions = {
44
name: string;
55
schema: Schema;
6+
value?: string;
67
};
78

89
export type ModelDefinition = {

lib/mongoose.providers.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ export function createMongooseProviders(
1313
...(option.discriminators || []).map((d) => ({
1414
provide: getModelToken(d.name, connectionName),
1515
useFactory: (model: Model<Document>) =>
16-
model.discriminator(d.name, d.schema),
16+
model.discriminator(d.name, d.schema, d.value),
1717
inject: [getModelToken(option.name, connectionName)],
1818
})),
1919
{
@@ -56,7 +56,7 @@ export function createMongooseAsyncProviders(
5656
...(option.discriminators || []).map((d) => ({
5757
provide: getModelToken(d.name, connectionName),
5858
useFactory: (model: Model<Document>) =>
59-
model.discriminator(d.name, d.schema),
59+
model.discriminator(d.name, d.schema, d.value),
6060
inject: [getModelToken(option.name, connectionName)],
6161
})),
6262
];

0 commit comments

Comments
 (0)