Spaces:
Sleeping
Sleeping
File size: 2,027 Bytes
05c5ed5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | "use client";
import { createDebounce, generateUUID } from "lib/utils";
import {
CodeRunnerOptions,
CodeRunnerResult,
CodeWorkerRequest,
CodeWorkerResponse,
} from "./code-runner.interface";
export function callCodeRunWorker(
type: "javascript" | "python",
option: CodeRunnerOptions,
): Promise<CodeRunnerResult> {
let tk: NodeJS.Timeout;
const terminateDebounce = createDebounce();
const terminate = () => {
terminateDebounce(() => {
worker.terminate();
}, 5000);
};
let isWorking = true;
const worker = new Worker(new URL("./worker.ts", import.meta.url));
const promise = new Promise<CodeRunnerResult>((resolve) => {
const id = generateUUID();
const request: CodeWorkerRequest = {
id,
type,
code: option.code,
timeout: option.timeout,
};
setTimeout(() => {
worker.postMessage(request);
}, 1000); // for boot-up effect
worker.onmessage = (event) => {
const response = event.data as CodeWorkerResponse;
if (response.id !== id) return;
if (response.type === "log") {
option.onLog?.(response.entry);
if (!isWorking) terminate();
} else {
resolve(response.result as CodeRunnerResult);
clearTimeout(tk);
terminate();
}
};
});
const race = Promise.race([
promise,
new Promise<CodeRunnerResult>((timeout) => {
tk = setTimeout(() => {
const errorResult: CodeRunnerResult = {
success: false,
logs: [
{
type: "error",
args: [
{
type: "data",
value: JSON.stringify({
type: "error",
message: "Timeout",
}),
},
],
},
],
error: "Timeout",
};
timeout(errorResult);
terminate();
}, option.timeout || 40000);
}),
]);
return race.finally(() => {
isWorking = false;
});
}
|