2022-03-26 13:31:13 +00:00
|
|
|
|
namespace ASC.Api.Core.Extensions;
|
|
|
|
|
public static class ServiceCollectionExtension
|
2022-02-14 11:18:58 +00:00
|
|
|
|
{
|
2022-04-01 16:12:53 +00:00
|
|
|
|
public static void AddCacheNotify(this IServiceCollection services, IConfiguration configuration)
|
|
|
|
|
{
|
|
|
|
|
var redisConfiguration = configuration.GetSection("Redis").Get<RedisConfiguration>();
|
|
|
|
|
var kafkaConfiguration = configuration.GetSection("kafka").Get<KafkaSettings>();
|
|
|
|
|
var rabbitMQConfiguration = configuration.GetSection("RabbitMQ").Get<RabbitMQSettings>();
|
|
|
|
|
|
|
|
|
|
if (redisConfiguration != null)
|
|
|
|
|
{
|
|
|
|
|
services.AddStackExchangeRedisExtensions<NewtonsoftSerializer>(redisConfiguration);
|
|
|
|
|
|
|
|
|
|
services.AddSingleton(typeof(ICacheNotify<>), typeof(RedisCacheNotify<>));
|
|
|
|
|
}
|
|
|
|
|
else if (rabbitMQConfiguration != null)
|
|
|
|
|
{
|
|
|
|
|
services.AddSingleton(typeof(ICacheNotify<>), typeof(RabbitMQCache<>));
|
|
|
|
|
}
|
|
|
|
|
else if (kafkaConfiguration != null)
|
|
|
|
|
{
|
|
|
|
|
services.AddSingleton(typeof(ICacheNotify<>), typeof(KafkaCacheNotify<>));
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
services.AddSingleton(typeof(ICacheNotify<>), typeof(MemoryCacheNotify<>));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static void AddDistributedCache(this IServiceCollection services, IConfiguration configuration)
|
|
|
|
|
{
|
|
|
|
|
var redisConfiguration = configuration.GetSection("Redis").Get<RedisConfiguration>();
|
|
|
|
|
|
|
|
|
|
if (redisConfiguration != null)
|
|
|
|
|
{
|
|
|
|
|
services.AddStackExchangeRedisCache(config =>
|
|
|
|
|
{
|
|
|
|
|
config.ConfigurationOptions = redisConfiguration.ConfigurationOptions;
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
services.AddDistributedMemoryCache();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
public static void AddEventBus(this IServiceCollection services, IConfiguration configuration)
|
2022-02-14 11:18:58 +00:00
|
|
|
|
{
|
2022-03-26 13:31:13 +00:00
|
|
|
|
services.AddSingleton<IEventBusSubscriptionsManager, InMemoryEventBusSubscriptionsManager>();
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
var rabbitMQConfiguration = configuration.GetSection("RabbitMQ").Get<RabbitMQSettings>();
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
if (rabbitMQConfiguration != null)
|
|
|
|
|
{
|
|
|
|
|
services.AddSingleton<IRabbitMQPersistentConnection>(sp =>
|
2022-02-28 14:48:55 +00:00
|
|
|
|
{
|
2022-03-26 13:31:13 +00:00
|
|
|
|
var cfg = sp.GetRequiredService<IConfiguration>();
|
|
|
|
|
var settings = cfg.GetSection("RabbitMQ").Get<RabbitMQSettings>();
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
var logger = sp.GetRequiredService<IOptionsMonitor<ILog>>();
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
var factory = new ConnectionFactory()
|
|
|
|
|
{
|
|
|
|
|
HostName = settings.HostName,
|
2022-04-01 16:12:53 +00:00
|
|
|
|
DispatchConsumersAsync = true
|
2022-03-26 13:31:13 +00:00
|
|
|
|
};
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
if (!string.IsNullOrEmpty(settings.UserName))
|
|
|
|
|
{
|
|
|
|
|
factory.UserName = settings.UserName;
|
|
|
|
|
}
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
if (!string.IsNullOrEmpty(settings.Password))
|
|
|
|
|
{
|
|
|
|
|
factory.Password = settings.Password;
|
|
|
|
|
}
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
var retryCount = 5;
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
if (!string.IsNullOrEmpty(cfg["core:eventBus:connectRetryCount"]))
|
|
|
|
|
{
|
|
|
|
|
retryCount = int.Parse(cfg["core:eventBus:connectRetryCount"]);
|
|
|
|
|
}
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
return new DefaultRabbitMQPersistentConnection(factory, logger, retryCount);
|
|
|
|
|
});
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
services.AddSingleton<IEventBus, EventBusRabbitMQ>(sp =>
|
|
|
|
|
{
|
|
|
|
|
var cfg = sp.GetRequiredService<IConfiguration>();
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
var rabbitMQPersistentConnection = sp.GetRequiredService<IRabbitMQPersistentConnection>();
|
|
|
|
|
var iLifetimeScope = sp.GetRequiredService<ILifetimeScope>();
|
|
|
|
|
var logger = sp.GetRequiredService<IOptionsMonitor<ILog>>();
|
|
|
|
|
var eventBusSubcriptionsManager = sp.GetRequiredService<IEventBusSubscriptionsManager>();
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
var serializer = new ASC.EventBus.Serializers.ProtobufSerializer();
|
2022-03-05 16:46:20 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
var subscriptionClientName = "asc_event_bus_default_queue";
|
2022-02-28 14:48:55 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
if (!string.IsNullOrEmpty(cfg["core:eventBus:subscriptionClientName"]))
|
|
|
|
|
{
|
|
|
|
|
subscriptionClientName = cfg["core:eventBus:subscriptionClientName"];
|
|
|
|
|
}
|
2022-02-28 14:48:55 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
var retryCount = 5;
|
2022-02-28 14:48:55 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
if (!string.IsNullOrEmpty(cfg["core:eventBus:connectRetryCount"]))
|
|
|
|
|
{
|
|
|
|
|
retryCount = int.Parse(cfg["core:eventBus:connectRetryCount"]);
|
|
|
|
|
}
|
2022-02-28 14:48:55 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
return new EventBusRabbitMQ(rabbitMQPersistentConnection, logger, iLifetimeScope, eventBusSubcriptionsManager, serializer, subscriptionClientName, retryCount);
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2022-04-01 16:12:53 +00:00
|
|
|
|
throw new NotImplementedException("EventBus: Provider not found.");
|
2022-03-26 13:31:13 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
/// <remarks>
|
|
|
|
|
/// Add a IHostedService for given type.
|
|
|
|
|
/// Only one copy of this instance type will active in multi process architecture.
|
|
|
|
|
/// </remarks>
|
|
|
|
|
public static void AddActivePassiveHostedService<T>(this IServiceCollection services) where T : class, IHostedService
|
|
|
|
|
{
|
|
|
|
|
var diHelper = new DIHelper(services);
|
2022-02-14 11:18:58 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
diHelper.TryAdd<IRegisterInstanceDao<T>, RegisterInstanceDao<T>>();
|
|
|
|
|
diHelper.TryAdd<IRegisterInstanceManager<T>, RegisterInstanceManager<T>>();
|
2022-02-23 12:30:08 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
services.AddHostedService<RegisterInstanceWorkerService<T>>();
|
2022-02-28 14:48:55 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
diHelper.TryAdd<T>();
|
|
|
|
|
services.AddHostedService<T>();
|
2022-02-23 12:30:08 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
}
|
2022-04-01 16:12:53 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
public static void AddDistributedTaskQueue(this IServiceCollection services)
|
|
|
|
|
{
|
|
|
|
|
services.AddTransient<DistributedTaskQueue>();
|
2022-02-23 12:30:08 +00:00
|
|
|
|
|
2022-03-26 13:31:13 +00:00
|
|
|
|
services.AddSingleton<IDistributedTaskQueueFactory, DefaultDistributedTaskQueueFactory>();
|
2022-02-23 12:30:08 +00:00
|
|
|
|
}
|
2022-02-14 11:18:58 +00:00
|
|
|
|
}
|