158 lines
3.8 KiB
TypeScript
158 lines
3.8 KiB
TypeScript
import cron from "node-cron";
|
|
import moment from "moment";
|
|
|
|
import { config } from "../config";
|
|
import Logger from "../logger";
|
|
|
|
import MailQueueJob from "./MailQueueJob";
|
|
import FixNodeFilenamesJob from "./FixNodeFilenamesJob";
|
|
import NodeInformationRetrievalJob from "./NodeInformationRetrievalJob";
|
|
import MonitoringMailsSendingJob from "./MonitoringMailsSendingJob";
|
|
import OfflineNodesDeletionJob from "./OfflineNodesDeletionJob";
|
|
|
|
export enum JobResultState {
|
|
OKAY = "okay",
|
|
WARNING = "warning",
|
|
}
|
|
|
|
export type JobResult = {
|
|
state: JobResultState;
|
|
message?: string;
|
|
};
|
|
|
|
export function jobResultOkay(message?: string): JobResult {
|
|
return {
|
|
state: JobResultState.OKAY,
|
|
message,
|
|
};
|
|
}
|
|
|
|
export function jobResultWarning(message?: string): JobResult {
|
|
return {
|
|
state: JobResultState.WARNING,
|
|
message,
|
|
};
|
|
}
|
|
|
|
export interface Job {
|
|
name: string;
|
|
description: string;
|
|
|
|
run(): Promise<JobResult>;
|
|
}
|
|
|
|
export enum TaskState {
|
|
IDLE = "idle",
|
|
RUNNING = "running",
|
|
FAILED = "failed",
|
|
}
|
|
|
|
export class Task {
|
|
constructor(
|
|
public id: number,
|
|
public name: string,
|
|
public description: string,
|
|
public schedule: string,
|
|
public job: Job,
|
|
public runningSince: moment.Moment | null,
|
|
public lastRunStarted: moment.Moment | null,
|
|
public lastRunDuration: number | null,
|
|
public state: TaskState,
|
|
public result: JobResult | null,
|
|
public enabled: boolean
|
|
) {}
|
|
|
|
run(): void {
|
|
if (this.runningSince || !this.enabled) {
|
|
// job is still running, skip execution
|
|
return;
|
|
}
|
|
|
|
this.runningSince = moment();
|
|
this.lastRunStarted = this.runningSince;
|
|
this.state = TaskState.RUNNING;
|
|
|
|
const done = (state: TaskState, result: JobResult | null): void => {
|
|
const now = moment();
|
|
const duration = now.diff(this.runningSince || now);
|
|
Logger.tag("jobs").profile("[%sms]\t%s", duration, this.name);
|
|
|
|
this.runningSince = null;
|
|
this.lastRunDuration = duration;
|
|
this.state = state;
|
|
this.result = result;
|
|
};
|
|
|
|
this.job
|
|
.run()
|
|
.then((result) => {
|
|
done(TaskState.IDLE, result);
|
|
})
|
|
.catch((err) => {
|
|
Logger.tag("jobs").error("Job %s failed: %s", this.name, err);
|
|
done(TaskState.FAILED, null);
|
|
});
|
|
}
|
|
}
|
|
|
|
type Tasks = { [key: string]: Task };
|
|
|
|
const tasks: Tasks = {};
|
|
|
|
let taskId = 1;
|
|
function nextTaskId(): number {
|
|
const id = taskId;
|
|
taskId += 1;
|
|
return id;
|
|
}
|
|
|
|
function schedule(expr: string, job: Job): void {
|
|
Logger.tag("jobs").info("Scheduling job: %s %s", expr, job.name);
|
|
|
|
const id = nextTaskId();
|
|
|
|
const task = new Task(
|
|
id,
|
|
job.name,
|
|
job.description,
|
|
expr,
|
|
job,
|
|
null,
|
|
null,
|
|
null,
|
|
TaskState.IDLE,
|
|
null,
|
|
true
|
|
);
|
|
|
|
cron.schedule(expr, () => task.run());
|
|
|
|
tasks["" + id] = task;
|
|
}
|
|
|
|
export function init() {
|
|
Logger.tag("jobs").info("Scheduling background jobs...");
|
|
|
|
try {
|
|
schedule("0 */1 * * * *", MailQueueJob);
|
|
schedule("15 */1 * * * *", FixNodeFilenamesJob);
|
|
|
|
if (config.client.monitoring.enabled) {
|
|
schedule("30 */15 * * * *", NodeInformationRetrievalJob);
|
|
schedule("45 */5 * * * *", MonitoringMailsSendingJob);
|
|
schedule("0 0 3 * * *", OfflineNodesDeletionJob); // every night at 3:00
|
|
}
|
|
} catch (error) {
|
|
Logger.tag("jobs").error(
|
|
"Error during scheduling of background jobs:",
|
|
error
|
|
);
|
|
throw error;
|
|
}
|
|
|
|
Logger.tag("jobs").info("Scheduling of background jobs done.");
|
|
}
|
|
|
|
export function getTasks(): Tasks {
|
|
return tasks;
|
|
}
|