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( }; function addTypeAnnotation(span: TextSpan) { + context.cancellationToken.throwIfCancellationRequested(); + const nodeWithDiag = getTokenAtPosition(sourceFile, span.start); const expandoFunction = findExpandoFunction(nodeWithDiag); @@ -331,6 +333,8 @@ function withContext( } 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( } 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;