spee.ch/server/models/index.js

88 lines
2.5 KiB
JavaScript
Raw Normal View History

const Sequelize = require('sequelize');
const logger = require('winston');
2018-04-18 20:07:54 +02:00
const Certificate = require('./certificate.js');
const Channel = require('./channel.js');
const Claim = require('./claim.js');
const File = require('./file.js');
const User = require('./user.js');
2018-04-28 00:17:09 +02:00
const Blocked = require('./blocked.js');
2018-06-28 00:11:25 +02:00
const Tor = require('./tor.js');
2018-06-06 04:34:24 +02:00
const {database, username, password} = require('@config/mysqlConfig');
if (!database || !username || !password) {
logger.warn('missing database, user, or password from mysqlConfig');
}
2018-03-29 21:21:23 +02:00
// set sequelize options
const sequelize = new Sequelize(database, username, password, {
host : 'localhost',
dialect : 'mysql',
2018-05-15 00:23:50 +02:00
dialectOptions: {
decimalNumbers: true
},
2018-03-29 21:21:23 +02:00
logging : false,
pool : {
max : 5,
min : 0,
idle : 10000,
acquire: 10000,
},
2018-05-15 00:23:50 +02:00
operatorsAliases: false,
2018-03-29 21:21:23 +02:00
});
// establish mysql connection
sequelize
.authenticate()
.then(() => {
logger.info('Sequelize has established mysql connection successfully.');
})
.catch(err => {
logger.error('Sequelize was unable to connect to the database:', err);
});
// manually add each model to the db object (note: make this dynamic)
const db = {};
db['Certificate'] = sequelize.import('Certificate', Certificate);
db['Channel'] = sequelize.import('Channel', Channel);
db['Claim'] = sequelize.import('Claim', Claim);
db['File'] = sequelize.import('File', File);
db['User'] = sequelize.import('User', User);
2018-04-28 00:17:09 +02:00
db['Blocked'] = sequelize.import('Blocked', Blocked);
2018-06-28 00:11:25 +02:00
db['Tor'] = sequelize.import('Tor', Tor);
2018-03-29 21:21:23 +02:00
// run model.association for each model in the db object that has an association
logger.info('associating db models...');
Object.keys(db).forEach(modelName => {
if (db[modelName].associate) {
logger.info('Associating model:', modelName);
db[modelName].associate(db);
}
});
// add sequelize/Sequelize to db
db.sequelize = sequelize;
db.Sequelize = Sequelize;
// add an 'upsert' method to the db object
db.upsert = (Model, values, condition, tableName) => {
return Model
.findOne({
where: condition,
})
.then(obj => {
if (obj) { // update
logger.debug(`updating record in db.${tableName}`);
return obj.update(values);
} else { // insert
logger.debug(`creating record in db.${tableName}`);
return Model.create(values);
}
})
.catch(function (error) {
logger.error(`${tableName}.upsert error`, error);
throw error;
});
};
module.exports = db;