From da9433fd1310066212d661300583264f9c2daf10 Mon Sep 17 00:00:00 2001
From: Tomas Matousek <tomat@microsoft.com>
Date: Wed, 5 Jun 2024 11:55:08 -0700
Subject: [PATCH] Ignore workspaces that do not have
 IEditAndContinueWorkspaceService

---
 .../EditAndContinueDiagnosticSource_OpenDocument.cs             | 2 +-
 .../EditAndContinueDiagnosticSource_Workspace.cs                | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/LanguageServer/Protocol/Features/EditAndContinue/EditAndContinueDiagnosticSource_OpenDocument.cs b/src/LanguageServer/Protocol/Features/EditAndContinue/EditAndContinueDiagnosticSource_OpenDocument.cs
index deb7877c1b234..da2a491dd33c1 100644
--- a/src/LanguageServer/Protocol/Features/EditAndContinue/EditAndContinueDiagnosticSource_OpenDocument.cs
+++ b/src/LanguageServer/Protocol/Features/EditAndContinue/EditAndContinueDiagnosticSource_OpenDocument.cs
@@ -28,7 +28,7 @@ public override async Task<ImmutableArray<DiagnosticData>> GetDiagnosticsAsync(R
             var services = designTimeSolution.Services;
 
             // avoid creating and synchronizing compile-time solution if Hot Reload/EnC session is not active
-            if (services.GetRequiredService<IEditAndContinueWorkspaceService>().SessionTracker is not { IsSessionActive: true } sessionStateTracker)
+            if (services.GetService<IEditAndContinueWorkspaceService>()?.SessionTracker is not { IsSessionActive: true } sessionStateTracker)
             {
                 return [];
             }
diff --git a/src/LanguageServer/Protocol/Features/EditAndContinue/EditAndContinueDiagnosticSource_Workspace.cs b/src/LanguageServer/Protocol/Features/EditAndContinue/EditAndContinueDiagnosticSource_Workspace.cs
index b84fee77d2030..4e229286b1e7a 100644
--- a/src/LanguageServer/Protocol/Features/EditAndContinue/EditAndContinueDiagnosticSource_Workspace.cs
+++ b/src/LanguageServer/Protocol/Features/EditAndContinue/EditAndContinueDiagnosticSource_Workspace.cs
@@ -37,7 +37,7 @@ public override Task<ImmutableArray<DiagnosticData>> GetDiagnosticsAsync(Request
 
     public static async ValueTask<ImmutableArray<IDiagnosticSource>> CreateWorkspaceDiagnosticSourcesAsync(Solution solution, Func<Document, bool> isDocumentOpen, CancellationToken cancellationToken)
     {
-        if (solution.Services.GetRequiredService<IEditAndContinueWorkspaceService>().SessionTracker is not { IsSessionActive: true } sessionStateTracker)
+        if (solution.Services.GetService<IEditAndContinueWorkspaceService>()?.SessionTracker is not { IsSessionActive: true } sessionStateTracker)
         {
             return [];
         }