You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.2 KiB
93 lines
2.2 KiB
1 month ago
|
import {
|
||
|
stderr,
|
||
|
stdout
|
||
|
} from "../chunk-ACQHDOFQ.js";
|
||
|
import {
|
||
|
getHandler,
|
||
|
throwInNextTick
|
||
|
} from "../chunk-E2J7JLFN.js";
|
||
|
import "../chunk-6LX4VMOV.js";
|
||
|
|
||
|
// src/entry/process.ts
|
||
|
process.__tinypool_state__ = {
|
||
|
isChildProcess: true,
|
||
|
isTinypoolWorker: true,
|
||
|
workerData: null,
|
||
|
workerId: Number(process.env.TINYPOOL_WORKER_ID)
|
||
|
};
|
||
|
var memoryUsage = process.memoryUsage.bind(process);
|
||
|
var send = process.send.bind(process);
|
||
|
process.on("message", (message) => {
|
||
|
if (!message || !message.__tinypool_worker_message__)
|
||
|
return;
|
||
|
if (message.source === "pool") {
|
||
|
const { filename, name } = message;
|
||
|
(async function() {
|
||
|
if (filename !== null) {
|
||
|
await getHandler(filename, name);
|
||
|
}
|
||
|
send(
|
||
|
{
|
||
|
ready: true,
|
||
|
source: "pool",
|
||
|
__tinypool_worker_message__: true
|
||
|
},
|
||
|
() => {
|
||
|
}
|
||
|
);
|
||
|
})().catch(throwInNextTick);
|
||
|
return;
|
||
|
}
|
||
|
if (message.source === "port") {
|
||
|
onMessage(message).catch(throwInNextTick);
|
||
|
return;
|
||
|
}
|
||
|
throw new Error(`Unexpected TinypoolWorkerMessage ${JSON.stringify(message)}`);
|
||
|
});
|
||
|
async function onMessage(message) {
|
||
|
const { taskId, task, filename, name } = message;
|
||
|
let response;
|
||
|
try {
|
||
|
const handler = await getHandler(filename, name);
|
||
|
if (handler === null) {
|
||
|
throw new Error(`No handler function exported from ${filename}`);
|
||
|
}
|
||
|
const result = await handler(task);
|
||
|
response = {
|
||
|
source: "port",
|
||
|
__tinypool_worker_message__: true,
|
||
|
taskId,
|
||
|
result,
|
||
|
error: null,
|
||
|
usedMemory: memoryUsage().heapUsed
|
||
|
};
|
||
|
if (stdout()?.writableLength > 0) {
|
||
|
await new Promise((resolve) => process.stdout.write("", resolve));
|
||
|
}
|
||
|
if (stderr()?.writableLength > 0) {
|
||
|
await new Promise((resolve) => process.stderr.write("", resolve));
|
||
|
}
|
||
|
} catch (error) {
|
||
|
response = {
|
||
|
source: "port",
|
||
|
__tinypool_worker_message__: true,
|
||
|
taskId,
|
||
|
result: null,
|
||
|
error: serializeError(error),
|
||
|
usedMemory: memoryUsage().heapUsed
|
||
|
};
|
||
|
}
|
||
|
send(response);
|
||
|
}
|
||
|
function serializeError(error) {
|
||
|
if (error instanceof Error) {
|
||
|
return {
|
||
|
...error,
|
||
|
name: error.name,
|
||
|
stack: error.stack,
|
||
|
message: error.message
|
||
|
};
|
||
|
}
|
||
|
return String(error);
|
||
|
}
|