floriday/floriday.js

79 lines
1.9 KiB
JavaScript
Raw Permalink Normal View History

2023-05-19 11:47:36 +00:00
import { checkCon, closeCon } from './models/sequelize.js';
import * as utils from './utils.js';
2023-04-06 08:56:52 +00:00
import chalk from 'chalk';
2023-06-15 09:29:19 +00:00
import yml from 'js-yaml';
import fs from 'fs';
const env = process.env;
2023-06-15 09:29:19 +00:00
const flModels = yml.load(fs.readFileSync('./models/models.yml', 'utf8'));
let cycle = 1, stopSchedule = false;
class Floriday {
2023-04-24 10:46:06 +00:00
async start() {
try {
2023-05-08 10:18:14 +00:00
await utils.checkConfig();
2023-06-15 10:50:58 +00:00
await utils.checkToken();
await utils.separator('Init');
2023-04-24 10:46:06 +00:00
} catch (err) {
utils.handler('critical', err);
2023-04-24 10:46:06 +00:00
}
};
2023-04-24 10:46:06 +00:00
async tryConn() {
2023-05-17 05:36:58 +00:00
while (true)
try {
await utils.handler('warning', new Error('Awaiting a response from the database...'));
2023-05-19 11:47:36 +00:00
await utils.sleep(env.DB_RECON_TIMEOUT);
await checkCon();
2023-05-17 05:36:58 +00:00
await this.schedule();
}
catch (err) {};
};
2023-04-06 08:56:52 +00:00
2023-04-24 10:46:06 +00:00
async schedule () {
try {
const intervalTime = JSON.parse(env.IS_PRODUCTION)
? env.MS_PRODUCTION_SCHEDULE
: env.MS_TEST_SCHEDULE;
while (!stopSchedule) {
2023-04-24 10:46:06 +00:00
try {
await this.trunk();
2023-05-09 09:59:21 +00:00
await new Promise(resolve => setTimeout(resolve, intervalTime));
} catch (err) {
2023-04-24 10:46:06 +00:00
await this.tryConn();
2023-05-09 09:59:21 +00:00
await new Promise(resolve => setTimeout(resolve, intervalTime));
2023-04-24 10:46:06 +00:00
}
2023-05-09 09:59:21 +00:00
}
2023-04-24 10:46:06 +00:00
} catch (err) {
throw err;
};
};
2023-04-06 08:56:52 +00:00
2023-04-24 10:46:06 +00:00
async trunk() {
try{
await utils.checkConnections();
for (let model of flModels) {
if (stopSchedule) return;
2023-06-12 12:42:33 +00:00
await utils.syncModel(model);
}
2023-05-22 05:54:45 +00:00
await utils.separator(`${cycle++} ${(cycle == 2 ? 'Cycle' : 'Cycles')}`);
2023-04-24 10:46:06 +00:00
} catch (err) {
2023-07-03 05:35:18 +00:00
if (includes('ER_SOCKET_UNEXPECTED_CLOSE','ECONNREFUSED').includes(err?.original?.code) || (err?.message).substring(0, 24) === 'SequelizeConnectionError')
2023-06-30 05:47:19 +00:00
throw err;
utils.handler('critical', err);
2023-04-24 10:46:06 +00:00
}
};
2023-04-24 10:46:06 +00:00
async stop() {
try {
stopSchedule = true;
await closeCon();
console.warn(chalk.dim('\nBye, come back soon 👋'))
} catch (err) {
utils.handler('critical', err);
}
};
};
export default Floriday;