Add Shutdown(CancellationToken token)

OmniSharp · Feb 14, 2024 · 71ef7e4 · 71ef7e4
1 parent 68680df
commit 71ef7e4
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/Client/LanguageClient.cs b/src/Client/LanguageClient.cs
@@ -420,10 +420,20 @@ private void RegisterCapabilities(ClientCapabilities capabilities)
         }
 
         public async Task Shutdown()
+        {
+            await Shutdown(CancellationToken.None);
+        }
+
+        public async Task Shutdown(CancellationToken token)
         {
             if (_connection.IsOpen)
             {
-                await this.RequestShutdown().ConfigureAwait(false);
+                try
+                {
+                    await this.RequestShutdown(token).ConfigureAwait(false);
+                }
+                catch (TaskCanceledException) { }
+
                 this.SendExit();
             }