|
1 | 1 | 'use strict';
|
2 |
| -var fs = require('fs'); |
3 |
| -var path = require('path'); |
4 |
| -var Sequelize = require('sequelize'); |
5 |
| -var basename = path.basename(module.filename); |
6 |
| -var env = process.env.NODE_ENV || 'development'; |
7 |
| -var config = require('../config/config.json')[env]; |
8 |
| -var db = {}; |
9 |
| -if (config.use_env_variable) { |
10 |
| - var sequelize = new Sequelize(process.env[config.use_env_variable]); |
11 |
| -} |
12 |
| -else { |
13 |
| - var sequelize = new Sequelize(config.database, config.username, config.password, config); |
14 |
| -} |
| 2 | + let fs = require('fs'); |
| 3 | + let path = require('path'); |
| 4 | + let Sequelize = require('sequelize'); |
| 5 | + let basename = path.basename(module.filename); |
| 6 | + const config = require('../config/env/config')(); |
| 7 | + let env = config.env || 'development'; |
| 8 | + let db = {}; |
| 9 | + let sequelize; |
| 10 | + |
| 11 | + if (config.db_url) { |
| 12 | + sequelize = new Sequelize(config.db_url); |
| 13 | + } else { |
| 14 | + sequelize = new Sequelize(config.db, config.username, config.password); |
| 15 | + } |
15 | 16 | fs
|
16 |
| - .readdirSync(__dirname) |
17 |
| - .filter(function (file) { |
18 |
| - return (file.indexOf('.') !== 0) && (file !== basename) && (file.slice(-3) === '.js'); |
19 |
| -}) |
20 |
| - .forEach(function (file) { |
21 |
| - var model = sequelize['import'](path.join(__dirname, file)); |
22 |
| - db[model.name] = model; |
23 |
| -}); |
| 17 | + .readdirSync(__dirname) |
| 18 | + .filter((file) => { |
| 19 | + return (file.indexOf('.') !== 0) && (file !== basename) && (file.slice(-3) === '.js'); |
| 20 | + }) |
| 21 | + .forEach((file) => { |
| 22 | + var model = sequelize['import'](path.join(__dirname, file)); |
| 23 | + db[model.name] = model; |
| 24 | + }); |
24 | 25 | Object.keys(db).forEach(function (modelName) {
|
25 |
| - if (db[modelName].associate) { |
26 |
| - db[modelName].associate(db); |
27 |
| - } |
28 |
| -}); |
29 |
| -db.sequelize = sequelize; |
30 |
| -db.Sequelize = Sequelize; |
31 |
| -module.exports = db; |
| 26 | + if (db[modelName].associate) { |
| 27 | + db[modelName].associate(db); |
| 28 | + } |
| 29 | + }); |
| 30 | + db.sequelize = sequelize; |
| 31 | + db.Sequelize = Sequelize; |
| 32 | + module.exports = db; |
0 commit comments