wtrpc

Type safe node js worker threads

Define a contract contract.ts

type RpcContract = {
    scan: (req: {pid: number, pattern: string}) => number[]
}

Implement the contract as a worker

worker.ts

import { WorkerRpc } from '@wtrpc/core'
import { RpcContract } from './contract.js'
import { parentPort } from 'node:worker_threads';

const worker = new WorkerRpc<RpcContract>({
    scan(req: {pid: number, pattern: string}): number[] {
        return [-1]
    }
})

// Bind the worker to the worker_thread parent port if it exists
if (parentPort) worker.bind(parentPort);

pool.ts

import { WorkerRpcPool } from '@wtrpc/core'
import { RpcContract } from './contract.js'

const workerUrl = new URL('./worker.js', import.meta.url);
const threadCount = os.cpus().length;

const pool = new WorkerRpcPool<RpcContract>(threadCount, workerUrl);

// Run the task on the worker thread pool

const tasks = [
    { pid: 37 },
    { pid: 50 },
    { pid: 99 }
]

const offsets = tasks.map(t => pool.run('scan', { pid: t.pid, pattern: '00 ?? 00 ?? 07' }));
const results = await Promise.all(offsets)
// [-1]