From bc31156d2c098e28cb065f285005a4a1fa824e75 Mon Sep 17 00:00:00 2001
From: Jake Bailey <5341706+jakebailey@users.noreply.github.com>
Date: Mon, 3 Jun 2024 10:39:36 -0700
Subject: [PATCH] Merge 14f9e666d49afb2ebcd93f88a3cc0ce183769ba1 into
 5041ab992f527d7f9d82d892f04c040c1d162a02

---
 src/services/codefixes/fixMissingTypeAnnotationOnExports.ts | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/services/codefixes/fixMissingTypeAnnotationOnExports.ts b/src/services/codefixes/fixMissingTypeAnnotationOnExports.ts
index 70ceb0c31eba3..440fc4d05c447 100644
--- 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;