feat(core): new worker workspace engine (#9257)

packages/common/infra/src/op/__tests__/message.spec.ts

@@ -50,7 +50,6 @@ describe('message', () => {
       removeEventListener: vi.fn(),
     };
     ctx.handler = new CustomMessageHandler(ctx.receivePort);
-    ctx.handler.listen();
   });
 
   it('should ignore unknown message type', ctx => {

packages/common/infra/src/op/client.ts

@@ -162,16 +162,19 @@ export class OpClient<Ops extends OpSchema> extends AutoMessageHandler {
     op: Op,
     ...args: OpInput<Ops, Op>
   ): Observable<Out> {
-    const payload = args[0];
-
-    const msg = {
-      type: 'subscribe',
-      id: this.nextCallId(op),
-      name: op as string,
-      payload,
-    } satisfies SubscribeMessage;
-
     const sub$ = new Observable<Out>(ob => {
+      const payload = args[0];
+
+      const msg = {
+        type: 'subscribe',
+        id: this.nextCallId(op),
+        name: op as string,
+        payload,
+      } satisfies SubscribeMessage;
+
+      const transferables = fetchTransferables(payload);
+      this.port.postMessage(msg, { transfer: transferables });
+
       this.obs.set(msg.id, ob);
 
       return () => {
@@ -184,9 +187,6 @@ export class OpClient<Ops extends OpSchema> extends AutoMessageHandler {
       };
     });
 
-    const transferables = fetchTransferables(payload);
-    this.port.postMessage(msg, { transfer: transferables });
-
     return sub$;
   }
 

packages/common/infra/src/op/consumer.ts

@@ -1,6 +1,7 @@
 import EventEmitter2 from 'eventemitter2';
 import { defer, from, fromEvent, Observable, of, take, takeUntil } from 'rxjs';
 
+import { MANUALLY_STOP } from '../utils';
 import {
   AutoMessageHandler,
   type CallMessage,
@@ -45,7 +46,7 @@ export class OpConsumer<Ops extends OpSchema> extends AutoMessageHandler {
     };
   }
 
-  private readonly handleCallMessage: MessageHandlers['call'] = async msg => {
+  private readonly handleCallMessage: MessageHandlers['call'] = msg => {
     const abortController = new AbortController();
     this.processing.set(msg.id, abortController);
 
@@ -119,7 +120,7 @@ export class OpConsumer<Ops extends OpSchema> extends AutoMessageHandler {
       return;
     }
 
-    abortController.abort();
+    abortController.abort(MANUALLY_STOP);
   };
 
   register<Op extends OpNames<Ops>>(op: Op, handler: OpHandler<Ops, Op>) {
@@ -181,7 +182,7 @@ export class OpConsumer<Ops extends OpSchema> extends AutoMessageHandler {
     super.close();
     this.registeredOpHandlers.clear();
     this.processing.forEach(controller => {
-      controller.abort();
+      controller.abort(MANUALLY_STOP);
     });
     this.processing.clear();
     this.eventBus.removeAllListeners();

packages/common/infra/src/op/message.ts

@@ -134,7 +134,9 @@ export abstract class AutoMessageHandler {
   private listening = false;
   protected abstract handlers: Partial<MessageHandlers>;
 
-  constructor(protected readonly port: MessageCommunicapable) {}
+  constructor(protected readonly port: MessageCommunicapable) {
+    this.listen();
+  }
 
   protected handleMessage = ignoreUnknownEvent((msg: Messages) => {
     const handler = this.handlers[msg.type];
@@ -145,7 +147,7 @@ export abstract class AutoMessageHandler {
     handler(msg as any);
   });
 
-  listen() {
+  protected listen() {
     if (this.listening) {
       return;
     }