ckaczor/sqltoolsservice
1
0
Fork 0
mirror of https://github.com/ckaczor/sqltoolsservice.git synced 2026-02-16 10:58:30 -05:00
Code Issues Packages Projects Releases Wiki Activity

Enable Script option for Server Properties (#2184)

Browse Source
This commit is contained in:
Barbara Valdez
2023-08-22 18:47:45 -07:00
committed by GitHub
parent 6497f5d72d
commit 58923bc75e
1 changed files with 28 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
src/Microsoft.SqlTools.ServiceLayer/ObjectManagement/ObjectTypes/Server/ServerHandler.cs
View File

@@ -3,7 +3,6 @@
// Licensed under the MIT license. See LICENSE file in the project root for full license information. // Licensed under the MIT license. See LICENSE file in the project root for full license information.
// //
using System;
using System.Threading.Tasks; using System.Threading.Tasks;
using Microsoft.SqlTools.ServiceLayer.Connection; using Microsoft.SqlTools.ServiceLayer.Connection;
using Microsoft.SqlTools.ServiceLayer.Management; using Microsoft.SqlTools.ServiceLayer.Management;
@@ -75,38 +74,55 @@ namespace Microsoft.SqlTools.ServiceLayer.ObjectManagement
public override Task Save(ServerViewContext context, ServerInfo obj) public override Task Save(ServerViewContext context, ServerInfo obj)
{ {
UpdateServerProperties(context.Parameters, obj); UpdateServerProperties(context.Parameters, obj, RunType.RunNow);
return Task.CompletedTask; return Task.CompletedTask;
} }
public override Task<string> Script(ServerViewContext context, ServerInfo obj) public override Task<string> Script(ServerViewContext context, ServerInfo obj)
{ {
throw new NotSupportedException("ServerHandler does not support Script method"); var script = UpdateServerProperties(
context.Parameters,
obj,
RunType.ScriptToWindow);
return Task.FromResult(script);
} }
private void UpdateServerProperties(InitializeViewRequestParams viewParams, ServerInfo serverInfo) private string UpdateServerProperties(InitializeViewRequestParams viewParams, ServerInfo serverInfo, RunType runType)
{ {
if (viewParams != null) ConnectionInfo connInfo = this.GetConnectionInfo(viewParams.ConnectionUri);
{
ConnectionInfo connInfo = this.GetConnectionInfo(viewParams.ConnectionUri);
CDataContainer dataContainer = CDataContainer.CreateDataContainer(connInfo);
ServerPrototype prototype = new ServerPrototype(dataContainer); using (var dataContainer = CDataContainer.CreateDataContainer(connInfo))
prototype.ApplyInfoToPrototype(serverInfo); {
ConfigureServer(dataContainer, ConfigAction.Update, RunType.RunNow, prototype); try
{
ServerPrototype prototype = new ServerPrototype(dataContainer);
prototype.ApplyInfoToPrototype(serverInfo);
return ConfigureServer(dataContainer, ConfigAction.Update, runType, prototype);
}
finally
{
dataContainer.ServerConnection.Disconnect();
}
} }
} }
private void ConfigureServer(CDataContainer dataContainer, ConfigAction configAction, RunType runType, ServerPrototype prototype) private string ConfigureServer(CDataContainer dataContainer, ConfigAction configAction, RunType runType, ServerPrototype prototype)
{ {
using (var actions = new ServerActions(dataContainer, prototype, configAction)) using (var actions = new ServerActions(dataContainer, prototype, configAction))
{ {
string sqlScript = string.Empty;
var executionHandler = new ExecutionHandler(actions); var executionHandler = new ExecutionHandler(actions);
executionHandler.RunNow(runType, this); executionHandler.RunNow(runType, this);
if (executionHandler.ExecutionResult == ExecutionMode.Failure) if (executionHandler.ExecutionResult == ExecutionMode.Failure)
{ {
throw executionHandler.ExecutionFailureException; throw executionHandler.ExecutionFailureException;
} }
if (runType == RunType.ScriptToWindow)
{
sqlScript = executionHandler.ScriptTextFromLastRun;
}
return sqlScript;
} }
} }
} }