Cancellable: set token in more places

dotnet · Jan 29, 2025 · 900521f · 900521f
1 parent 2edab12
commit 900521f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/Compiler/Service/BackgroundCompiler.fs b/src/Compiler/Service/BackgroundCompiler.fs
@@ -503,6 +503,7 @@ type internal BackgroundCompiler
         match tryGetBuilder options with
         | Some getBuilder ->
             async {
+                do! Cancellable.UseToken()
                 match! getBuilder with
                 | builderOpt, creationDiags when builderOpt.IsNone || not builderOpt.Value.IsReferencesInvalidated ->
                     return builderOpt, creationDiags

diff --git a/src/Compiler/Service/ServiceAnalysis.fs b/src/Compiler/Service/ServiceAnalysis.fs
@@ -6,6 +6,7 @@ open System
 open System.Collections.Generic
 open System.Runtime.CompilerServices
 open Internal.Utilities.Library
+open FSharp.Compiler
 open FSharp.Compiler.CodeAnalysis
 open FSharp.Compiler.Symbols
 open FSharp.Compiler.Syntax.PrettyNaming
@@ -301,6 +302,7 @@ module UnusedOpens =
     /// Async to allow cancellation.
     let getUnusedOpens (checkFileResults: FSharpCheckFileResults, getSourceLineStr: int -> string) : Async<range list> =
         async {
+            do! Cancellable.UseToken()
             if checkFileResults.OpenDeclarations.Length = 0 then
                 return []
             else