Created
May 2, 2018 18:40
-
-
Save bradyclifford/4836d090355978ea9c37cbd80e16878b to your computer and use it in GitHub Desktop.
.Net Core Service Collection
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
using System; | |
using System.Collections.Generic; | |
using System.Linq; | |
using Microsoft.Extensions.Configuration; | |
using Microsoft.Extensions.DependencyInjection; | |
using Microsoft.Extensions.Options; | |
namespace Web.Extensions | |
{ | |
public static class ServiceCollectionExtensions | |
{ | |
public static IServiceCollection ConfigureSettings<TSettings>(this IServiceCollection services, | |
IConfiguration configuration) where TSettings : class, new() | |
{ | |
return services.ConfigureSettings(configuration, | |
provider => provider.GetRequiredService<IOptions<TSettings>>().Value); | |
} | |
public static IServiceCollection ConfigureSettings<TSettings>(this IServiceCollection services, | |
IConfiguration configuration, Func<IServiceProvider, TSettings> implementationFactory) | |
where TSettings : class, new() | |
{ | |
var settingsName = typeof(TSettings).Name; | |
services.Configure<TSettings>(configuration.GetSection(settingsName)); | |
services.AddSingleton(implementationFactory); | |
return services; | |
} | |
public static IServiceCollection Replace<TService, TImplementation>(this IServiceCollection services) | |
where TImplementation : TService | |
{ | |
return services.Replace<TService>(typeof(TImplementation)); | |
} | |
public static IServiceCollection Replace<TService>(this IServiceCollection services, Type implementationType) | |
{ | |
return services.Replace(typeof(TService), implementationType); | |
} | |
public static IServiceCollection Replace(this IServiceCollection services, Type serviceType, | |
Type implementationType) | |
{ | |
if (services == null) | |
{ | |
throw new ArgumentNullException(nameof(services)); | |
} | |
if (serviceType == null) | |
{ | |
throw new ArgumentNullException(nameof(serviceType)); | |
} | |
if (implementationType == null) | |
{ | |
throw new ArgumentNullException(nameof(implementationType)); | |
} | |
if (!services.TryGetDescriptors(serviceType, out var descriptors)) | |
{ | |
throw new ArgumentException($"No services found for {serviceType.FullName}.", nameof(serviceType)); | |
} | |
foreach (var descriptor in descriptors) | |
{ | |
var index = services.IndexOf(descriptor); | |
services.Insert(index, descriptor.WithImplementationType(implementationType)); | |
services.Remove(descriptor); | |
} | |
return services; | |
} | |
private static bool TryGetDescriptors(this IServiceCollection services, Type serviceType, | |
out ICollection<ServiceDescriptor> descriptors) | |
{ | |
return (descriptors = services.Where(service => service.ServiceType == serviceType).ToArray()).Any(); | |
} | |
private static ServiceDescriptor WithImplementationType(this ServiceDescriptor descriptor, | |
Type implementationType) | |
{ | |
return new ServiceDescriptor(descriptor.ServiceType, implementationType, descriptor.Lifetime); | |
} | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment