using System;

using System.Threading;

using System.Threading.Tasks;

using System.Net.Http;

using Microsoft.Rest;

using Microsoft.Azure.Management.ResourceManager;

using Microsoft.Azure.Management.ResourceManager.Models;

#nullable enable

public class CorrelationIdHandler : DelegatingHandler

{

    private readonly string? correlationId;

​

    public CorrelationIdHandler(string? correlationId)

    {

        this.correlationId = correlationId;

    }

​

    protected override Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)

    {

        if (correlationId is not null)

        {

            request.Headers.TryAddWithoutValidation("x-ms-correlation-request-id", correlationId);

        }

​

        return base.SendAsync(request, cancellationToken);

    }

}

​

public class ArmClientFactory

{

    private readonly ServiceClientCredentials credentials;

    public ArmClientFactory(ServiceClientCredentials credentials)

    {

        this.credentials = credentials;

    }

​

    public IResourceManagementClient Create(string? correlationId)

        => new ResourceManagementClient(credentials, new DelegatingHandler[] {

            new CorrelationIdHandler(correlationId),    

        });

}

​

public class ArmManagementGroupScopeOperations

{

    private readonly ArmClientFactory clientFactory;

    private readonly string managementGroupId;

    private readonly string location;

    public ArmManagementGroupScopeOperations(ArmClientFactory clientFactory, string managementGroupId, string location)

    {

        this.clientFactory = clientFactory;

        this.managementGroupId = managementGroupId;

        this.location = location;

    }

​

    public Func<string, Task<DeploymentValidateResult>> ValidateFunc(

        string deploymentName,

        DeploymentProperties deploymentProperties,

        CancellationToken cancellationToken)

    {

        var deployment = new ScopedDeployment

        {

            Properties = deploymentProperties,

            Location = this.location,

        };

        return (correlationId) => clientFactory.Create(correlationId).Deployments.ValidateAtManagementGroupScopeAsync(

            this.managementGroupId,

            deploymentName,

            deployment,

            cancellationToken);

    }

}

​

public class Program

{

    public static async Task Main()

    {

        var clientFactory = new ArmClientFactory(new TokenCredentials("blah"));     

        var ops = new ArmManagementGroupScopeOperations(clientFactory, "myMgId", "West US");

        var validate = ops.ValidateFunc("main", new(), CancellationToken.None);

        var correlationId = Guid.NewGuid().ToString();

        await validate(correlationId);

    }

}