AzureIdentityHttpClient.cs

using Azure.Core;
using System.Text.Json;

namespace Sandbox;

public sealed class AzureIdentityHttpClient(HttpClient httpClient)
{
    public async Task<T?> SendAsync<T>(
        Func<HttpResponseMessage, JsonSerializerOptions, CancellationToken, ValueTask<T>> handler,
        HttpMethod httpMethod,
        JsonSerializerOptions jsonSerializerOptions,
        TokenRequestContext tokenRequestContext,
        Uri uri,
        HttpCompletionOption httpCompletionOption = default,
        CancellationToken cancellationToken = default
    ) {
        using var httpRequestMessage = new HttpRequestMessage(
            method: httpMethod,
            requestUri: uri
        );

        httpRequestMessage.Options.Set(
            key: AzureIdentityAuthenticationHandler.TokenRequestContextKey,
            value: tokenRequestContext
        );

        using var httpResponseMessage = await httpClient
            .SendAsync(
                cancellationToken: cancellationToken,
                completionOption: httpCompletionOption,
                request: httpRequestMessage
            )
            .ConfigureAwait(continueOnCapturedContext: false);

        return await handler(
                arg1: httpResponseMessage,
                arg2: jsonSerializerOptions,
                arg3: cancellationToken
            )
            .ConfigureAwait(continueOnCapturedContext: false);
    }
}