fix: disable ALS stores when no longer needed

Author: chris-olszewski

packages/worker/src/workflow/reusable-vm.ts

@@ -210,7 +210,15 @@ export class ReusableVMWorkflowCreator implements WorkflowCreator {
    * Cleanup the pre-compiled script
    */
   public async destroy(): Promise<void> {
-    // TODO: disable AsyncLocalStorage used for cancellation/update scopes
+    if (this._context) {
+      vm.runInContext(
+        `{
+          __TEMPORAL__.api.destroy();
+        }`,
+        this._context,
+        { timeout: this.isolateExecutionTimeoutMs, displayErrors: true }
+      );
+    }
     globalHandlers.removeWorkflowBundle(this.workflowBundle);
     delete this._context;
   }

packages/worker/src/workflow/vm.ts

@@ -126,6 +126,7 @@ export class VMWorkflowCreator implements WorkflowCreator {
 export class VMWorkflow extends BaseVMWorkflow {
   public async dispose(): Promise<void> {
     this.workflowModule.dispose();
+    this.workflowModule.destroy();
     VMWorkflowCreator.workflowByRunId.delete(this.runId);
     delete this.context;
   }

packages/workflow/src/worker-interface.ts

@@ -10,6 +10,8 @@ import { WorkflowInterceptorsFactory } from './interceptors';
 import { WorkflowCreateOptionsInternal } from './interfaces';
 import { Activator } from './internals';
 import { setActivatorUntyped, getActivator } from './global-attributes';
+import { disableStorage } from './cancellation-scope';
+import { disableUpdateStorage } from './update-scope';
 
 // Export the type for use on the "worker" side
 export { PromiseStackStore } from './internals';
@@ -261,6 +263,11 @@ export function dispose(): void {
   }
 }
 
+export function destroy(): void {
+  disableStorage();
+  disableUpdateStorage();
+}
+
 function isWorkflowFunctionWithOptions(obj: any): obj is WorkflowFunctionWithOptions<any[], any> {
   if (obj == null) return false;
   return Object.hasOwn(obj, 'workflowDefinitionOptions');