public PSKqlScriptResource(KqlScriptResource kqlScriptResource, string workspaceName)
{
this.WorkspaceName = workspaceName;
this.Id = kqlScriptResource?.Id;
this.Name = kqlScriptResource?.Name;
this.Type = kqlScriptResource?.Type;
this.Properties = new PSKqlScript(kqlScriptResource?.Properties);
}
public override void ExecuteCmdlet()
{
if (this.IsParameterBound(c => c.WorkspaceObject))
{
this.WorkspaceName = this.WorkspaceObject.Name;
}
if (!this.IsParameterBound(c => c.Name))
{
string path = this.TryResolvePath(DefinitionFile);
this.Name = Path.GetFileNameWithoutExtension(path);
}
if (this.ShouldProcess(this.WorkspaceName, String.Format(Resources.SettingSynapseKqlScript, this.Name, this.WorkspaceName)))
{
string query = this.ReadFileAsText(DefinitionFile);
KqlScriptResource kqlScript = new KqlScriptResource
{
Properties = new KqlScript
{
Content = new KqlScriptContent
{
Query = query,
Metadata = new KqlScriptContentMetadata
{
Language = "kql"
}
}
}
};
if (this.IsParameterBound(c => c.KustoPoolName))
{
kqlScript.Properties.Content.CurrentConnection = new KqlScriptContentCurrentConnection
{
PoolName = this.KustoPoolName,
DatabaseName = this.KustoPoolDatabaseName
};
}
WriteObject(new PSKqlScriptResource(SynapseAnalyticsClient.CreateOrUpdateKqlScript(this.Name, kqlScript), this.WorkspaceName));
}
}
public async Task <Response <KqlScriptResource> > GetByNameAsync(string kqlScriptName, CancellationToken cancellationToken = default)
{
if (kqlScriptName == null)
{
throw new ArgumentNullException(nameof(kqlScriptName));
}
using var message = CreateGetByNameRequest(kqlScriptName);
await _pipeline.SendAsync(message, cancellationToken).ConfigureAwait(false);
switch (message.Response.Status)
{
case 200:
{
KqlScriptResource value = default;
using var document = await JsonDocument.ParseAsync(message.Response.ContentStream, default, cancellationToken).ConfigureAwait(false);
value = KqlScriptResource.DeserializeKqlScriptResource(document.RootElement);
return(Response.FromValue(value, message.Response));
}
internal HttpMessage CreateCreateOrUpdateRequest(string kqlScriptName, KqlScriptResource kqlScript)
{
var message = _pipeline.CreateMessage();
var request = message.Request;
request.Method = RequestMethod.Put;
var uri = new RawRequestUriBuilder();
uri.Reset(endpoint);
uri.AppendPath("/kqlScripts/", false);
uri.AppendPath(kqlScriptName, true);
uri.AppendQuery("api-version", "2021-11-01-preview", true);
request.Uri = uri;
request.Headers.Add("Accept", "application/json");
request.Headers.Add("Content-Type", "application/json");
var content = new Utf8JsonRequestContent();
content.JsonWriter.WriteObjectValue(kqlScript);
request.Content = content;
return(message);
}
public Response CreateOrUpdate(string kqlScriptName, KqlScriptResource kqlScript, CancellationToken cancellationToken = default)
{
if (kqlScriptName == null)
{
throw new ArgumentNullException(nameof(kqlScriptName));
}
if (kqlScript == null)
{
throw new ArgumentNullException(nameof(kqlScript));
}
using var message = CreateCreateOrUpdateRequest(kqlScriptName, kqlScript);
_pipeline.Send(message, cancellationToken);
switch (message.Response.Status)
{
case 200:
case 202:
return(message.Response);
default:
throw _clientDiagnostics.CreateRequestFailedException(message.Response);
}
}
public async Task <Response> CreateOrUpdateAsync(string kqlScriptName, KqlScriptResource kqlScript, CancellationToken cancellationToken = default)
{
if (kqlScriptName == null)
{
throw new ArgumentNullException(nameof(kqlScriptName));
}
if (kqlScript == null)
{
throw new ArgumentNullException(nameof(kqlScript));
}
using var message = CreateCreateOrUpdateRequest(kqlScriptName, kqlScript);
await _pipeline.SendAsync(message, cancellationToken).ConfigureAwait(false);
switch (message.Response.Status)
{
case 200:
case 202:
return(message.Response);
default:
throw await _clientDiagnostics.CreateRequestFailedExceptionAsync(message.Response).ConfigureAwait(false);
}
}
public KqlScriptResource CreateOrUpdateKqlScript(string kqlScriptName, KqlScriptResource kqlScript)
{
var operation = _kqlScriptClient.StartCreateOrUpdate(kqlScriptName, kqlScript);
return(operation.Poll().Value);
}
public virtual KqlScriptCreateOrUpdateOperation StartCreateOrUpdate(string kqlScriptName, KqlScriptResource kqlScript, CancellationToken cancellationToken = default)
{
if (kqlScriptName == null)
{
throw new ArgumentNullException(nameof(kqlScriptName));
}
if (kqlScript == null)
{
throw new ArgumentNullException(nameof(kqlScript));
}
using var scope = _clientDiagnostics.CreateScope("KqlScriptClient.StartCreateOrUpdate");
scope.Start();
try
{
var originalResponse = RestClient.CreateOrUpdate(kqlScriptName, kqlScript, cancellationToken);
return(new KqlScriptCreateOrUpdateOperation(_clientDiagnostics, _pipeline, RestClient.CreateCreateOrUpdateRequest(kqlScriptName, kqlScript).Request, originalResponse));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
