-
-
Notifications
You must be signed in to change notification settings - Fork 119
/
HostBuilderExtensions.cs
103 lines (84 loc) · 3.92 KB
/
HostBuilderExtensions.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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
using System;
using System.Collections.Generic;
using LamarCodeGeneration.Util;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
namespace Lamar.Microsoft.DependencyInjection
{
public static class HostBuilderExtensions
{
/// <summary>
/// Apply Lamar service overrides regardless of the order of .Net service registrations. This is primarily
/// meant for test automation scenarios
/// </summary>
/// <param name="builder"></param>
/// <param name="overrides"></param>
/// <returns></returns>
public static IHostBuilder OverrideServices(this IHostBuilder builder, Action<ServiceRegistry> overrides)
{
return builder.ConfigureServices(x => x.OverrideServices(overrides));
}
/// <summary>
/// Shortcut to replace the built in DI container with Lamar using service registrations
/// dependent upon the application's environment and configuration.
/// </summary>
/// <param name="builder"></param>
/// <param name="registry"></param>
/// <returns></returns>
public static IHostBuilder UseLamar(this IHostBuilder builder, Action<HostBuilderContext, ServiceRegistry> configure = null)
{
return builder
.UseServiceProviderFactory<ServiceRegistry>(new LamarServiceProviderFactory())
.UseServiceProviderFactory<IServiceCollection>(new LamarServiceProviderFactory())
.ConfigureServices((context, services) =>
{
var registry = new ServiceRegistry(services);
configure?.Invoke(context, registry);
services.Clear();
services.AddRange(registry);
#if NET6_0_OR_GREATER
services.AddSingleton<IServiceProviderIsService>(s => (IServiceProviderIsService) s.GetRequiredService<IContainer>());
#endif
});
}
/// <summary>
/// Shortcut to replace the built in DI container with Lamar using service registrations
/// dependent upon the application's environment and configuration.
/// </summary>
/// <param name="builder"></param>
/// <param name="registry"></param>
/// <returns></returns>
public static IHostBuilder UseLamar(this IHostBuilder builder, Action<ServiceRegistry> configure)
{
return builder.UseLamar((c, s) => configure.Invoke(s));
}
/// <summary>
/// Overrides the internal DI container with Lamar, optionally using a Lamar ServiceRegistry
/// for additional service registrations
/// </summary>
/// <param name="services"></param>
/// <param name="registry"></param>
/// <returns></returns>
public static IServiceCollection AddLamar(this IServiceCollection services, ServiceRegistry registry = null)
{
services.AddSingleton<ILoggerFactory, LoggerFactory>(sp =>
new LoggerFactory(
sp.GetRequiredService<IEnumerable<ILoggerProvider>>(),
sp.GetRequiredService<IOptionsMonitor<LoggerFilterOptions>>()
));
services.AddSingleton<IServiceProviderFactory<ServiceRegistry>, LamarServiceProviderFactory>();
services.AddSingleton<IServiceProviderFactory<IServiceCollection>, LamarServiceProviderFactory>();
#if NET6_0_OR_GREATER
services.AddSingleton<IServiceProviderIsService>(s => (IServiceProviderIsService) s.GetRequiredService<IContainer>());
#endif
registry ??= new ServiceRegistry();
foreach (var descriptor in registry)
{
services.Add(descriptor);
}
return services;
}
}
}