Merge pull request #1902 from PathogenDavid/issue-1901

Fix redundant assembly resolution resulting in duplicate assembly loading

Bonsai/Bonsai.csproj

@@ -102,6 +102,7 @@
       </_IlRepackCommand>
     </PropertyGroup>
     <Exec WorkingDirectory="$(RepackedOutputPath)" Command="$(_IlRepackCommand.Replace('%0D', ' ').Replace('%0A', ' '))" UseCommandProcessor="false" />
+    <Copy SourceFiles="$(TargetDir)NuGet.config" DestinationFolder="$(RepackedOutputPath)" />
   </Target>
 
   <Target Name="UseRepackOutputForPackage" Condition="'$(UseRepackForBootstrapperPackage)' == 'true'" AfterTargets="BuiltProjectOutputGroup" DependsOnTargets="ConfigureRepack">

Bonsai/SystemResourcesExtensionsSupport.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Concurrent;
 using System.Diagnostics;
 using System.Reflection;
 
@@ -16,19 +17,27 @@ internal static class SystemResourcesExtensionsSupport
         [Conditional("NETFRAMEWORK")]
         internal static void Initialize()
         {
+            // Assembly.Load(byte[]) always results in a new assembly so we need to manually de-duplicate them
+            ConcurrentDictionary<string, Assembly> resolveCache = new();
+
             AppDomain.CurrentDomain.AssemblyResolve += (_, args) =>
             {
                 var assemblyName = new AssemblyName(args.Name);
+
+                if (resolveCache.TryGetValue(assemblyName.Name, out var cachedResult))
+                    return cachedResult;
+
                 using var embeddedAssembly = typeof(SystemResourcesExtensionsSupport).Assembly.GetManifestResourceStream($"{nameof(Bonsai)}.{assemblyName.Name}.dll");
-                
+
                 if (embeddedAssembly is null)
-                    return null;
+                    return resolveCache.GetOrAdd(assemblyName.Name, (Assembly)null);
 
                 var assemblyBytes = new byte[embeddedAssembly.Length];
                 int readLength = embeddedAssembly.Read(assemblyBytes, 0, assemblyBytes.Length);
                 Debug.Assert(readLength == assemblyBytes.Length);
 
                 var result = Assembly.Load(assemblyBytes);
+                result = resolveCache.GetOrAdd(assemblyName.Name, result);
                 Debug.WriteLine($"Redirecting '{args.Name}' to embedded '{result.FullName}'");
                 return result;
             };