internal WorkloadGroup(ArmResource options, WorkloadGroupData resource) : base(options, resource.Id)
{
HasData = true;
_data = resource;
_clientDiagnostics = new ClientDiagnostics(ClientOptions);
_workloadGroupsRestClient = new WorkloadGroupsRestOperations(_clientDiagnostics, Pipeline, ClientOptions, BaseUri);
}
WorkloadGroup IOperationSource <WorkloadGroup> .CreateResult(Response response, CancellationToken cancellationToken)
{
using var document = JsonDocument.Parse(response.ContentStream);
var data = WorkloadGroupData.DeserializeWorkloadGroupData(document.RootElement);
return(new WorkloadGroup(_client, data));
}
public async Task <Response <WorkloadGroupData> > GetAsync(string subscriptionId, string resourceGroupName, string serverName, string databaseName, string workloadGroupName, CancellationToken cancellationToken = default)
{
Argument.AssertNotNullOrEmpty(subscriptionId, nameof(subscriptionId));
Argument.AssertNotNullOrEmpty(resourceGroupName, nameof(resourceGroupName));
Argument.AssertNotNullOrEmpty(serverName, nameof(serverName));
Argument.AssertNotNullOrEmpty(databaseName, nameof(databaseName));
Argument.AssertNotNullOrEmpty(workloadGroupName, nameof(workloadGroupName));
using var message = CreateGetRequest(subscriptionId, resourceGroupName, serverName, databaseName, workloadGroupName);
await _pipeline.SendAsync(message, cancellationToken).ConfigureAwait(false);
switch (message.Response.Status)
{
case 200:
{
WorkloadGroupData value = default;
using var document = await JsonDocument.ParseAsync(message.Response.ContentStream, default, cancellationToken).ConfigureAwait(false);
value = WorkloadGroupData.DeserializeWorkloadGroupData(document.RootElement);
return(Response.FromValue(value, message.Response));
}
internal WorkloadGroup(ArmClient armClient, WorkloadGroupData data) : this(armClient, data.Id)
{
HasData = true;
_data = data;
}
internal WorkloadGroupResource(ArmClient client, WorkloadGroupData data) : this(client, data.Id)
{
HasData = true;
_data = data;
}
public virtual async Task <ArmOperation <WorkloadGroupResource> > CreateOrUpdateAsync(WaitUntil waitUntil, string workloadGroupName, WorkloadGroupData data, CancellationToken cancellationToken = default)
{
Argument.AssertNotNullOrEmpty(workloadGroupName, nameof(workloadGroupName));
Argument.AssertNotNull(data, nameof(data));
using var scope = _workloadGroupClientDiagnostics.CreateScope("WorkloadGroupCollection.CreateOrUpdate");
scope.Start();
try
{
var response = await _workloadGroupRestClient.CreateOrUpdateAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Name, Id.Name, workloadGroupName, data, cancellationToken).ConfigureAwait(false);
var operation = new SqlArmOperation <WorkloadGroupResource>(new WorkloadGroupOperationSource(Client), _workloadGroupClientDiagnostics, Pipeline, _workloadGroupRestClient.CreateCreateOrUpdateRequest(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Name, Id.Name, workloadGroupName, data).Request, response, OperationFinalStateVia.Location);
if (waitUntil == WaitUntil.Completed)
{
await operation.WaitForCompletionAsync(cancellationToken).ConfigureAwait(false);
}
return(operation);
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public async virtual Task <WorkloadGroupCreateOrUpdateOperation> CreateOrUpdateAsync(string workloadGroupName, WorkloadGroupData parameters, bool waitForCompletion = true, CancellationToken cancellationToken = default)
{
if (workloadGroupName == null)
{
throw new ArgumentNullException(nameof(workloadGroupName));
}
if (parameters == null)
{
throw new ArgumentNullException(nameof(parameters));
}
using var scope = _clientDiagnostics.CreateScope("WorkloadGroupCollection.CreateOrUpdate");
scope.Start();
try
{
var response = await _workloadGroupsRestClient.CreateOrUpdateAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Name, Id.Name, workloadGroupName, parameters, cancellationToken).ConfigureAwait(false);
var operation = new WorkloadGroupCreateOrUpdateOperation(Parent, _clientDiagnostics, Pipeline, _workloadGroupsRestClient.CreateCreateOrUpdateRequest(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Name, Id.Name, workloadGroupName, parameters).Request, response);
if (waitForCompletion)
{
await operation.WaitForCompletionAsync(cancellationToken).ConfigureAwait(false);
}
return(operation);
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public virtual ArmOperation <WorkloadGroup> CreateOrUpdate(bool waitForCompletion, string workloadGroupName, WorkloadGroupData parameters, CancellationToken cancellationToken = default)
{
Argument.AssertNotNullOrEmpty(workloadGroupName, nameof(workloadGroupName));
Argument.AssertNotNull(parameters, nameof(parameters));
using var scope = _workloadGroupClientDiagnostics.CreateScope("WorkloadGroupCollection.CreateOrUpdate");
scope.Start();
try
{
var response = _workloadGroupRestClient.CreateOrUpdate(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Name, Id.Name, workloadGroupName, parameters, cancellationToken);
var operation = new SqlArmOperation <WorkloadGroup>(new WorkloadGroupOperationSource(Client), _workloadGroupClientDiagnostics, Pipeline, _workloadGroupRestClient.CreateCreateOrUpdateRequest(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Name, Id.Name, workloadGroupName, parameters).Request, response, OperationFinalStateVia.Location);
if (waitForCompletion)
{
operation.WaitForCompletion(cancellationToken);
}
return(operation);
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
