Reduce memory and cpu consumption of types registration in tests

Author: alex-kulakov

Orm/Xtensive.Orm.Tests.Framework/AssemblyExtensions.cs

@@ -4,16 +4,101 @@
 // Created by: Alexey Kulakov
 // Created:    2018.08.31
 
+using System;
+using System.Collections.Concurrent;
+using System.Collections.Generic;
 using System.Configuration;
+using System.Linq;
 using System.Reflection;
+using System.Runtime.CompilerServices;
 
 namespace Xtensive.Orm.Tests
 {
   public static class AssemblyExtensions
   {
+    private static readonly Type ObjectType = typeof(object);
+    private static readonly ConcurrentDictionary<Assembly, Type[]> TypesPerAssembly = new();
+
     public static System.Configuration.Configuration GetAssemblyConfiguration(this Assembly assembly)
     {
       return ConfigurationManager.OpenExeConfiguration(assembly.Location);
     }
+
+    public static IReadOnlyList<Type> GetTypesFromNamespaceCaching(this Assembly assembly, string @namespace)
+    {
+      // these two dummy mentions to not forget to sync filtration algorithm here and in the classes,
+      // in particular BaseType property, if the property changed then this algorighm should be changed as well
+      _ = nameof(Xtensive.IoC.ServiceTypeRegistrationProcessor.BaseType);
+      _ = nameof(Xtensive.Orm.Configuration.DomainTypeRegistrationHandler.BaseType);
+
+      var assemblyTypes = TypesPerAssembly.GetOrAdd(assembly, static (a) => {
+        var allTypes = a.GetTypes();
+        var list = new List<Type>(allTypes.Length);
+        foreach (var t in allTypes) {
+          // we ignore compiler generated types because usuallty they are
+          // at the end of sorted types
+          if (t.IsSubclassOf(ObjectType) && t.GetCustomAttribute<CompilerGeneratedAttribute>()==null)
+            list.Add(t);
+        }
+        return list.ToArray();
+      });
+
+      var range = FindRange(assemblyTypes, @namespace);
+      return new ArraySegment<Type>(assemblyTypes, range.first, range.last - range.first + 1);
+      
+
+      //type.IsSubclassOf(BaseType) && (ns.IsNullOrEmpty() || (type.FullName.IndexOf(ns + ".", StringComparison.InvariantCulture) >= 0));
+    }
+
+    private static (int first, int last) FindRange(Type[] types, string ns)
+    {
+      const int windowSize = 10;
+
+      // we rely on the fact that types are sorted by full name, that means types of same namespace are go one by one
+      // which gives us to optimize search - we find first type that has desired namespace, then we try to find last one
+      // and then we return the part of original array as result
+      var firstHit = -1;
+      var lastHit = -1;
+
+      for (int headIndex = 0, count = types.Length ; headIndex < count; headIndex++) {
+        var head = types[headIndex];
+        if (head.FullName.IndexOf(ns + ".", StringComparison.InvariantCulture) >= 0) {
+          firstHit = headIndex;
+          break;
+        }
+      }
+
+      var isOutOfRange = false;
+      do {
+        lastHit = firstHit + windowSize;
+        if (lastHit > types.Length - 1) {
+          lastHit = types.Length - 1;
+        }
+        var tail = types[lastHit];
+        if (tail.FullName.IndexOf(ns + ".", StringComparison.InvariantCulture) < 0)
+          isOutOfRange = true;
+      }
+      while (!isOutOfRange);
+
+      for (int tailIndex = lastHit; tailIndex > firstHit; tailIndex--) {
+        var tail = types[tailIndex];
+        if (tail.FullName.IndexOf(ns + ".", StringComparison.InvariantCulture) >= 0) {
+          lastHit = tailIndex;
+          break;
+        }
+      }
+
+      return (firstHit, lastHit);
+    }
+  }
+
+  public static class TypeRegistryExtensions
+  {
+    public static void RegisterCaching(this Xtensive.Orm.Configuration.DomainTypeRegistry _this, Assembly assembly, string @namespace)
+    {
+      foreach (var t in assembly.GetTypesFromNamespaceCaching(@namespace)) {
+        _this.Register(t);
+      }
+    }
   }
 }

Orm/Xtensive.Orm.Tests/Interfaces/DoubleIndexedInterfaceTest.cs

@@ -1,4 +1,4 @@
-// Copyright (C) 2011 Xtensive LLC.
+// Copyright (C) 2011 Xtensive LLC.
 // All rights reserved.
 // For conditions of distribution and use, see license.
 // Created by: Denis Krjuchkov
@@ -59,7 +59,7 @@ public class DoubleIndexedInterfaceTest : AutoBuildTest
     protected override DomainConfiguration BuildConfiguration()
     {
       var config = base.BuildConfiguration();
-      config.Types.Register(typeof (ITest).Assembly, typeof (ITest).Namespace);
+      config.Types.RegisterCaching(typeof (ITest).Assembly, typeof (ITest).Namespace);
       return config;
     }
 

Orm/Xtensive.Orm.Tests/Interfaces/InterfaceImplementationTest.cs

@@ -70,7 +70,7 @@ public class InterfaceImplementationTest
     public void CompsiteIndexTest()
     {
       var config = DomainConfigurationFactory.Create();
-      config.Types.Register(typeof(IHasName).Assembly, typeof(IHasName).Namespace);
+      config.Types.RegisterCaching(typeof(IHasName).Assembly, typeof(IHasName).Namespace);
       var domain = GetClassTableDomain(config);
       using (var session = domain.OpenSession())
       using (var t = session.OpenTransaction()) {
@@ -99,7 +99,7 @@ public void CombinedTest()
     public void InternalTest(Func<DomainConfiguration, Domain> generator)
     {
       var config = DomainConfigurationFactory.Create();
-      config.Types.Register(typeof(IHasName).Assembly, typeof(IHasName).Namespace);
+      config.Types.RegisterCaching(typeof(IHasName).Assembly, typeof(IHasName).Namespace);
       var domain = generator(config);
       using (var session = domain.OpenSession())
       using (var t = session.OpenTransaction()) {

Orm/Xtensive.Orm.Tests/Interfaces/InterfaceTest.cs

@@ -109,7 +109,7 @@ public class InterfaceTest : AutoBuildTest
     protected override DomainConfiguration BuildConfiguration()
     {
       var config = base.BuildConfiguration();
-      config.Types.Register(typeof (Person1).Assembly, typeof (Person1).Namespace);
+      config.Types.RegisterCaching(typeof (Person1).Assembly, typeof (Person1).Namespace);
       return config;
     }
 

Orm/Xtensive.Orm.Tests/Interfaces/KeyStructureConflictTest.cs

@@ -50,7 +50,7 @@ public class KeyStructureConflictTest : AutoBuildTest
     protected override DomainConfiguration BuildConfiguration()
     {
       var config = base.BuildConfiguration();
-      config.Types.Register(typeof (Root1).Assembly, typeof (Root1).Namespace);
+      config.Types.RegisterCaching(typeof (Root1).Assembly, typeof (Root1).Namespace);
       return config;
     }
 

Orm/Xtensive.Orm.Tests/Interfaces/TypeIdModeConflictTest.cs

@@ -39,7 +39,7 @@ public class TypeIdModeConflictTest : AutoBuildTest
     protected override DomainConfiguration BuildConfiguration()
     {
       var config = base.BuildConfiguration();
-      config.Types.Register(typeof (Root1).Assembly, typeof (Root1).Namespace);
+      config.Types.RegisterCaching(typeof (Root1).Assembly, typeof (Root1).Namespace);
       return config;
     }
 

Orm/Xtensive.Orm.Tests/Interfaces/UnusedTypeRemovalTest.cs

@@ -62,7 +62,7 @@ public void MainTest()
     protected override DomainConfiguration BuildConfiguration()
     {
       var config = base.BuildConfiguration();
-      config.Types.Register(typeof (IFirst).Assembly, typeof (IFirst).Namespace);
+      config.Types.RegisterCaching(typeof (IFirst).Assembly, typeof (IFirst).Namespace);
       return config;
     }
 

Orm/Xtensive.Orm.Tests/Issues/BatchingCommandProcessorFrenzy.cs

@@ -1,4 +1,4 @@
-// Copyright (C) 2012 Xtensive LLC.
+// Copyright (C) 2012 Xtensive LLC.
 // All rights reserved.
 // For conditions of distribution and use, see license.
 // Created by: Denis Krjuchkov
@@ -27,7 +27,7 @@ public class BatchingCommandProcessorFrenzy : AutoBuildTest
     protected override DomainConfiguration BuildConfiguration()
     {
       var configuration = base.BuildConfiguration();
-      configuration.Types.Register(typeof (FrenzyEntity).Assembly, typeof (FrenzyEntity).Namespace);
+      configuration.Types.RegisterCaching(typeof (FrenzyEntity).Assembly, typeof (FrenzyEntity).Namespace);
       return configuration;
     }
 

Orm/Xtensive.Orm.Tests/Issues/ClientProfileEntitySetTest.cs

@@ -1,4 +1,4 @@
-// Copyright (C) 2012 Xtensive LLC.
+// Copyright (C) 2012 Xtensive LLC.
 // All rights reserved.
 // For conditions of distribution and use, see license.
 // Created by: Denis Krjuchkov
@@ -48,7 +48,7 @@ public class ClientProfileEntitySetTest : AutoBuildTest
     protected override DomainConfiguration BuildConfiguration()
     {
       var configuration = base.BuildConfiguration();
-      configuration.Types.Register(typeof (Owner).Assembly, typeof (Owner).Namespace);
+      configuration.Types.RegisterCaching(typeof (Owner).Assembly, typeof (Owner).Namespace);
       configuration.Sessions.Default.Options = SessionOptions.ClientProfile;
       return configuration;
     }

Orm/Xtensive.Orm.Tests/Issues/CustomCompilerException.cs

@@ -116,7 +116,7 @@ public class CustomCompilerException : AutoBuildTest
         protected override DomainConfiguration BuildConfiguration()
         {
             var config = base.BuildConfiguration();
-            config.Types.Register(typeof (FundBase).Assembly, typeof (FundBase).Namespace);
+            config.Types.RegisterCaching(typeof (FundBase).Assembly, typeof (FundBase).Namespace);
             return config;
         }