Merge pull request #25 from yfital/interception_works_once

Fix multiple intreceptors issue due to .net core logic change

Author: ENikS

src/Interceptors/InstanceInterceptors/InterfaceInterception/InterfaceInterceptorClassGenerator.Desktop.cs

@@ -176,7 +176,7 @@ private void CreateTypeBuilder()
         {
             TypeAttributes newAttributes = TypeAttributes.Public | TypeAttributes.Class;
 
-            ModuleBuilder moduleBuilder = GetModuleBuilder();
+            ModuleBuilder moduleBuilder = InterceptorClassGenerator.CreateModuleBuilder(AssemblyBuilder);
             _typeBuilder = moduleBuilder.DefineType(CreateTypeName(), newAttributes);
 
             _mainInterfaceMapper = DefineGenericArguments();

src/Interceptors/InstanceInterceptors/InterfaceInterception/InterfaceInterceptorClassGenerator.cs

@@ -0,0 +1,25 @@
+using System;
+using System.Collections.Concurrent;
+using System.Reflection.Emit;
+
+namespace Unity.Interception.Interceptors
+{
+    public static class InterceptorClassGenerator
+    {
+        private static readonly ConcurrentDictionary<AssemblyBuilder, ModuleBuilder> ModuleByAssembly =
+            new ConcurrentDictionary<AssemblyBuilder, ModuleBuilder>();
+
+        public static ModuleBuilder CreateModuleBuilder(AssemblyBuilder assemblyBuilder)
+        {
+            return ModuleByAssembly.GetOrAdd(assemblyBuilder, assembly =>
+            {
+                string moduleName = Guid.NewGuid().ToString("N");
+#if DEBUG_SAVE_GENERATED_ASSEMBLY
+            return AssemblyBuilder.DefineDynamicModule(moduleName, moduleName + ".dll", true);
+#else
+                return assembly.DefineDynamicModule(moduleName);
+#endif
+            });
+        }
+    }
+}

src/Interceptors/InterceptorClassGenerator.Desktop.cs

@@ -214,7 +214,7 @@ private void CreateTypeBuilder()
 
             Type baseClass = GetGenericType(_typeToIntercept);
 
-            ModuleBuilder moduleBuilder = InterceptingClassGenerator.GetModuleBuilder();
+            ModuleBuilder moduleBuilder = InterceptorClassGenerator.CreateModuleBuilder(AssemblyBuilder);
             _typeBuilder = moduleBuilder.DefineType(
                 "DynamicModule.ns.Wrapped_" + _typeToIntercept.Name + "_" + Guid.NewGuid().ToString("N"),
                 newAttributes,