[NEW][APPS] New Scheduler API (#19290)
Co-authored-by: Douglas Gubert <douglas.gubert@gmail.com>pull/19379/head
Thassio Victor
5 years ago
committed by
GitHub
parent
96c3c75624
commit
97bdfb56ae
@ -0,0 +1,62 @@ |
|||||||
|
import Agenda from 'agenda'; |
||||||
|
import { MongoInternals } from 'meteor/mongo'; |
||||||
|
|
||||||
|
function _callProcessor(processor) { |
||||||
|
return (job) => processor(job.attrs.data); |
||||||
|
} |
||||||
|
|
||||||
|
export class AppSchedulerBridge { |
||||||
|
constructor(orch) { |
||||||
|
this.orch = orch; |
||||||
|
this.scheduler = new Agenda({ |
||||||
|
mongo: MongoInternals.defaultRemoteCollectionDriver().mongo.client.db(), |
||||||
|
collection: 'rocketchat_agenda_jobs', |
||||||
|
}); |
||||||
|
this.isConnected = false; |
||||||
|
} |
||||||
|
|
||||||
|
async registerProcessors(processors = [], appId) { |
||||||
|
this.orch.debugLog(`The App ${ appId } is registering job processors`, processors); |
||||||
|
processors.forEach(({ id, processor }) => this.scheduler.define(id, _callProcessor(processor))); |
||||||
|
} |
||||||
|
|
||||||
|
async scheduleOnce(job, appId) { |
||||||
|
this.orch.debugLog(`The App ${ appId } is scheduling an onetime job`, job); |
||||||
|
await this.startScheduler(); |
||||||
|
await this.scheduler.schedule(job.when, job.id, job.data || {}); |
||||||
|
} |
||||||
|
|
||||||
|
async scheduleRecurring(job, appId) { |
||||||
|
this.orch.debugLog(`The App ${ appId } is scheduling a recurring job`, job); |
||||||
|
await this.startScheduler(); |
||||||
|
await this.scheduler.every(job.cron, job.id, job.data || {}); |
||||||
|
} |
||||||
|
|
||||||
|
async cancelJob(jobId, appId) { |
||||||
|
this.orch.debugLog(`The App ${ appId } is canceling a job`, jobId); |
||||||
|
await this.startScheduler(); |
||||||
|
try { |
||||||
|
await this.scheduler.cancel({ name: jobId }); |
||||||
|
} catch (e) { |
||||||
|
console.error(e); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
async cancelAllJobs(appId) { |
||||||
|
this.orch.debugLog(`Canceling all jobs of App ${ appId }`); |
||||||
|
await this.startScheduler(); |
||||||
|
const matcher = new RegExp(`^_${ appId }`); |
||||||
|
try { |
||||||
|
await this.scheduler.cancel({ name: { $regex: matcher } }); |
||||||
|
} catch (e) { |
||||||
|
console.error(e); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
async startScheduler() { |
||||||
|
if (!this.isConnected) { |
||||||
|
await this.scheduler.start(); |
||||||
|
this.isConnected = true; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue