/// <summary>
/// Periodically calls the server till the long-running operation completes.
/// </summary>
/// <param name="pollingInterval">
/// The interval between status requests to the server.
/// The interval can change based on information returned from the server.
/// For example, the server might communicate to the client that there is not reason to poll for status change sooner than some time.
/// </param>
/// <param name="cancellationToken">A <see cref="CancellationToken"/> used for the periodical service calls.</param>
/// <returns>The last HTTP response received from the server.</returns>
/// <remarks>
/// This method will periodically call UpdateStatusAsync till HasCompleted is true.
/// An API call is then made to retrieve the status of the documents.
/// </remarks>
public async override ValueTask <Response <AsyncPageable <DocumentStatusResult> > > WaitForCompletionAsync(TimeSpan pollingInterval, CancellationToken cancellationToken = default)
{
while (true)
{
await UpdateStatusAsync(cancellationToken).ConfigureAwait(false);
if (HasCompleted)
{
var response = await _serviceClient.GetOperationDocumentsStatusAsync(new Guid(Id), cancellationToken : cancellationToken).ConfigureAwait(false);
_firstPage = Page.FromValues(response.Value.Value, response.Value.NextLink, response.GetRawResponse());
async Task <Page <DocumentStatusResult> > NextPageFunc(string nextLink, int?pageSizeHint)
{
// TODO: diagnostics scope?
try
{
Response <DocumentStatusResponse> response = await _serviceClient.GetOperationDocumentsStatusNextPageAsync(nextLink, new Guid(Id), cancellationToken : cancellationToken).ConfigureAwait(false);
return(Page.FromValues(response.Value.Value, response.Value.NextLink, response.GetRawResponse()));
}
catch (Exception)
{
throw;
}
}
var result = PageableHelpers.CreateAsyncEnumerable(_ => Task.FromResult(_firstPage), NextPageFunc);
return(Response.FromValue(result, response));
}
await Task.Delay(pollingInterval, cancellationToken).ConfigureAwait(false);
}
}
/// <summary>
/// Get the status of a all documents in the operation.
/// </summary>
/// <param name="cancellationToken">A <see cref="CancellationToken"/> used for the service call.</param>
public virtual AsyncPageable <DocumentStatusDetail> GetAllDocumentsStatusAsync(CancellationToken cancellationToken = default)
{
async Task <Page <DocumentStatusDetail> > FirstPageFunc(int?pageSizeHint)
{
using DiagnosticScope scope = _diagnostics.CreateScope($"{nameof(DocumentTranslationOperation)}.{nameof(GetAllDocumentsStatusAsync)}");
scope.Start();
try
{
var response = await _serviceClient.GetOperationDocumentsStatusAsync(new Guid(Id), null, null, cancellationToken).ConfigureAwait(false);
return(Page.FromValues(response.Value.Value, response.Value.NextLink, response.GetRawResponse()));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
async Task <Page <DocumentStatusDetail> > NextPageFunc(string nextLink, int?pageSizeHint)
{
using DiagnosticScope scope = _diagnostics.CreateScope($"{nameof(DocumentTranslationOperation)}.{nameof(GetAllDocumentsStatusAsync)}");
scope.Start();
try
{
var response = await _serviceClient.GetOperationDocumentsStatusNextPageAsync(nextLink, new Guid(Id), cancellationToken : cancellationToken).ConfigureAwait(false);
return(Page.FromValues(response.Value.Value, response.Value.NextLink, response.GetRawResponse()));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
return(PageableHelpers.CreateAsyncEnumerable(FirstPageFunc, NextPageFunc));
}
