Last active
August 16, 2024 14:36
-
-
Save ReallyLiri/c669c60db2109554d5ce47e03613a7a9 to your computer and use it in GitHub Desktop.
ASP.NET dependency injection extensions
This file contains hidden or 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 System.Reflection; | |
using Microsoft.AspNetCore.Routing; | |
using Microsoft.Extensions.DependencyInjection; | |
namespace ServiceCollectionExtensions | |
{ | |
public static class ServiceCollectionExtensions | |
{ | |
public static ConstructorInfo GetConstructor(this Type type) | |
{ | |
var constructors = type.GetConstructors(); | |
if (constructors.Length != 1) | |
{ | |
throw new ArgumentException($"{type} doesn't have exactly one constructor"); | |
} | |
var constructor = constructors[0]; | |
return constructor; | |
} | |
public static IReadOnlyDictionary<string, object> ToPropertiesDictionary(this object obj) | |
=> new RouteValueDictionary(obj); | |
public static void AddSingletonWithConstructorParams<TService, TImplementation>( | |
this IServiceCollection services, | |
object paramsWithNames | |
) where TImplementation : class, TService where TService : class | |
=> services.AddSingletonWithConstructorParams( | |
typeof(TService), | |
typeof(TImplementation), | |
paramsWithNames | |
); | |
public static void AddSingletonWithConstructorParams( | |
this IServiceCollection services, | |
Type serviceType, | |
Type implementationType, | |
object paramsWithNames | |
) | |
{ | |
if (paramsWithNames == null) | |
{ | |
throw new ArgumentException("null argument"); | |
} | |
if (!paramsWithNames.GetType() | |
.Name.Contains("AnonymousType")) | |
{ | |
// due to ambiguous invocation of this method with a single arg | |
services.AddSingletonWithConstructorParams( | |
serviceType, | |
implementationType, | |
new[] {paramsWithNames} | |
); | |
return; | |
} | |
var constructor = implementationType.GetConstructor(); | |
var overridingParams = paramsWithNames.ToPropertiesDictionary(); | |
foreach (var parameterInfo in constructor.GetParameters()) | |
{ | |
if (overridingParams.ContainsKey(parameterInfo.Name)) | |
{ | |
var paramValue = overridingParams[parameterInfo.Name]; | |
if (paramValue != null && !parameterInfo.ParameterType.IsInstanceOfType(paramValue)) | |
{ | |
throw new ArgumentException($"Argument of type {paramValue.GetType()} cannot be supplied as parameter named {parameterInfo.Name} for type {implementationType}"); | |
} | |
} | |
} | |
AddSingletonWithParamsCallback( | |
services, | |
serviceType, | |
constructor, | |
parameterInfo => overridingParams.ContainsKey(parameterInfo.Name), | |
parameterInfo => overridingParams[parameterInfo.Name] | |
); | |
} | |
public static void AddSingletonWithConstructorParams<TService, TImplementation>( | |
this IServiceCollection services, | |
params object[] parameters | |
) where TImplementation : class, TService where TService : class | |
=> services.AddSingletonWithConstructorParams( | |
typeof(TService), | |
typeof(TImplementation), | |
parameters | |
); | |
public static void AddSingletonWithConstructorParams( | |
this IServiceCollection services, | |
Type serviceType, | |
Type implementationType, | |
params object[] parameters | |
) | |
{ | |
var constructor = implementationType.GetConstructor(); | |
if (parameters.Any(param => param == null)) | |
{ | |
throw new ArgumentException($"Cannot supply null ctor params to {nameof(AddSingletonWithConstructorParams)}"); | |
} | |
var overridingValuesByParameterName = ConstructOverridingValuesByParameterName(constructor, parameters); | |
services.AddSingletonWithParamsCallback( | |
serviceType, | |
constructor, | |
parameterInfo => overridingValuesByParameterName.ContainsKey(parameterInfo.Name), | |
parameterInfo => overridingValuesByParameterName[parameterInfo.Name] | |
); | |
} | |
private static Dictionary<string, object> ConstructOverridingValuesByParameterName(ConstructorInfo constructor, params object[] parameters) | |
{ | |
var overridingValuesByParameterName = new Dictionary<string, object>(); | |
foreach (var parameterInfo in constructor.GetParameters()) | |
{ | |
var matchingOverridingParameters = parameters.Where(param => parameterInfo.ParameterType.IsInstanceOfType(param)) | |
.ToList(); | |
if (!matchingOverridingParameters.Any()) | |
{ | |
continue; | |
} | |
if (matchingOverridingParameters.Count > 1) | |
{ | |
throw new ArgumentException($"More than one of the input parameters matches the ctor param {parameterInfo.Name} of type {parameterInfo.ParameterType}"); | |
} | |
overridingValuesByParameterName[parameterInfo.Name] = matchingOverridingParameters[0]; | |
} | |
return overridingValuesByParameterName; | |
} | |
private static void AddSingletonWithParamsCallback( | |
this IServiceCollection services, | |
Type serviceType, | |
ConstructorInfo constructor, | |
Predicate<ParameterInfo> parameterPredicate, | |
Func<ParameterInfo, object> parameterValueProvider | |
) | |
=> services.AddSingleton( | |
serviceType, | |
serviceProvider => InvokeConstructor( | |
constructor, | |
parameterPredicate, | |
parameterValueProvider, | |
serviceProvider | |
) | |
); | |
private static object InvokeConstructor( | |
ConstructorInfo constructor, | |
Predicate<ParameterInfo> parameterPredicate, | |
Func<ParameterInfo, object> parameterValueProvider, | |
IServiceProvider serviceProvider | |
) | |
=> constructor.Invoke( | |
constructor.GetParameters() | |
.Select( | |
parameterInfo => parameterPredicate(parameterInfo) | |
? parameterValueProvider(parameterInfo) | |
: serviceProvider.GetService(parameterInfo.ParameterType) | |
) | |
.ToArray() | |
); | |
public static TService GetRequiredServiceWithConstructorParams<TService, TImplementation>(this IServiceProvider serviceProvider, params object[] parameters) where TImplementation : TService where TService : class | |
{ | |
var constructor = typeof(TImplementation).GetConstructor(); | |
var overridingValuesByParameterName = ConstructOverridingValuesByParameterName(constructor, parameters); | |
return InvokeConstructor( | |
constructor, | |
parameterInfo => overridingValuesByParameterName.ContainsKey(parameterInfo.Name), | |
parameterInfo => overridingValuesByParameterName[parameterInfo.Name], | |
serviceProvider | |
) as TService; | |
} | |
public static void AddBuilder<TService, TImplementation, TArg>(this IServiceCollection serviceCollection) where TImplementation : class, TService where TService : class | |
=> serviceCollection.AddSingleton( | |
serviceProvider => | |
{ | |
return new Func<TArg, TService>( | |
arg => serviceProvider.GetRequiredServiceWithConstructorParams<TService, TImplementation>(arg) | |
); | |
} | |
); | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Thanks!
AddBuilder
can be used when your service requires some dynamic or unknown value at registration time (i.e read from outside source, or recreated for each request with different config).If you have a service
MyService : IMyService
, that accepts some interfaces in its constructor, along somelong timeout
, you could registerserviceCollection.AddBuilder<IMyService, MyService, long>()
and then consume it usingFunc<long, IMyService
, which will create a new instance on each invocation, allowing you to inject all the non-long
constructor parameters.This is just a shortcut to registering the
Func<>
on your own like I referenced above to @Ewerton