src/Transports/MassTransit.Azure.ServiceBus.Core/Testing/AzureServiceBusTestHarnessHostedService.cs - develop - vshmidt/masstransit

src/Transports/MassTransit.Azure.ServiceBus.Core/Testing/AzureServiceBusTestHarnessHostedService.cs

78 строк3 KB

Chris Patterson

Added net10.0 build target

12 ноя 2025, 18:39

12 ноя 2025, 18:397b97076

namespace MassTransit.Testing
{
    using System.Collections.Generic;
    using System.Linq;
    using System.Threading;
    using System.Threading.Tasks;
    using Azure;
    using Azure.Messaging.ServiceBus.Administration;
    using Internals;
    using Microsoft.Extensions.Hosting;
    using Microsoft.Extensions.Logging;
    using Microsoft.Extensions.Options;


    public class AzureServiceBusTestHarnessHostedService :
        IHostedService
    {
        readonly ILogger<AzureServiceBusTestHarnessHostedService> _logger;
        readonly AzureServiceBusTestHarnessOptions _testOptions;
        readonly AzureServiceBusTransportOptions _transportOptions;

        public AzureServiceBusTestHarnessHostedService(IOptions<AzureServiceBusTransportOptions> transportOptions,
            IOptions<AzureServiceBusTestHarnessOptions> testOptions, ILogger<AzureServiceBusTestHarnessHostedService> logger)
        {
            _logger = logger;
            _transportOptions = transportOptions.Value;
            _testOptions = testOptions.Value;
        }

        public async Task StartAsync(CancellationToken cancellationToken)
        {
            if (_testOptions.CleanNamespace)
                await Clean();
        }

        public Task StopAsync(CancellationToken cancellationToken)
        {
            return Task.CompletedTask;
        }

        async Task Clean()
        {
            var managementClient = new ServiceBusAdministrationClient(_transportOptions.ConnectionString);

            var topicCount = 0;
            var queueCount = 0;

            AsyncPageable<TopicProperties> pageableTopics = managementClient.GetTopicsAsync();
            IList<TopicProperties> topics = await pageableTopics.ToListAsync();
            while (topics.Count > 0)
            {
                foreach (var topic in topics)
                {
                    await managementClient.DeleteTopicAsync(topic.Name);
                    topicCount++;
                }

                topics = await managementClient.GetTopicsAsync().ToListAsync();
            }

            AsyncPageable<QueueProperties> pageableQueues = managementClient.GetQueuesAsync();
            IList<QueueProperties> queues = await pageableQueues.ToListAsync();
            while (queues.Count > 0)
            {
                foreach (var queue in queues)
                {
                    await managementClient.DeleteQueueAsync(queue.Name);
                    queueCount++;
                }

                queues = await managementClient.GetQueuesAsync().ToListAsync();
            }

            if (topicCount > 0 || queueCount > 0)
                _logger.LogInformation("Removed {QueueCount} queue(s), {TopicCount} topics(s)", queueCount, topicCount);
        }
    }
}