118 lines
2.7 KiB
TypeScript
118 lines
2.7 KiB
TypeScript
import * as store from "./store.ts";
|
|
import { Task } from "../lib/types.ts";
|
|
import * as notifications from "./notifications.ts";
|
|
|
|
export interface QueueTask extends Task {
|
|
status?: string;
|
|
device: string;
|
|
}
|
|
|
|
export interface StageTask extends Task {
|
|
devices: string[];
|
|
}
|
|
|
|
const MAX_QUEUE = 100;
|
|
|
|
const queue: Set<QueueTask> = new Set();
|
|
const staging: Set<StageTask> = new Set();
|
|
|
|
function canQueue(tasks: QueueTask[]): boolean {
|
|
let count = queue.size;
|
|
for (const task of tasks) if (!queue.has(task)) ++count;
|
|
return count <= MAX_QUEUE;
|
|
}
|
|
|
|
export function queueTask(...tasks: QueueTask[]): void {
|
|
if (!canQueue(tasks)) {
|
|
notifications.push("error", "Too many tasks in queue");
|
|
return;
|
|
}
|
|
|
|
for (const task of tasks) {
|
|
task.status = "queued";
|
|
queue.add(task);
|
|
}
|
|
}
|
|
|
|
export function deleteTask(task: QueueTask): void {
|
|
queue.delete(task);
|
|
}
|
|
|
|
export function getQueue(): Set<QueueTask> {
|
|
return queue;
|
|
}
|
|
|
|
export function clear(): void {
|
|
queue.clear();
|
|
}
|
|
|
|
export function getStaging(): Set<StageTask> {
|
|
return staging;
|
|
}
|
|
|
|
export function clearStaging(): void {
|
|
staging.clear();
|
|
}
|
|
|
|
export function stageSpv(task: StageTask): void {
|
|
if (queue.size + task.devices.length > MAX_QUEUE) {
|
|
notifications.push("error", "Too many tasks in queue");
|
|
return;
|
|
}
|
|
staging.add(task);
|
|
}
|
|
|
|
export function stageDownload(task: StageTask): void {
|
|
if (queue.size + task.devices.length > MAX_QUEUE) {
|
|
notifications.push("error", "Too many tasks in queue");
|
|
return;
|
|
}
|
|
staging.add(task);
|
|
}
|
|
|
|
export function commit(
|
|
tasks: QueueTask[],
|
|
callback: (
|
|
deviceId: string,
|
|
err: Error,
|
|
conReqStatus: string,
|
|
_tasks: QueueTask[],
|
|
) => void,
|
|
): Promise<void> {
|
|
const devices: { [deviceId: string]: QueueTask[] } = {};
|
|
|
|
if (!canQueue(tasks))
|
|
return Promise.reject(new Error("Too many tasks in queue"));
|
|
|
|
for (const t of tasks) {
|
|
devices[t.device] = devices[t.device] || [];
|
|
devices[t.device].push(t);
|
|
t.status = "queued";
|
|
queue.add(t);
|
|
}
|
|
|
|
return new Promise((resolve) => {
|
|
let counter = 1;
|
|
for (const [deviceId, tasks2] of Object.entries(devices)) {
|
|
++counter;
|
|
store
|
|
.postTasks(deviceId, tasks2)
|
|
.then((connectionRequestStatus) => {
|
|
for (const t of tasks2) {
|
|
if (t.status === "pending") t.status = "stale";
|
|
else if (t.status === "done") queue.delete(t);
|
|
}
|
|
callback(deviceId, null, connectionRequestStatus, tasks2);
|
|
if (--counter === 0) resolve();
|
|
})
|
|
.catch((err) => {
|
|
for (const t of tasks2) t.status = "stale";
|
|
callback(deviceId, err, null, tasks2);
|
|
if (--counter === 0) resolve();
|
|
});
|
|
}
|
|
|
|
if (--counter === 0) resolve();
|
|
});
|
|
}
|