89 lines
3.8 KiB
C#
89 lines
3.8 KiB
C#
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.DependencyInjection.Extensions;
|
|
using Microsoft.Extensions.Logging;
|
|
using Microsoft.Extensions.Options;
|
|
using MongoDB.Driver;
|
|
using StellaOps.Feedser.Core.Jobs;
|
|
using StellaOps.Feedser.Storage.Mongo.Advisories;
|
|
using StellaOps.Feedser.Storage.Mongo.ChangeHistory;
|
|
using StellaOps.Feedser.Storage.Mongo.Documents;
|
|
using StellaOps.Feedser.Storage.Mongo.Dtos;
|
|
using StellaOps.Feedser.Storage.Mongo.Exporting;
|
|
using StellaOps.Feedser.Storage.Mongo.JpFlags;
|
|
using StellaOps.Feedser.Storage.Mongo.MergeEvents;
|
|
using StellaOps.Feedser.Storage.Mongo.PsirtFlags;
|
|
using StellaOps.Feedser.Storage.Mongo.Migrations;
|
|
|
|
namespace StellaOps.Feedser.Storage.Mongo;
|
|
|
|
public static class ServiceCollectionExtensions
|
|
{
|
|
public static IServiceCollection AddMongoStorage(this IServiceCollection services, Action<MongoStorageOptions> configureOptions)
|
|
{
|
|
ArgumentNullException.ThrowIfNull(services);
|
|
ArgumentNullException.ThrowIfNull(configureOptions);
|
|
|
|
services.AddOptions<MongoStorageOptions>()
|
|
.Configure(configureOptions)
|
|
.PostConfigure(static options => options.EnsureValid());
|
|
|
|
services.TryAddSingleton(TimeProvider.System);
|
|
|
|
services.AddSingleton<IMongoClient>(static sp =>
|
|
{
|
|
var options = sp.GetRequiredService<IOptions<MongoStorageOptions>>().Value;
|
|
return new MongoClient(options.ConnectionString);
|
|
});
|
|
|
|
services.AddSingleton(static sp =>
|
|
{
|
|
var options = sp.GetRequiredService<IOptions<MongoStorageOptions>>().Value;
|
|
var client = sp.GetRequiredService<IMongoClient>();
|
|
var settings = new MongoDatabaseSettings
|
|
{
|
|
ReadConcern = ReadConcern.Majority,
|
|
WriteConcern = WriteConcern.WMajority,
|
|
ReadPreference = ReadPreference.PrimaryPreferred,
|
|
};
|
|
|
|
var database = client.GetDatabase(options.GetDatabaseName(), settings);
|
|
var writeConcern = database.Settings.WriteConcern.With(wTimeout: options.CommandTimeout);
|
|
return database.WithWriteConcern(writeConcern);
|
|
});
|
|
|
|
services.AddSingleton<MongoBootstrapper>();
|
|
services.AddSingleton<IJobStore, MongoJobStore>();
|
|
services.AddSingleton<ILeaseStore, MongoLeaseStore>();
|
|
services.AddSingleton<ISourceStateRepository, MongoSourceStateRepository>();
|
|
services.AddSingleton<IDocumentStore, DocumentStore>();
|
|
services.AddSingleton<IDtoStore, DtoStore>();
|
|
services.AddSingleton<IAdvisoryStore, AdvisoryStore>();
|
|
services.AddSingleton<IChangeHistoryStore, MongoChangeHistoryStore>();
|
|
services.AddSingleton<IJpFlagStore, JpFlagStore>();
|
|
services.AddSingleton<IPsirtFlagStore, PsirtFlagStore>();
|
|
services.AddSingleton<IMergeEventStore, MergeEventStore>();
|
|
services.AddSingleton<IExportStateStore, ExportStateStore>();
|
|
services.TryAddSingleton<ExportStateManager>();
|
|
|
|
services.AddSingleton<IMongoCollection<JobRunDocument>>(static sp =>
|
|
{
|
|
var database = sp.GetRequiredService<IMongoDatabase>();
|
|
return database.GetCollection<JobRunDocument>(MongoStorageDefaults.Collections.Jobs);
|
|
});
|
|
|
|
services.AddSingleton<IMongoCollection<JobLeaseDocument>>(static sp =>
|
|
{
|
|
var database = sp.GetRequiredService<IMongoDatabase>();
|
|
return database.GetCollection<JobLeaseDocument>(MongoStorageDefaults.Collections.Locks);
|
|
});
|
|
|
|
services.AddHostedService<RawDocumentRetentionService>();
|
|
|
|
services.AddSingleton<MongoMigrationRunner>();
|
|
services.AddSingleton<IMongoMigration, EnsureDocumentExpiryIndexesMigration>();
|
|
services.AddSingleton<IMongoMigration, EnsureGridFsExpiryIndexesMigration>();
|
|
|
|
return services;
|
|
}
|
|
}
|