/
Config.cs
43 lines (38 loc) · 1.36 KB
/
Config.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
using Core.Aggregates;
using Core.OpenTelemetry;
using Core.OptimisticConcurrency;
using Marten;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
namespace Core.Marten.Repository;
public static class Config
{
public static IServiceCollection AddMartenRepository<T>(
this IServiceCollection services,
bool withAppendScope = true,
bool withTelemetry = true
) where T : class, IAggregate
{
services.AddScoped<IMartenRepository<T>, MartenRepository<T>>();
if (withAppendScope)
services.Decorate<IMartenRepository<T>>(
(inner, sp) => new MartenRepositoryWithETagDecorator<T>(
inner,
sp.GetRequiredService<IExpectedResourceVersionProvider>(),
sp.GetRequiredService<INextResourceVersionProvider>()
)
);
if (withTelemetry)
{
services.Decorate<IMartenRepository<T>>(
(inner, sp) => new MartenRepositoryWithTracingDecorator<T>(
inner,
sp.GetRequiredService<IDocumentSession>(),
sp.GetRequiredService<IActivityScope>(),
sp.GetRequiredService<ILogger<MartenRepositoryWithTracingDecorator<T>>>()
)
);
}
return services;
}
}