毕设专用git仓库
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

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);
}