Get Connection String request handler (#665)

This commit is contained in:
Karl Burtram
2018-07-25 15:03:00 -04:00
committed by GitHub
parent c7ac3aeaae
commit 7c96ceb501
4 changed files with 90 additions and 5 deletions

View File

@@ -14,11 +14,11 @@ using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.SqlTools.Hosting.Protocol;
using Microsoft.SqlServer.Management.Common;
using Microsoft.SqlTools.ServiceLayer.Connection.Contracts;
using Microsoft.SqlTools.ServiceLayer.Connection.ReliableConnection;
using Microsoft.SqlTools.ServiceLayer.LanguageServices;
using Microsoft.SqlTools.ServiceLayer.LanguageServices.Contracts;
using Microsoft.SqlServer.Management.Common;
using Microsoft.SqlTools.ServiceLayer.Utility;
using Microsoft.SqlTools.Utility;
@@ -949,7 +949,8 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection
serviceHost.SetRequestHandler(CancelConnectRequest.Type, HandleCancelConnectRequest);
serviceHost.SetRequestHandler(DisconnectRequest.Type, HandleDisconnectRequest);
serviceHost.SetRequestHandler(ListDatabasesRequest.Type, HandleListDatabasesRequest);
serviceHost.SetRequestHandler(ChangeDatabaseRequest.Type, HandleChangeDatabase);
serviceHost.SetRequestHandler(ChangeDatabaseRequest.Type, HandleChangeDatabaseRequest);
serviceHost.SetRequestHandler(GetConnectionStringRequest.Type, HandleGetConnectionStringRequest);
}
/// <summary>
@@ -1252,10 +1253,37 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection
return connectionBuilder;
}
/// <summary>
/// Handles a request to get a connection string for the provided connection
/// </summary>
public async Task HandleGetConnectionStringRequest(
GetConnectionStringParams connStringParams,
RequestContext<string> requestContext)
{
await Task.Run(async () =>
{
string connectionString = string.Empty;
ConnectionInfo info;
if (TryFindConnection(connStringParams.OwnerUri, out info))
{
try
{
connectionString = BuildConnectionString(info.ConnectionDetails);
}
catch (Exception e)
{
await requestContext.SendError(e.ToString());
}
}
await requestContext.SendResult(connectionString);
});
}
/// <summary>
/// Handles a request to change the database for a connection
/// </summary>
public async Task HandleChangeDatabase(
public async Task HandleChangeDatabaseRequest(
ChangeDatabaseParams changeDatabaseParams,
RequestContext<bool> requestContext)
{