87 lines
2.1 KiB
Plaintext
87 lines
2.1 KiB
Plaintext
"use strict";
|
|
import {
|
|
stderr,
|
|
stdout
|
|
} from "../chunk-OECBSOR6.js";
|
|
import {
|
|
getHandler,
|
|
throwInNextTick
|
|
} from "../chunk-DSRZHYCS.js";
|
|
import "../chunk-T6A5DJAH.js";
|
|
|
|
// src/entry/process.ts
|
|
process.__tinypool_state__ = {
|
|
isChildProcess: true,
|
|
isTinypoolWorker: true,
|
|
workerData: null,
|
|
workerId: Number(process.env.TINYPOOL_WORKER_ID)
|
|
};
|
|
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);
|
|
}
|
|
process.send({
|
|
ready: true,
|
|
source: "pool",
|
|
__tinypool_worker_message__: true
|
|
});
|
|
})().catch(throwInNextTick);
|
|
return;
|
|
}
|
|
if (message.source === "port") {
|
|
return onMessage(message).catch(throwInNextTick);
|
|
}
|
|
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: process.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: process.memoryUsage().heapUsed
|
|
};
|
|
}
|
|
process.send(response);
|
|
}
|
|
function serializeError(error) {
|
|
if (error instanceof Error) {
|
|
return {
|
|
...error,
|
|
name: error.name,
|
|
stack: error.stack,
|
|
message: error.message
|
|
};
|
|
}
|
|
return String(error);
|
|
}
|