swcs/mixtape
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ability to use collection attribute on base interfaces

Browse Source
This commit is contained in:
Tobias Klika
2020-05-28 01:14:26 +02:00
parent ec1d409f3f
commit 65f7421ef4
4 changed files with 34 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zero.Core/Extensions/ServiceCollectionExtensions.cs
+5 -36
View File
@@ -1,4 +1,5 @@
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.DependencyInjection.Extensions;
using System;
using System.Collections.Generic;
using System.Linq;
@@ -53,11 +54,11 @@ namespace zero.Core.Extensions
/// <summary>
/// Adds or overrides an implementation
/// </summary>
public static void Replace<TService, TImplementation>(this IServiceCollection services)
public static void Replace<TService, TImplementation>(this IServiceCollection services, ServiceLifetime lifetime = ServiceLifetime.Transient)
where TService : class
where TImplementation : class, TService
{
services.Remove<TService>(out ServiceLifetime lifetime);
services.RemoveAll<TService>();
services.Add(new ServiceDescriptor(typeof(TService), typeof(TImplementation), lifetime));
}
@@ -65,44 +66,12 @@ namespace zero.Core.Extensions
/// <summary>
/// Adds or overrides an implementation
/// </summary>
public static void Replace<TService, TImplementation>(this IServiceCollection services, Func<IServiceProvider, TImplementation> implementationFactory)
public static void Replace<TService, TImplementation>(this IServiceCollection services, Func<IServiceProvider, TImplementation> implementationFactory, ServiceLifetime lifetime = ServiceLifetime.Transient)
where TService : class
where TImplementation : class, TService
{
services.Remove<TService>(out ServiceLifetime lifetime);
services.RemoveAll<TService>();
services.Add(new ServiceDescriptor(typeof(TService), implementationFactory, lifetime));
}
/// <summary>
/// Removes all instances of a registered service
/// </summary>
public static bool Remove<TService>(this IServiceCollection services) => services.Remove<TService>(out ServiceLifetime lifetime);
/// <summary>
/// Removes all instances of a registered service
/// </summary>
public static bool Remove<TService>(this IServiceCollection services, out ServiceLifetime lifetime)
{
Type serviceType = typeof(TService);
IEnumerable<ServiceDescriptor> descriptors = services.Where(x => x.ServiceType == serviceType);
if (descriptors.Any())
{
lifetime = descriptors.First().Lifetime;
foreach (ServiceDescriptor descriptor in descriptors)
{
services.Remove(descriptor);
}
}
else
{
lifetime = ServiceLifetime.Transient;
}
return descriptors.Any();
}
}
}