feat(tmongodbplugin): add schema registration

Itee · Aug 13, 2019 · cc1b772 · cc1b772
1 parent 0511264
commit cc1b772
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/sources/mongodb/TMongoDBPlugin.js b/sources/mongodb/TMongoDBPlugin.js
@@ -27,6 +27,10 @@ class TMongoDBPlugin extends TAbstractDatabasePlugin {
         this._schemas = value
     }
 
+    addSchema( value ) {
+        this._schemas.push(value)
+    }
+
     get types () {
         return this._types
     }
@@ -144,10 +148,13 @@ class TMongoDBPlugin extends TAbstractDatabasePlugin {
     }
 
     beforeRegisterRoutes ( Mongoose ) {
+
         super.beforeRegisterRoutes( Mongoose )
 
         this._registerTypes( Mongoose )
         TMongoDBPlugin._registerTypesTo( Mongoose, this.__dirname )
+
+        this._registerSchemas( Mongoose )
         TMongoDBPlugin._registerSchemasTo( Mongoose, this.__dirname )
 
     }
@@ -163,6 +170,17 @@ class TMongoDBPlugin extends TAbstractDatabasePlugin {
 
     }
 
+    _registerSchemas ( Mongoose ) {
+
+        for ( let schema of this._schemas ) {
+
+            console.log( `Register schema: ${schema.name}` )
+            schema( Mongoose )
+
+        }
+
+    }
+
 }
 
 export { TMongoDBPlugin }