feat(#75): Add QueueInitializeAsync to ManualDi.Async

Initialization was missing async queueing functionality

Author: PereViader

ManualDi.Async/ManualDi.Async.Tests/TestDiContainerBindings.cs

@@ -37,7 +37,7 @@ public async Task TestQueueInject()
     }
 
     [Test]
-    public async Task TestQueueInitialization()
+    public async Task TestQueueInitialize()
     {
         var initializationDelegate = Substitute.For<ContainerDelegate>();
 
@@ -48,6 +48,18 @@ public async Task TestQueueInitialization()
         initializationDelegate.Received(1).Invoke(Arg.Is<IDiContainer>(container));
     }
 
+    [Test]
+    public async Task TestQueueInitializeAsync()
+    {
+        var initializationDelegate = Substitute.For<AsyncContainerDelegate>();
+
+        await using var container = await new DiContainerBindings()
+            .Install(b => b.QueueInitializeAsync(initializationDelegate))
+            .Build(CancellationToken.None);
+
+        await initializationDelegate.Received(1).Invoke(Arg.Is<IDiContainer>(container), Arg.Any<CancellationToken>());
+    }
+    
     [Test]
     public async Task TestQueueStartup()
     {

ManualDi.Async/ManualDi.Async/Building/DiContainerBindings.cs

@@ -12,7 +12,7 @@ public sealed class DiContainerBindings
     {
         internal readonly Dictionary<IntPtr, Binding> bindingsByType;
         private readonly List<ContainerDelegate> injectDelegates;
-        private readonly List<ContainerDelegate> initializeDelegates;
+        private readonly List<object> initializeDelegates;
         private readonly List<object> startupDelegates;
         private readonly List<object> disposables;
 
@@ -74,6 +74,11 @@ public void QueueInitialize(ContainerDelegate containerDelegate)
             initializeDelegates.Add(containerDelegate);
         }
 
+        public void QueueInitializeAsync(AsyncContainerDelegate asyncContainerDelegate)
+        {
+            initializeDelegates.Add(asyncContainerDelegate);
+        }
+        
         [Obsolete("Use QueueInitialize instead")]
         public void QueueInitialization(ContainerDelegate containerDelegate)
         {
@@ -178,7 +183,17 @@ internal Binding<TApparent, TConcrete> BindAsSubContainer<TApparent, TConcrete>(
 
                     foreach (var initializationDelegate in initializeDelegates)
                     {
-                        initializationDelegate.Invoke(subContainer);
+                        switch (initializationDelegate)
+                        {
+                            case AsyncContainerDelegate asyncInitializationDelegate:
+                                await asyncInitializationDelegate.Invoke(subContainer, ct);
+                                break;
+                            case ContainerDelegate syncInitializationDelegate:
+                                syncInitializationDelegate.Invoke(subContainer);
+                                break;
+                            default:
+                                throw new InvalidOperationException($"Unexpected initialization delegate type: {initializationDelegate.GetType()}");
+                        }
                     }
 
                     foreach (var startupDelegate in startupDelegates)
@@ -218,9 +233,19 @@ public async ValueTask<DiContainer> Build(CancellationToken ct)
                     injectDelegate.Invoke(diContainer);
                 }
 
-                foreach (var initializeDelegate in initializeDelegates)
+                foreach (var initializationDelegate in initializeDelegates)
                 {
-                    initializeDelegate.Invoke(diContainer);
+                    switch (initializationDelegate)
+                    {
+                        case AsyncContainerDelegate asyncInitializationDelegate:
+                            await asyncInitializationDelegate.Invoke(diContainer, ct);
+                            break;
+                        case ContainerDelegate syncInitializationDelegate:
+                            syncInitializationDelegate.Invoke(diContainer);
+                            break;
+                        default:
+                            throw new InvalidOperationException($"Unexpected initialization delegate type: {initializationDelegate.GetType()}");
+                    }
                 }
 
                 foreach (var startupDelegate in startupDelegates)