From d8ac6f1d27a75f5f1261fb9b58954be8f49de8fd Mon Sep 17 00:00:00 2001
From: vby <vby@nemetos.com>
Date: Wed, 25 Dec 2024 23:24:36 +0100
Subject: [PATCH] Added keyed services support

---
 .../ServiceCollectionExtensions.Decoration.cs   | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/Scrutor/ServiceCollectionExtensions.Decoration.cs b/src/Scrutor/ServiceCollectionExtensions.Decoration.cs
index 189ffad7..3905d606 100644
--- a/src/Scrutor/ServiceCollectionExtensions.Decoration.cs
+++ b/src/Scrutor/ServiceCollectionExtensions.Decoration.cs
@@ -25,6 +25,23 @@ public static IServiceCollection Decorate<TService, TDecorator>(this IServiceCol
         return services.Decorate(typeof(TService), typeof(TDecorator));
     }
 
+    /// <summary>
+    /// Decorates all registered services of type <typeparamref name="TService"/>
+    /// using the specified type <typeparamref name="TDecorator"/>.
+    /// </summary>
+    /// <param name="services">The services to add to.</param>
+    /// <param name="serviceKey">The <see cref="ServiceDescriptor.ServiceKey"/> of the service.</param>
+    /// <exception cref="DecorationException">If no service of the type <typeparamref name="TService"/> has been registered.</exception>
+    /// <exception cref="ArgumentNullException">If the <paramref name="services"/> argument is <c>null</c>.</exception>
+    public static IServiceCollection DecorateKeyed<TService, TDecorator>(this IServiceCollection services, string serviceKey)
+        where TDecorator : TService
+    {
+        Preconditions.NotNull(services, nameof(services));
+        Preconditions.NotNull(serviceKey, nameof(serviceKey));
+
+        return services.Decorate(DecorationStrategy.WithType(typeof(TService), serviceKey, typeof(TDecorator)));
+    }
+
     /// <summary>
     /// Decorates all registered services of type <typeparamref name="TService"/>
     /// using the specified type <typeparamref name="TDecorator"/>.