@mvp-rockets/namma-generator
Version:
A generator to generate mvp-rockets projects
41 lines (33 loc) • 1.11 kB
JavaScript
const fs = require('fs');
const path = require('path');
require('pg').defaults.parseInt8 = true;
const Sequelize = require('sequelize');
const basename = path.basename(__filename);
const config = require('config/config');
if (config.app?.useGlobalTransaction) {
const cls = require('cls-hooked');
const namespace = cls.getNamespace(config.clsNameSpace);
Sequelize.useCLS(namespace);
}
const db = {};
const dbConfig = {
logging: process.env.APP_ENV === "dev",
...config.db
};
const sequelize = new Sequelize(dbConfig.database, dbConfig.username, dbConfig.password, dbConfig);
fs
.readdirSync(__dirname)
.filter((file) => (file.indexOf('.') !== 0) && (file !== basename) && (file.slice(-3) === '.js'))
.forEach((file) => {
const model = require(path.join(__dirname, file))(sequelize, Sequelize.DataTypes);
db[model.name] = model;
});
Object.keys(db).forEach((modelName) => {
if (db[modelName].associate) {
db[modelName].associate(db);
}
});
sequelize.options.logging = false;
db.sequelize = sequelize;
db.Sequelize = Sequelize;
module.exports = db;