From 3d9a7e44382fd43713ba0eaddac64199a22a56f9 Mon Sep 17 00:00:00 2001
From: Daniel Clark <dan_clark@outlook.com>
Date: Fri, 9 Dec 2022 23:11:11 +0000
Subject: [PATCH] Fix initialization

---
 .../Lifecycles/Initialization.cs               |  6 +++++-
 .../Lifecycles/InitializeHandler.cs            | 18 ++++--------------
 2 files changed, 9 insertions(+), 15 deletions(-)

diff --git a/Toolkit.Foundation/Lifecycles/Initialization.cs b/Toolkit.Foundation/Lifecycles/Initialization.cs
index 0869d04..f7febce 100644
--- a/Toolkit.Foundation/Lifecycles/Initialization.cs
+++ b/Toolkit.Foundation/Lifecycles/Initialization.cs
@@ -1,4 +1,6 @@
-namespace Toolkit.Foundation;
+using System.Diagnostics;
+
+namespace Toolkit.Foundation;
 
 public class Initialization : IInitialization
 {
@@ -15,7 +17,9 @@ public class Initialization : IInitialization
         {
             if (initializer is not null)
             {
+                Trace.WriteLine(initializer.GetType());
                 await initializer.InitializeAsync();
+                Trace.WriteLine("Done");
             }
         }
     }
diff --git a/Toolkit.Foundation/Lifecycles/InitializeHandler.cs b/Toolkit.Foundation/Lifecycles/InitializeHandler.cs
index 3f4348d..e5dbef3 100644
--- a/Toolkit.Foundation/Lifecycles/InitializeHandler.cs
+++ b/Toolkit.Foundation/Lifecycles/InitializeHandler.cs
@@ -1,29 +1,19 @@
 using Mediator;
-using System.Diagnostics;
 
 namespace Toolkit.Foundation;
 
 public class InitializeHandler : IRequestHandler<Initialize>
 {
-    private readonly IEnumerable<IInitializable?> initializers;
+    private readonly IInitialization initialization;
 
-    public InitializeHandler(IEnumerable<IInitializable?> initializers)
+    public InitializeHandler(IInitialization initialization)
     {
-        this.initializers = initializers;
+        this.initialization = initialization;
     }
 
     public async ValueTask<Unit> Handle(Initialize request, CancellationToken cancellationToken)
     {
-        foreach (IInitializable? initializer in initializers)
-        {
-            if (initializer is not null)
-            {
-                Trace.WriteLine(initializer.GetType());
-                await initializer.InitializeAsync();
-                Trace.WriteLine("Done");
-            }
-        }
-
+        await initialization.InitializeAsync();
         return default;
     }
 }