This commit is contained in:
MARCROCK22 2024-03-16 15:17:30 -04:00
parent e63264bdf1
commit c8e6f43d4f
2 changed files with 3 additions and 6 deletions

View File

@ -1,5 +1,5 @@
import { randomUUID } from 'node:crypto'; import { randomUUID } from 'node:crypto';
import { parentPort, type MessagePort } from 'node:worker_threads'; import { parentPort } from 'node:worker_threads';
import type { WorkerData } from '../../websocket'; import type { WorkerData } from '../../websocket';
import type { WorkerSendCacheRequest } from '../../websocket/discord/worker'; import type { WorkerSendCacheRequest } from '../../websocket/discord/worker';
import type { Adapter } from './types'; import type { Adapter } from './types';
@ -8,10 +8,7 @@ export class WorkerAdapter implements Adapter {
isAsync = true; isAsync = true;
promises = new Map<string, { resolve: (value: unknown) => void; timeout: NodeJS.Timeout }>(); promises = new Map<string, { resolve: (value: unknown) => void; timeout: NodeJS.Timeout }>();
constructor( constructor(public workerData: WorkerData) {}
readonly parent: MessagePort | NodeJS.Process,
public workerData: WorkerData,
) {}
postMessage(body: any) { postMessage(body: any) {
if (parentPort) return parentPort.postMessage(body); if (parentPort) return parentPort.postMessage(body);

View File

@ -65,7 +65,7 @@ export class WorkerClient<Ready extends boolean = boolean> extends BaseClient {
(manager ?? process).on('message', (data: ManagerMessages) => this.handleManagerMessages(data)); (manager ?? process).on('message', (data: ManagerMessages) => this.handleManagerMessages(data));
this.setServices({ this.setServices({
cache: { cache: {
adapter: new WorkerAdapter(manager ?? process, workerData), adapter: new WorkerAdapter(workerData),
disabledCache: options?.disabledCache, disabledCache: options?.disabledCache,
}, },
}); });