Use a timeout when calling RequestShutdown()

OmniSharp · Feb 13, 2024 · 4fc7cb1 · 4fc7cb1
1 parent 68680df
commit 4fc7cb1
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/Client/LanguageClient.cs b/src/Client/LanguageClient.cs
@@ -423,7 +423,16 @@ public async Task Shutdown()
         {
             if (_connection.IsOpen)
             {
-                await this.RequestShutdown().ConfigureAwait(false);
+                try
+                {
+                    // this can get stuck forever, so use a timeout.
+                    var cancelSource = new CancellationTokenSource();
+                    cancelSource.CancelAfter(5 * 1000);
+
+                    await this.RequestShutdown(cancelSource.Token).ConfigureAwait(false);
+                }
+                catch (TaskCanceledException) { }
+
                 this.SendExit();
             }