74 lines
3.4 KiB
C#
74 lines
3.4 KiB
C#
using Microsoft.Extensions.Configuration;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using StellaOps.Infrastructure.Postgres.Options;
|
|
using StellaOps.Scheduler.Models;
|
|
using StellaOps.Scheduler.Persistence.Postgres;
|
|
using StellaOps.Scheduler.Persistence.Postgres.Repositories;
|
|
|
|
namespace StellaOps.Scheduler.Persistence.Extensions;
|
|
|
|
/// <summary>
|
|
/// Extension methods for configuring Scheduler persistence services.
|
|
/// </summary>
|
|
public static class SchedulerPersistenceExtensions
|
|
{
|
|
/// <summary>
|
|
/// Adds Scheduler PostgreSQL persistence services using configuration section.
|
|
/// </summary>
|
|
/// <param name="services">Service collection.</param>
|
|
/// <param name="configuration">Configuration root.</param>
|
|
/// <param name="sectionName">Configuration section name for PostgreSQL options.</param>
|
|
/// <returns>Service collection for chaining.</returns>
|
|
public static IServiceCollection AddSchedulerPersistence(
|
|
this IServiceCollection services,
|
|
IConfiguration configuration,
|
|
string sectionName = "Postgres:Scheduler")
|
|
{
|
|
services.Configure<PostgresOptions>(sectionName, configuration.GetSection(sectionName));
|
|
services.AddSingleton<SchedulerDataSource>();
|
|
|
|
// Register repositories
|
|
services.AddScoped<IJobRepository, JobRepository>();
|
|
services.AddScoped<ITriggerRepository, TriggerRepository>();
|
|
services.AddScoped<IWorkerRepository, WorkerRepository>();
|
|
services.AddScoped<IDistributedLockRepository, DistributedLockRepository>();
|
|
services.AddScoped<IJobHistoryRepository, JobHistoryRepository>();
|
|
services.AddScoped<IMetricsRepository, MetricsRepository>();
|
|
services.AddScoped<IGraphJobRepository, GraphJobRepository>();
|
|
services.AddScoped<IRunRepository, RunRepository>();
|
|
services.AddScoped<IScheduleRepository, ScheduleRepository>();
|
|
services.AddScoped<IImpactSnapshotRepository, ImpactSnapshotRepository>();
|
|
services.AddScoped<IPolicyRunJobRepository, PolicyRunJobRepository>();
|
|
services.AddScoped<IFailureSignatureRepository, FailureSignatureRepository>();
|
|
services.AddSingleton<IRunSummaryService, RunSummaryService>();
|
|
|
|
return services;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Adds Scheduler PostgreSQL persistence services with explicit options.
|
|
/// </summary>
|
|
/// <param name="services">Service collection.</param>
|
|
/// <param name="configureOptions">Options configuration action.</param>
|
|
/// <returns>Service collection for chaining.</returns>
|
|
public static IServiceCollection AddSchedulerPersistence(
|
|
this IServiceCollection services,
|
|
Action<PostgresOptions> configureOptions)
|
|
{
|
|
services.Configure(configureOptions);
|
|
services.AddSingleton<SchedulerDataSource>();
|
|
|
|
// Register repositories
|
|
services.AddScoped<IJobRepository, JobRepository>();
|
|
services.AddScoped<ITriggerRepository, TriggerRepository>();
|
|
services.AddScoped<IWorkerRepository, WorkerRepository>();
|
|
services.AddScoped<IDistributedLockRepository, DistributedLockRepository>();
|
|
services.AddScoped<IJobHistoryRepository, JobHistoryRepository>();
|
|
services.AddScoped<IMetricsRepository, MetricsRepository>();
|
|
services.AddScoped<IGraphJobRepository, GraphJobRepository>();
|
|
services.AddScoped<IFailureSignatureRepository, FailureSignatureRepository>();
|
|
|
|
return services;
|
|
}
|
|
}
|