🤖 Pick PR #58755 (Check cancellation token in isolate...) into releas…

…e-5.5 (#58759) Co-authored-by: Jake Bailey <[email protected]>
microsoft · Jun 3, 2024 · 93585d0 · 93585d0
1 parent 35d2952
commit 93585d0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/services/codefixes/fixMissingTypeAnnotationOnExports.ts b/src/services/codefixes/fixMissingTypeAnnotationOnExports.ts
@@ -254,6 +254,8 @@ function withContext<T>(
     };
 
     function addTypeAnnotation(span: TextSpan) {
+        context.cancellationToken.throwIfCancellationRequested();
+
         const nodeWithDiag = getTokenAtPosition(sourceFile, span.start);
 
         const expandoFunction = findExpandoFunction(nodeWithDiag);
@@ -331,6 +333,8 @@ function withContext<T>(
     }
 
     function addInlineAssertion(span: TextSpan): DiagnosticOrDiagnosticAndArguments | undefined {
+        context.cancellationToken.throwIfCancellationRequested();
+
         const nodeWithDiag = getTokenAtPosition(sourceFile, span.start);
         const expandoFunction = findExpandoFunction(nodeWithDiag);
         // No inline assertions for expando members
@@ -406,6 +410,8 @@ function withContext<T>(
     }
 
     function extractAsVariable(span: TextSpan): DiagnosticOrDiagnosticAndArguments | undefined {
+        context.cancellationToken.throwIfCancellationRequested();
+
         const nodeWithDiag = getTokenAtPosition(sourceFile, span.start);
         const targetNode = findBestFittingNode(nodeWithDiag, span) as Expression;
         if (!targetNode || isValueSignatureDeclaration(targetNode) || isValueSignatureDeclaration(targetNode.parent)) return;