fixing PreserveAttribute to ensure container is loaded.

dansiegel · Dec 26, 2019 · 181a763 · 181a763
1 parent d3a4385
commit 181a763
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/src/Prism.Container.Extensions/ContainerLocator.cs b/src/Prism.Container.Extensions/ContainerLocator.cs
@@ -29,6 +29,19 @@ public static IContainerExtension Locate()
             return _locatedExtension.Container;
         }
 
+        internal static void LocatePreservedReference(Type type)
+        {
+            if (_locatedExtension is null)
+            {
+                if (!LocatedAssembly(new[] { type.Assembly }, out _locatedExtension))
+                {
+                    throw new DllNotFoundException("No assembly containing a reference to an IContainerExtension implementation could be found. You must have a reference to one of the DI Container Extensions in your final project");
+                }
+
+                Console.WriteLine($"Located Container: {_locatedExtension.Container.GetType().FullName}");
+            }
+        }
+
         private static bool LocatedAssembly(IEnumerable<Assembly> assemblies, out ContainerExtensionAttribute attribute)
         {
             attribute = null;

diff --git a/src/Prism.Container.Extensions/PreserveAttribute.cs b/src/Prism.Container.Extensions/PreserveAttribute.cs
@@ -1,5 +1,6 @@
 using System;
 using System.ComponentModel;
+using Prism.Ioc;
 
 namespace Prism.Container.Extensions
 {
@@ -36,6 +37,10 @@ public PreserveAttribute()
         public PreserveAttribute(Type referenceType)
             : this(true, false)
         {
+            if(typeof(IContainerExtension).IsAssignableFrom(referenceType))
+            {
+                ContainerLocator.LocatePreservedReference(referenceType);
+            }
         }
     }
 }