diff --git a/lib/sequence.js b/lib/sequence.js index 20f7922..e7741c8 100644 --- a/lib/sequence.js +++ b/lib/sequence.js @@ -1,6 +1,5 @@ const _ = require('lodash'); const async = require('async'); -const mongoose = require('mongoose'); const SequenceArchive = require('./sequence_archive'); const sequenceArchive = SequenceArchive.getSingleton(); @@ -9,11 +8,11 @@ let Sequence; const resolve = (path, obj) => path.split('.').reduce((prev, curr) => (prev ? prev[curr] : null), obj); -module.exports = function SequenceFactory(connection) { - if (arguments.length !== 1) { - throw new Error( - 'Please, pass mongoose while requiring mongoose-sequence: https://github.com/ramiel/mongoose-sequence#requiring', - ); +module.exports = function SequenceFactory(mongoose) { + const connection = mongoose.connection; + + if (!connection || !connection.modelNames) { + throw new Error('Please pass the mongoose module as argument'); } /**