forked from meanjs/mean
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(config): Mongo Seed 2.0 (meanjs#1808)
feat(config): Mongo Seed 2.0 Adds a more configurable and easily extended MongoDB Seed feature. Adds additional options at the collection, and collection item level to allow more control over how each environment config handles the seeding feature. Enforces seed order based on the order of the environment's seeding configuration object. Removes the previous SeedDB config file. Adds chalk to messages logged to the console for readability. Refactors the Mongo Seed configuration tests. Adds Gulp tasks to perform Mongo Seed operations for default, prod, and test environment configurations. Also, adds accommodating npm scripts.
- Loading branch information
Showing
11 changed files
with
1,083 additions
and
497 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,153 @@ | ||
| 'use strict'; | ||
|
|
||
| var _ = require('lodash'), | ||
| config = require('../config'), | ||
| mongoose = require('mongoose'), | ||
| chalk = require('chalk'); | ||
|
|
||
| exports.start = start; | ||
|
|
||
| function start(seedConfig) { | ||
| return new Promise(function (resolve, reject) { | ||
| seedConfig = seedConfig || {}; | ||
|
|
||
| var options = seedConfig.options || (config.seedDB ? _.clone(config.seedDB.options, true) : {}); | ||
| var collections = seedConfig.collections || (config.seedDB ? _.clone(config.seedDB.collections, true) : []); | ||
|
|
||
| if (!collections.length) { | ||
| return resolve(); | ||
| } | ||
|
|
||
| var seeds = collections | ||
| .filter(function (collection) { | ||
| return collection.model; | ||
| }); | ||
|
|
||
| // Use the reduction pattern to ensure we process seeding in desired order. | ||
| seeds.reduce(function (p, item) { | ||
| return p.then(function () { | ||
| return seed(item, options); | ||
| }); | ||
| }, Promise.resolve()) // start with resolved promise for initial previous (p) item | ||
| .then(onSuccessComplete) | ||
| .catch(onError); | ||
|
|
||
| // Local Promise handlers | ||
|
|
||
| function onSuccessComplete() { | ||
| if (options.logResults) { | ||
| console.log(); | ||
| console.log(chalk.bold.green('Database Seeding: Mongo Seed complete!')); | ||
| console.log(); | ||
| } | ||
|
|
||
| return resolve(); | ||
| } | ||
|
|
||
| function onError(err) { | ||
| if (options.logResults) { | ||
| console.log(); | ||
| console.log(chalk.bold.red('Database Seeding: Mongo Seed Failed!')); | ||
| console.log(chalk.bold.red('Database Seeding: ' + err)); | ||
| console.log(); | ||
| } | ||
|
|
||
| return reject(err); | ||
| } | ||
|
|
||
| }); | ||
| } | ||
|
|
||
| function seed(collection, options) { | ||
| // Merge options with collection options | ||
| options = _.merge(options || {}, collection.options || {}); | ||
|
|
||
| return new Promise(function (resolve, reject) { | ||
| const Model = mongoose.model(collection.model); | ||
| const docs = collection.docs; | ||
|
|
||
| var skipWhen = collection.skip ? collection.skip.when : null; | ||
|
|
||
| if (!Model.seed) { | ||
| return reject(new Error('Database Seeding: Invalid Model Configuration - ' + collection.model + '.seed() not implemented')); | ||
| } | ||
|
|
||
| if (!docs || !docs.length) { | ||
| return resolve(); | ||
| } | ||
|
|
||
| // First check if we should skip this collection | ||
| // based on the collection's "skip.when" option. | ||
| // NOTE: If it exists, "skip.when" should be a qualified | ||
| // Mongoose query that will be used with Model.find(). | ||
| skipCollection() | ||
| .then(seedDocuments) | ||
| .then(function () { | ||
| return resolve(); | ||
| }) | ||
| .catch(function (err) { | ||
| return reject(err); | ||
| }); | ||
|
|
||
| function skipCollection() { | ||
| return new Promise(function (resolve, reject) { | ||
| if (!skipWhen) { | ||
| return resolve(false); | ||
| } | ||
|
|
||
| Model | ||
| .find(skipWhen) | ||
| .exec(function (err, results) { | ||
| if (err) { | ||
| return reject(err); | ||
| } | ||
|
|
||
| if (results && results.length) { | ||
| return resolve(true); | ||
| } | ||
|
|
||
| return resolve(false); | ||
| }); | ||
| }); | ||
| } | ||
|
|
||
| function seedDocuments(skipCollection) { | ||
| return new Promise(function (resolve, reject) { | ||
|
|
||
| if (skipCollection) { | ||
| return onComplete([{ message: chalk.yellow('Database Seeding: ' + collection.model + ' collection skipped') }]); | ||
| } | ||
|
|
||
| var workload = docs | ||
| .filter(function (doc) { | ||
| return doc.data; | ||
| }) | ||
| .map(function (doc) { | ||
| return Model.seed(doc.data, { overwrite: doc.overwrite }); | ||
| }); | ||
|
|
||
| Promise.all(workload) | ||
| .then(onComplete) | ||
| .catch(onError); | ||
|
|
||
| // Local Closures | ||
|
|
||
| function onComplete(responses) { | ||
| if (options.logResults) { | ||
| responses.forEach(function (response) { | ||
| if (response.message) { | ||
| console.log(chalk.magenta(response.message)); | ||
| } | ||
| }); | ||
| } | ||
|
|
||
| return resolve(); | ||
| } | ||
|
|
||
| function onError(err) { | ||
| return reject(err); | ||
| } | ||
| }); | ||
| } | ||
| }); | ||
| } |
Oops, something went wrong.