Try to increase throughput
This commit is contained in:
@@ -23,21 +23,35 @@ for (let i = 0; i < numWorkers; i++) {
|
|||||||
workerPool.push({ worker, inUse: false });
|
workerPool.push({ worker, inUse: false });
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const requestQueue: ((
|
||||||
|
value:
|
||||||
|
| { worker: Worker; inUse: boolean }
|
||||||
|
| PromiseLike<{ worker: Worker; inUse: boolean }>
|
||||||
|
) => void)[] = [];
|
||||||
|
|
||||||
function getAvailableWorker() {
|
function getAvailableWorker() {
|
||||||
return new Promise((resolve) => {
|
return new Promise((resolve) => {
|
||||||
const check = () => {
|
|
||||||
const availableWorker = workerPool.find((w) => !w.inUse);
|
const availableWorker = workerPool.find((w) => !w.inUse);
|
||||||
if (availableWorker) {
|
if (availableWorker) {
|
||||||
availableWorker.inUse = true;
|
availableWorker.inUse = true;
|
||||||
resolve(availableWorker);
|
resolve(availableWorker);
|
||||||
} else {
|
} else {
|
||||||
setTimeout(check, 100); // Check again in 100ms.
|
requestQueue.push(resolve);
|
||||||
}
|
}
|
||||||
};
|
|
||||||
check();
|
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function releaseWorker(worker) {
|
||||||
|
if (requestQueue.length > 0) {
|
||||||
|
const nextRequest = requestQueue.shift();
|
||||||
|
if (nextRequest) {
|
||||||
|
nextRequest(worker);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
worker.inUse = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
async function handler(req: Request): Promise<Response> {
|
async function handler(req: Request): Promise<Response> {
|
||||||
if (req.method !== "POST" || new URL(req.url).pathname !== "/execute") {
|
if (req.method !== "POST" || new URL(req.url).pathname !== "/execute") {
|
||||||
return new Response("Not Found", { status: 404 });
|
return new Response("Not Found", { status: 404 });
|
||||||
@@ -95,7 +109,7 @@ async function handler(req: Request): Promise<Response> {
|
|||||||
} catch (e) {
|
} catch (e) {
|
||||||
return new Response(`Bad Request: ${e.message}`, { status: 400 });
|
return new Response(`Bad Request: ${e.message}`, { status: 400 });
|
||||||
} finally {
|
} finally {
|
||||||
availableWorker.inUse = false; // Release the worker.
|
releaseWorker(availableWorker); // Release the worker.
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user