ckaczor/sqltoolsservice
1
0
Fork 0
mirror of https://github.com/ckaczor/sqltoolsservice.git synced 2026-01-13 17:23:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

Transfer URI Correctly (#2265)

Browse Source 
* transfer uri regardless of queryrunner presence

* remove extra space
This commit is contained in:
Christopher Suh
2023-10-09 11:18:51 -07:00
committed by GitHub
parent dacddd250c
commit f0d3a0d546
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/Microsoft.SqlTools.ServiceLayer/Connection/ConnectionService.cs
View File

@@ -925,6 +925,7 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection
{ {
return false; return false;
} }
info.OwnerUri = newOwnerUri;
OwnerToConnectionMap.Remove(originalOwnerUri); OwnerToConnectionMap.Remove(originalOwnerUri);
OwnerToConnectionMap.Add(newOwnerUri, info); OwnerToConnectionMap.Add(newOwnerUri, info);
return true; return true;

2
src/Microsoft.SqlTools.ServiceLayer/QueryExecution/QueryExecutionService.cs
View File

@@ -363,13 +363,13 @@ namespace Microsoft.SqlTools.ServiceLayer.QueryExecution
{ {
string OriginalOwnerUri = changeUriParams.OriginalOwnerUri; string OriginalOwnerUri = changeUriParams.OriginalOwnerUri;
string NewOwnerUri = changeUriParams.NewOwnerUri; string NewOwnerUri = changeUriParams.NewOwnerUri;
ConnectionService.ReplaceUri(OriginalOwnerUri, NewOwnerUri);
// Attempt to load the query // Attempt to load the query
Query query; Query query;
if (!ActiveQueries.TryRemove(OriginalOwnerUri, out query)) if (!ActiveQueries.TryRemove(OriginalOwnerUri, out query))
{ {
throw new Exception("Uri: " + OriginalOwnerUri + " is not associated with an active query."); throw new Exception("Uri: " + OriginalOwnerUri + " is not associated with an active query.");
} }
ConnectionService.ReplaceUri(OriginalOwnerUri, NewOwnerUri);
query.ConnectionOwnerURI = NewOwnerUri; query.ConnectionOwnerURI = NewOwnerUri;
ActiveQueries.TryAdd(NewOwnerUri, query); ActiveQueries.TryAdd(NewOwnerUri, query);
return Task.FromResult(true); return Task.FromResult(true);