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();
  }