utils/structures/Queue.ts

import DiscordClient from "../../client/Client";
import { v4 as uuid } from 'uuid';
import QueuedJob, { IQueuedJob } from "../../models/QueuedJob";

export interface QueueOptions {
    runAfter?: number;
    id?: string;
    runAt?: Date;
    model: IQueuedJob;
}

export default abstract class Queue {
    protected completed = false;
    protected runOn: number = 0;
    public readonly id: string;
    protected readonly model: IQueuedJob;
    protected readonly timeout: NodeJS.Timeout;

    constructor(protected client: DiscordClient, { runAfter, id, runAt, model }: QueueOptions) {
        if (runAfter !== 0 && runAfter !== 0 && !runAfter && !runAt) {
            throw new Error("One of runAfter or runAt must be specified for creating a queue");
        }

        this.runOn = runAfter ? Date.now() + runAfter : runAt!.getTime();
        this.id = id ?? uuid();
        this.model = model;

        const ms = this.runOn - Date.now();

        this.timeout = setTimeout(async () => {
            await this.run();
            await this.finish();
        }, ms < 0 ? 0 : ms);

        console.log('Queue created: ', this.constructor.name, this.id);
    }

    get data() {
        return this.model.data;
    }

    async finish() {
        this.client.queueManager.removeQueue(this);
        console.log("Job complete: ", this.constructor.name);
    }

    async cancel() {
        clearTimeout(this.timeout);
        await this.model.delete();
        this.client.queueManager.removeQueue(this);
    }

    abstract execute(data?: { [key: string | number]: any }): Promise<any>;

    async run() {
        this.completed = true;
        this.model.delete();

        try {
            return await this.execute(this.model.data);
        }
        catch (e) {
            console.error(`An error occurred in queue job\nJob ID: ${this.id}`, e);
        }
    }
}
1	import DiscordClient from "../../client/Client";
2	import { v4 as uuid } from 'uuid';
3	import QueuedJob, { IQueuedJob } from "../../models/QueuedJob";
4
5	export interface QueueOptions {
6	runAfter?: number;
7	id?: string;
8	runAt?: Date;
9	model: IQueuedJob;
10	}
11
12	export default abstract class Queue {
13	protected completed = false;
14	protected runOn: number = 0;
15	public readonly id: string;
16	protected readonly model: IQueuedJob;
17	protected readonly timeout: NodeJS.Timeout;
18
19	constructor(protected client: DiscordClient, { runAfter, id, runAt, model }: QueueOptions) {
20	if (runAfter !== 0 && runAfter !== 0 && !runAfter && !runAt) {
21	throw new Error("One of runAfter or runAt must be specified for creating a queue");
22	}
23
24	this.runOn = runAfter ? Date.now() + runAfter : runAt!.getTime();
25	this.id = id ?? uuid();
26	this.model = model;
27
28	const ms = this.runOn - Date.now();
29
30	this.timeout = setTimeout(async () => {
31	await this.run();
32	await this.finish();
33	}, ms < 0 ? 0 : ms);
34
35	console.log('Queue created: ', this.constructor.name, this.id);
36	}
37
38	get data() {
39	return this.model.data;
40	}
41
42	async finish() {
43	this.client.queueManager.removeQueue(this);
44	console.log("Job complete: ", this.constructor.name);
45	}
46
47	async cancel() {
48	clearTimeout(this.timeout);
49	await this.model.delete();
50	this.client.queueManager.removeQueue(this);
51	}
52
53	abstract execute(data?: { [key: string \| number]: any }): Promise<any>;
54
55	async run() {
56	this.completed = true;
57	this.model.delete();
58
59	try {
60	return await this.execute(this.model.data);
61	}
62	catch (e) {
63	console.error(`An error occurred in queue job\nJob ID: ${this.id}`, e);
64	}
65	}
66	}
team@onesoftnet.eu.org	ViewVC Help
Powered by ViewVC 1.1.26