Merge pull request #700 from MatthieuG9/patch-1

Add an overload method RegisterEntryPoint
hadashiA · Aug 14, 2024 · e7f2955 · e7f2955
2 parents 75da356 + 8de6f2c
commit e7f2955
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/VContainer/Assets/VContainer/Runtime/Unity/ContainerBuilderUnityExtensions.cs b/VContainer/Assets/VContainer/Runtime/Unity/ContainerBuilderUnityExtensions.cs
@@ -106,6 +106,16 @@ public static RegistrationBuilder RegisterEntryPoint<T>(
             return builder.Register<T>(lifetime).AsImplementedInterfaces();
         }
 
+        public static RegistrationBuilder RegisterEntryPoint<TInterface>(
+            this IContainerBuilder builder,
+            Func<IObjectResolver, TInterface> implementationConfiguration,
+            Lifetime lifetime)
+        {
+            EntryPointsBuilder.EnsureDispatcherRegistered(builder);
+            return builder.Register(new FuncRegistrationBuilder(container => implementationConfiguration(container),
+                typeof(TInterface), lifetime)).AsImplementedInterfaces();
+        }
+
         public static void RegisterEntryPointExceptionHandler(
             this IContainerBuilder builder,
             Action<Exception> exceptionHandler)