ente/src/services/upload/queueProcessor.ts

67 lines
1.8 KiB
TypeScript
Raw Normal View History

2021-08-26 10:22:05 +00:00
interface RequestQueueItem {
2021-08-26 11:00:41 +00:00
request: (canceller?: RequestCanceller) => Promise<any>;
2021-08-26 10:22:05 +00:00
callback: (response) => void;
isCanceled: { status: boolean };
canceller: { exec: () => void };
}
2021-08-26 11:00:41 +00:00
interface RequestCanceller {
exec: () => void;
}
export default class QueueProcessor<T> {
2021-08-26 10:22:05 +00:00
private requestQueue: RequestQueueItem[] = [];
private requestInProcessing = 0;
constructor(private maxParallelProcesses: number) {}
2021-08-26 11:00:41 +00:00
public queueUpRequest(request: () => Promise<T>) {
2021-08-26 10:22:05 +00:00
const isCanceled = { status: false };
const canceller: RequestCanceller = {
exec: () => {
isCanceled.status = true;
},
};
2021-08-26 11:00:41 +00:00
const promise = new Promise<T>((resolve) => {
2021-08-26 10:22:05 +00:00
this.requestQueue.push({
request,
callback: resolve,
isCanceled,
canceller,
});
this.pollQueue();
});
2021-08-26 11:00:41 +00:00
return { promise, canceller };
2021-08-26 10:22:05 +00:00
}
async pollQueue() {
if (this.requestInProcessing < this.maxParallelProcesses) {
this.requestInProcessing++;
await this.processQueue();
this.requestInProcessing--;
}
}
public async processQueue() {
while (this.requestQueue.length > 0) {
const queueItem = this.requestQueue.pop();
let response: string;
if (queueItem.isCanceled.status) {
response = null;
} else {
try {
response = await queueItem.request(queueItem.canceller);
} catch (e) {
response = null;
}
2021-08-26 10:22:05 +00:00
}
queueItem.callback(response);
await this.processQueue();
}
}
}