feat: use virtuals factory in schema factory

nestjs · May 22, 2022 · de9b3fb · de9b3fb
1 parent 1bbf2e6
commit de9b3fb
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/factories/schema.factory.ts b/lib/factories/schema.factory.ts
@@ -3,6 +3,7 @@ import * as mongoose from 'mongoose';
 import { SchemaDefinition, SchemaDefinitionType } from 'mongoose';
 import { TypeMetadataStorage } from '../storages/type-metadata.storage';
 import { DefinitionsFactory } from './definitions.factory';
+import { VirtualsFactory } from './virtuals.factory';
 
 export class SchemaFactory {
   // TODO: remove unused, deprecated type argument
@@ -13,9 +14,13 @@ export class SchemaFactory {
     const schemaDefinition = DefinitionsFactory.createForClass(target);
     const schemaMetadata =
       TypeMetadataStorage.getSchemaMetadataByTarget(target);
-    return new mongoose.Schema<TClass>(
+    const schema = new mongoose.Schema<TClass>(
       schemaDefinition as SchemaDefinition<SchemaDefinitionType<TClass>>,
       schemaMetadata && schemaMetadata.options,
     );
+
+    VirtualsFactory.createForClass(target, schema);
+
+    return schema;
   }
 }