import {spawn, Pool, Worker, Thread} from "@chainsafe/threads"; import {Implementation, PointFormat} from "../../../../src/index.js"; import {PublicKey, Signature} from "../../../../src/types.js"; import {WorkerApi} from "./worker.js"; type ThreadType = { [K in keyof WorkerApi]: (...args: Parameters) => Promise>; }; import path from "path"; import {fileURLToPath} from "url"; const __dirname = path.dirname(fileURLToPath(import.meta.url)); export class BlsMultiThreadNaive { impl: Implementation; pool: Pool; format: PointFormat; constructor(impl: Implementation, workerCount?: number) { this.impl = impl; // Use compressed for herumi for now. // THe worker is not able to deserialize from uncompressed // `Error: err _wrapDeserialize` this.format = impl === "blst-native" ? PointFormat.uncompressed : PointFormat.compressed; this.pool = Pool( () => (spawn( // There is still an annoyance dealing with ESM imports here: // threads.js attempts to require.resolve any files passed to Worker, and // the esm module resolver requires the .js extension, even though the .js file does not actually exist. // The solution for now: // Pass in the script path as an absolute path and suppress threads.js default behavior when importing new Worker(path.join(__dirname, "./worker.js"), {suppressResolveScript: true, suppressTranspileTS: true}) ) as any) as Promise, workerCount ); } async destroy(): Promise { await this.pool.terminate(true); } async verify(pk: PublicKey, msg: Uint8Array, sig: Signature): Promise { return this.pool.queue((worker) => worker.verify(this.impl, pk.toBytes(PointFormat.uncompressed), msg, sig.toBytes(PointFormat.uncompressed)) ); } async verifyMultipleAggregateSignatures( sets: {publicKey: PublicKey; message: Uint8Array; signature: Signature}[] ): Promise { return this.pool.queue((worker) => worker.verifyMultipleAggregateSignatures( this.impl, sets.map((s) => ({ publicKey: s.publicKey.toBytes(PointFormat.uncompressed), message: s.message, signature: s.signature.toBytes(PointFormat.uncompressed), })) ) ); } }