96 lines
2.3 KiB
TypeScript
96 lines
2.3 KiB
TypeScript
import * as plugins from './taskbuffer.plugins';
|
|
import { Task } from './taskbuffer.classes.task';
|
|
|
|
// interfaces
|
|
import { Objectmap } from '@pushrocks/lik';
|
|
|
|
export class TaskManager {
|
|
taskMap = new plugins.lik.Objectmap<Task>();
|
|
private cronJobMap = new plugins.lik.Objectmap<ICronJob>();
|
|
constructor() {
|
|
// nothing here
|
|
}
|
|
|
|
/**
|
|
* checks if a task is already present
|
|
* @param taskNameArg
|
|
*/
|
|
getTaskByName(taskNameArg): Task {
|
|
return this.taskMap.find(itemArg => {
|
|
return itemArg.name === taskNameArg;
|
|
});
|
|
}
|
|
|
|
/**
|
|
* adds a Task to the TaskManager
|
|
* @param taskArg
|
|
*/
|
|
addTask(taskArg: Task): void {
|
|
if (!taskArg.name) {
|
|
throw new Error('taskArg needs a name to be added to taskManager');
|
|
}
|
|
this.taskMap.add(taskArg);
|
|
}
|
|
|
|
/**
|
|
* adds and schedules a task at once
|
|
* @param taskArg
|
|
* @param cronStringArg
|
|
*/
|
|
addAndScheduleTask(taskArg: Task, cronStringArg: string) {
|
|
this.addTask(taskArg);
|
|
let taskName = taskArg.name;
|
|
this.scheduleTaskByName(taskName, cronStringArg);
|
|
}
|
|
|
|
/**
|
|
* triggers a task in the TaskManagerByName
|
|
* @param taskNameArg
|
|
*/
|
|
triggerTaskByName(taskNameArg: string): Promise<any> {
|
|
let taskToTrigger = this.getTaskByName(taskNameArg);
|
|
if (!taskToTrigger) {
|
|
throw new Error(`There is no task with the name of ${taskNameArg}`);
|
|
}
|
|
return taskToTrigger.trigger();
|
|
}
|
|
|
|
/**
|
|
* schedules the task by name
|
|
* @param taskNameArg
|
|
*/
|
|
scheduleTaskByName(taskNameArg: string, cronStringArg: string) {
|
|
let taskToSchedule = this.getTaskByName(taskNameArg);
|
|
let job = new plugins.cron.CronJob({
|
|
cronTime: cronStringArg,
|
|
onTick: () => {
|
|
this.triggerTaskByName(taskNameArg);
|
|
},
|
|
start: true
|
|
});
|
|
this.cronJobMap.add({
|
|
taskNameArg: taskToSchedule.name,
|
|
cronString: cronStringArg,
|
|
job: job
|
|
});
|
|
}
|
|
|
|
descheduleTaskByName(taskNameArg: string) {
|
|
let descheduledCron = this.cronJobMap.findOneAndRemove(itemArg => {
|
|
return itemArg.taskNameArg === taskNameArg;
|
|
});
|
|
descheduledCron.job.stop();
|
|
}
|
|
/**
|
|
* returns all schedules of a specific task
|
|
* @param taskNameArg
|
|
*/
|
|
getSchedulesForTaskName(taskNameArg: string) {}
|
|
}
|
|
|
|
export interface ICronJob {
|
|
cronString: string;
|
|
taskNameArg: string;
|
|
job: any;
|
|
}
|