예제 #1
0
        public async virtual Task <DiskPoolUpdateOperation> UpdateAsync(bool waitForCompletion, DiskPoolUpdate diskPoolUpdatePayload, CancellationToken cancellationToken = default)
        {
            if (diskPoolUpdatePayload == null)
            {
                throw new ArgumentNullException(nameof(diskPoolUpdatePayload));
            }

            using var scope = _diskPoolClientDiagnostics.CreateScope("DiskPool.Update");
            scope.Start();
            try
            {
                var response = await _diskPoolRestClient.UpdateAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Name, diskPoolUpdatePayload, cancellationToken).ConfigureAwait(false);

                var operation = new DiskPoolUpdateOperation(ArmClient, _diskPoolClientDiagnostics, Pipeline, _diskPoolRestClient.CreateUpdateRequest(Id.SubscriptionId, Id.ResourceGroupName, Id.Name, diskPoolUpdatePayload).Request, response);
                if (waitForCompletion)
                {
                    await operation.WaitForCompletionAsync(cancellationToken).ConfigureAwait(false);
                }
                return(operation);
            }
            catch (Exception e)
            {
                scope.Failed(e);
                throw;
            }
        }
        public virtual async Task <ArmOperation <DiskPoolResource> > UpdateAsync(WaitUntil waitUntil, DiskPoolPatch patch, CancellationToken cancellationToken = default)
        {
            Argument.AssertNotNull(patch, nameof(patch));

            using var scope = _diskPoolClientDiagnostics.CreateScope("DiskPoolResource.Update");
            scope.Start();
            try
            {
                var response = await _diskPoolRestClient.UpdateAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Name, patch, cancellationToken).ConfigureAwait(false);

                var operation = new StoragePoolArmOperation <DiskPoolResource>(new DiskPoolOperationSource(Client), _diskPoolClientDiagnostics, Pipeline, _diskPoolRestClient.CreateUpdateRequest(Id.SubscriptionId, Id.ResourceGroupName, Id.Name, patch).Request, response, OperationFinalStateVia.AzureAsyncOperation);
                if (waitUntil == WaitUntil.Completed)
                {
                    await operation.WaitForCompletionAsync(cancellationToken).ConfigureAwait(false);
                }
                return(operation);
            }
            catch (Exception e)
            {
                scope.Failed(e);
                throw;
            }
        }