Class PersistentWorker

worker/model-worker-manager.ts:56–204

  | { requestId: string; success: false; error: string };

export class PersistentWorker {
  private worker: Worker;
  private view: Int32Array;
  private dataBuffer: Uint8Array;
  private running = true;
  private encoder = new TextEncoder();
  private decoder = new TextDecoder();
  private tmpDir: string;
  private timeout: number;
  private aesKey: AesKey;

  constructor(
    options: {
      tmpDir: string;
      config: {
        timeout: number;
        debug: boolean;
        logLevel: string;
      };
    }
  ) {
    this.tmpDir = options.tmpDir;
    this.timeout = options.config.timeout;
    this.aesKey = generateKey();

    const shared = new SharedArrayBuffer(SHARED_BUFFER_SIZE);
    this.view = new Int32Array(shared);
    this.dataBuffer = new Uint8Array(shared, 4, SHARED_BUFFER_SIZE - 4);

    this.worker = new Worker(new URL('./model-worker.js', import.meta.url), {
      workerData: {
        tmpDir: options.tmpDir,
        config: options.config,
        aesKey: this.aesKey,
        shared,
      },
    });
    
    this.worker.on('online', () => {
      getLogger().info('[WorkerManager] Worker Online');
    });
    
    this.worker.on('message', (msg) => {
      if (msg.type === 'log') {
        if (msg.level === 'warn') {
          getLogger().warn(`[WorkerManager] ${msg.message}`);
        } else if (msg.level === 'error') {
          getLogger().error(`[WorkerManager] ${msg.message}`);
        } else {
          getLogger().info(`[WorkerManager] ${msg.message}`);
        }
      }
    });
    
    this.worker.on('error', (err) => {
      getLogger().error(`[WorkerManager] Error: ${err}`);
      this.running = false;
    });