Remove data source from list when final use is disconnected

MarkMpn · Nov 6, 2023 · d4c8592 · d4c8592
1 parent cab37bd
commit d4c8592
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/MarkMpn.Sql4Cds.LanguageServer/Connection/ConnectionManager.cs b/MarkMpn.Sql4Cds.LanguageServer/Connection/ConnectionManager.cs
@@ -49,7 +49,10 @@ public Session Connect(ConnectionDetails connection, string ownerUri)
 
         public void Disconnect(string ownerUri)
         {
-            _connectedDataSource.TryRemove(ownerUri, out _);
+            if (_connectedDataSource.TryRemove(ownerUri, out var dataSourceName) &&
+                !_connectedDataSource.Any(kvp => kvp.Value == dataSourceName))
+                _dataSources.TryRemove(dataSourceName, out _);
+
             _connections.TryRemove(ownerUri, out _);
         }