Update FeatureEndpointExtensions.cs

loresoft · Dec 11, 2024 · b32cac3 · b32cac3
1 parent a7537f7
commit b32cac3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/MediatR.CommandQuery.Endpoints/FeatureEndpointExtensions.cs b/src/MediatR.CommandQuery.Endpoints/FeatureEndpointExtensions.cs
@@ -13,11 +13,14 @@ public static IServiceCollection AddFeatureEndpoints(this IServiceCollection ser
         return services;
     }
 
-    public static IEndpointConventionBuilder MapFeatureEndpoints(this IEndpointRouteBuilder builder, string prefix = "/api")
+    public static IEndpointConventionBuilder MapFeatureEndpoints(this IEndpointRouteBuilder builder, string prefix = "/api", string? serviceKey = null)
     {
         var featureGroup = builder.MapGroup(prefix);
 
-        var features = builder.ServiceProvider.GetServices<IFeatureEndpoint>();
+        var features = string.IsNullOrEmpty(serviceKey)
+            ? builder.ServiceProvider.GetServices<IFeatureEndpoint>()
+            : builder.ServiceProvider.GetKeyedServices<IFeatureEndpoint>(serviceKey);
+
         foreach (var feature in features)
             feature.AddRoutes(featureGroup);