const async = require("async"); class DeferredPromise { constructor() { this.promise = new Promise((resolve, reject) => { this.reject = reject; this.resolve = resolve; }); } } class MovingAverageCalculator { constructor() { this.count = 0; this._mean = 0; } update(newValue) { this.count++; const differential = (newValue - this._mean) / this.count; this._mean += differential; } get mean() { this.validate(); return this._mean; } validate() { if (this.count === 0) throw new Error("Mean is undefined"); } } class CoreClass { constructor(name) { this.name = name; this.status = "UNINITIALIZED"; // this.log("Core constructor"); this.jobQueue = async.priorityQueue( (job, callback) => this._runJob(job, callback), 10 // How many jobs can run concurrently ); this.jobQueue.pause(); this.runningJobs = []; this.priorities = {}; this.stage = 0; this.jobStatistics = {}; this.registerJobs(); } setStatus(status) { this.status = status; this.log("INFO", `Status changed to: ${status}`); if (this.status === "READY") this.jobQueue.resume(); else if (this.status === "FAIL" || this.status === "LOCKDOWN") this.jobQueue.pause(); } getStatus() { return this.status; } setStage(stage) { this.stage = stage; } getStage() { return this.stage; } _initialize() { this.setStatus("INITIALIZING"); this.initialize() .then(() => { this.setStatus("READY"); this.moduleManager.onInitialize(this); }) .catch((err) => { console.error(err); this.setStatus("FAILED"); this.moduleManager.onFail(this); }); } log() { let _arguments = Array.from(arguments); const type = _arguments[0]; _arguments.splice(0, 1); const start = `|${this.name.toUpperCase()}|`; const numberOfTabsNeeded = 4 - Math.ceil(start.length / 8); _arguments.unshift(`${start}${Array(numberOfTabsNeeded).join("\t")}`); if (type === "INFO") { _arguments[0] = _arguments[0] + "\x1b[36m"; _arguments.push("\x1b[0m"); console.log.apply(null, _arguments); } else if (type === "ERROR") { _arguments[0] = _arguments[0] + "\x1b[31m"; _arguments.push("\x1b[0m"); console.error.apply(null, _arguments); } } registerJobs() { let props = []; let obj = this; do { props = props.concat(Object.getOwnPropertyNames(obj)); } while ((obj = Object.getPrototypeOf(obj))); const jobNames = props .sort() .filter( (prop) => typeof this[prop] == "function" && prop === prop.toUpperCase() ); jobNames.forEach((jobName) => { this.jobStatistics[jobName] = { successful: 0, failed: 0, total: 0, averageTiming: new MovingAverageCalculator(), }; }); } runJob(name, payload, options = {}) { let deferredPromise = new DeferredPromise(); const job = { name, payload, onFinish: deferredPromise }; if (options.bypassQueue) { this._runJob(job, () => {}); } else { const priority = this.priorities[name] ? this.priorities[name] : 10; this.jobQueue.push(job, priority); } return deferredPromise.promise; } setModuleManager(moduleManager) { this.moduleManager = moduleManager; } _runJob(job, cb) { this.log("INFO", `Running job ${job.name}`); const startTime = Date.now(); this.runningJobs.push(job); const newThis = Object.assign( Object.create(Object.getPrototypeOf(this)), this ); newThis.runJob = (...args) => { if (args.length === 2) args.push({}); args[2].bypassQueue = true; return this.runJob.apply(this, args); }; this[job.name] .apply(newThis, [job.payload]) .then((response) => { this.log("INFO", `Ran job ${job.name} successfully`); this.jobStatistics[job.name].successful++; job.onFinish.resolve(response); }) .catch((error) => { this.log("INFO", `Running job ${job.name} failed`); this.jobStatistics[job.name].failed++; job.onFinish.reject(error); }) .finally(() => { const endTime = Date.now(); const executionTime = endTime - startTime; this.jobStatistics[job.name].total++; this.jobStatistics[job.name].averageTiming.update( executionTime ); this.runningJobs.splice(this.runningJobs.indexOf(job), 1); cb(); }); } } module.exports = CoreClass;