48 lines
1.2 KiB
TypeScript
48 lines
1.2 KiB
TypeScript
import { type MessagePort, Worker } from 'worker_threads';
|
|
import * as path from 'path';
|
|
|
|
export enum WorkerNameEnum {
|
|
runJs = 'runJs',
|
|
runPy = 'runPy'
|
|
}
|
|
|
|
type WorkerResponseType = { type: 'success' | 'error'; data: any };
|
|
|
|
export const getWorker = (name: WorkerNameEnum) => {
|
|
const baseUrl =
|
|
process.env.NODE_ENV === 'production' ? 'projects/sandbox/dist/worker' : 'dist/worker';
|
|
const workerPath = path.join(process.cwd(), baseUrl, `${name}.js`);
|
|
return new Worker(workerPath);
|
|
};
|
|
|
|
export const runWorker = <T = any>(name: WorkerNameEnum, params?: Record<string, any>) => {
|
|
return new Promise<T>((resolve, reject) => {
|
|
const worker = getWorker(name);
|
|
|
|
worker.postMessage(params);
|
|
|
|
worker.on('message', (msg: WorkerResponseType) => {
|
|
if (msg.type === 'error') return reject(msg.data);
|
|
|
|
resolve(msg.data);
|
|
worker.terminate();
|
|
});
|
|
|
|
worker.on('error', (err) => {
|
|
reject(err);
|
|
worker.terminate();
|
|
});
|
|
worker.on('messageerror', (err) => {
|
|
reject(err);
|
|
worker.terminate();
|
|
});
|
|
});
|
|
};
|
|
|
|
export const workerResponse = ({
|
|
parentPort,
|
|
...data
|
|
}: WorkerResponseType & { parentPort?: MessagePort }) => {
|
|
parentPort?.postMessage(data);
|
|
};
|