AzureServiceBusFactory.cs 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. using Microsoft.Azure.Cosmos.Table;
  2. using Microsoft.Extensions.Options;
  3. using Microsoft.Extensions.Logging;
  4. using System;
  5. using System.Collections.Generic;
  6. using System.Text;
  7. using Microsoft.Extensions.DependencyInjection;
  8. using Azure.Storage.Blobs;
  9. using Azure.Storage.Blobs.Models;
  10. using Azure.Storage.Blobs.Specialized;
  11. using StackExchange.Redis;
  12. using System.Collections.Concurrent;
  13. using Azure.Messaging.ServiceBus;
  14. namespace TEAMModelOS.SDK.DI
  15. {
  16. public class AzureServiceBusFactory
  17. {
  18. private readonly IServiceProvider _services;
  19. private readonly IOptionsMonitor<AzureServiceBusFactoryOptions> _optionsMonitor;
  20. private readonly ILogger _logger;
  21. private ConcurrentDictionary<string, ServiceBusClient> ServiceBusClients { get; } = new ConcurrentDictionary<string, ServiceBusClient>();
  22. public AzureServiceBusFactory(IServiceProvider services, IOptionsMonitor<AzureServiceBusFactoryOptions> optionsMonitor, ILogger<AzureServiceBusFactory> logger)
  23. {
  24. if (services == null) throw new ArgumentNullException(nameof(services));
  25. if (optionsMonitor == null) throw new ArgumentNullException(nameof(optionsMonitor));
  26. _services = services;
  27. _optionsMonitor = optionsMonitor;
  28. _logger = logger;
  29. }
  30. public ServiceBusClient GetServiceBusClient(string name = "Default")
  31. {
  32. try
  33. {
  34. var client = ServiceBusClients.GetOrAdd(name, x => new ServiceBusClient(_optionsMonitor.Get(name).ServiceBusConnectionString));
  35. return client;
  36. }
  37. catch (OptionsValidationException e)
  38. {
  39. _logger?.LogWarning(e, e.Message);
  40. return null;
  41. }
  42. }
  43. public TopicClient GetTopicClient(ReceiveMode receiveMode, string name = "Default")
  44. {
  45. try
  46. {
  47. ServiceBusConnectionStringBuilder builder = new ServiceBusConnectionStringBuilder(_optionsMonitor.Get(name).ServiceBusConnectionString);
  48. TopicClient client = new TopicClient(builder, receiveMode);
  49. return client;
  50. }
  51. catch (OptionsValidationException e)
  52. {
  53. _logger?.LogWarning(e, e.Message);
  54. return null;
  55. }
  56. }
  57. }
  58. }