Merge pull request #4 from AlonTalmi/SupportForValueTypesFuncRegistra…

…tion Support for Value Types delegate registration
AlonTalmi · Nov 14, 2023 · ab184ad · ab184ad
2 parents 8a1d59d + a4c069c
commit ab184ad
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/VContainer/Assets/VContainer/Runtime/ContainerBuilderExtensions.cs b/VContainer/Assets/VContainer/Runtime/ContainerBuilderExtensions.cs
@@ -84,17 +84,15 @@ public static RegistrationBuilder Register<TInterface>(
             this IContainerBuilder builder,
             Func<IObjectResolver, TInterface> implementationConfiguration,
             Lifetime lifetime)
-            where TInterface : class
-            => builder.Register(new FuncRegistrationBuilder(implementationConfiguration, typeof(TInterface), lifetime));
+            => builder.Register(new FuncRegistrationBuilder(resolver => implementationConfiguration(resolver), typeof(TInterface), lifetime)); 
 
         [MethodImpl(MethodImplOptions.AggressiveInlining)]
         public static RegistrationBuilder Register<TInterface>(
             this IContainerBuilder builder,
             Func<IObjectResolver, TInterface> implementationConfiguration,
             Lifetime lifetime,
             Action<TInterface, IObjectResolver> callback)
-            where TInterface : class
-            => builder.Register(new FuncRegistrationBuilderWithCallback(implementationConfiguration, typeof(TInterface), lifetime, (instance, resolver) => callback((TInterface)instance, resolver)));
+            => builder.Register(new FuncRegistrationBuilderWithCallback(resolver => implementationConfiguration(resolver), typeof(TInterface), lifetime, (instance, resolver) => callback((TInterface)instance, resolver)));
 
         [MethodImpl(MethodImplOptions.AggressiveInlining)]
         public static RegistrationBuilder RegisterInstance<TInterface>(