Add WithMetadata with Func<Type, IEnumerable<KeyValuePair<string, obj…

…ect?> for OpenGenericAssemblyScanning
autofac · Dec 9, 2021 · 2b1acdb · 2b1acdb
1 parent a6a4d8d
commit 2b1acdb
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/src/Autofac/RegistrationExtensions.OpenGenericAssemblyScanning.cs b/src/Autofac/RegistrationExtensions.OpenGenericAssemblyScanning.cs
@@ -298,5 +298,27 @@ public static IRegistrationBuilder<TLimit, OpenGenericScanningActivatorData, Dyn
         {
             return registration.As(t => t.GetOpenGenericImplementedInterfaces());
         }
+
+        /// <summary>
+        /// Specify how a type from a scanned assembly provides metadata.
+        /// </summary>
+        /// <typeparam name="TLimit">Registration limit type.</typeparam>
+        /// <typeparam name="TRegistrationStyle">Registration style.</typeparam>
+        /// <param name="registration">Registration to set metadata on.</param>
+        /// <param name="metadataMapping">A function mapping the type to a list of metadata items.</param>
+        /// <returns>Registration builder allowing the registration to be configured.</returns>
+        public static IRegistrationBuilder<TLimit, OpenGenericScanningActivatorData, TRegistrationStyle>
+            WithMetadata<TLimit, TRegistrationStyle>(
+                this IRegistrationBuilder<TLimit, OpenGenericScanningActivatorData, TRegistrationStyle> registration,
+                Func<Type, IEnumerable<KeyValuePair<string, object?>>> metadataMapping)
+        {
+            if (registration == null)
+            {
+                throw new ArgumentNullException(nameof(registration));
+            }
+
+            registration.ActivatorData.ConfigurationActions.Add((t, rb) => rb.WithMetadata(metadataMapping(t)));
+            return registration;
+        }
     }
 }