perf: cache CodeDom assembly paths

Author: kpkhxlgy0

MCPForUnity/Editor/Tools/ExecuteCode.cs

@@ -30,11 +30,13 @@ public static class ExecuteCode
 
         private static readonly List<HistoryEntry> _history = new List<HistoryEntry>();
         private static string[] _cachedAssemblyPaths;
+        private static string[] _cachedCodeDomAssemblyPaths;
 
         [UnityEditor.InitializeOnLoadMethod]
         private static void OnDomainReload()
         {
             _cachedAssemblyPaths = null;
+            _cachedCodeDomAssemblyPaths = null;
             RoslynCompiler.ResetCache();
         }
 
@@ -342,6 +344,10 @@ private static Assembly CodeDomCompile(string source, string[] assemblyPaths, ou
 
         internal static string[] FilterAssemblyPathsForCodeDom(string[] allPaths)
         {
+            var useCache = ReferenceEquals(allPaths, _cachedAssemblyPaths);
+            if (useCache && _cachedCodeDomAssemblyPaths != null)
+                return _cachedCodeDomAssemblyPaths;
+
             var hasNetstandard = allPaths.Any(p =>
                 string.Equals(Path.GetFileNameWithoutExtension(p), "netstandard", StringComparison.OrdinalIgnoreCase));
 
@@ -350,7 +356,10 @@ internal static string[] FilterAssemblyPathsForCodeDom(string[] allPaths)
                     !_codedomDuplicateAssemblies.Contains(Path.GetFileNameWithoutExtension(p))).ToArray()
                 : allPaths;
 
-            return DeduplicateAssemblyPathsForCodeDom(filtered);
+            var result = DeduplicateAssemblyPathsForCodeDom(filtered);
+            if (useCache)
+                _cachedCodeDomAssemblyPaths = result;
+            return result;
         }
 
         private static string[] DeduplicateAssemblyPathsForCodeDom(string[] paths)

TestProjects/UnityMCPTests/Assets/Tests/EditMode/Tools/ExecuteCodeTests.cs

@@ -2,6 +2,7 @@
 using System.CodeDom.Compiler;
 using System.IO;
 using System.Linq;
+using System.Reflection;
 using Microsoft.CSharp;
 using Newtonsoft.Json.Linq;
 using NUnit.Framework;
@@ -425,6 +426,53 @@ public void FilterAssemblyPathsForCodeDom_DuplicateNames_PrefersReferencedVersio
             }
         }
 
+        [Test]
+        public void FilterAssemblyPathsForCodeDom_CachedAssemblyPaths_ReusesResultUntilDomainReload()
+        {
+            var tempRoot = CreateTempDirectory();
+            var cachedAssemblyPathsField = typeof(ExecuteCode).GetField(
+                "_cachedAssemblyPaths",
+                BindingFlags.NonPublic | BindingFlags.Static);
+            var cachedCodeDomAssemblyPathsField = typeof(ExecuteCode).GetField(
+                "_cachedCodeDomAssemblyPaths",
+                BindingFlags.NonPublic | BindingFlags.Static);
+            var onDomainReload = typeof(ExecuteCode).GetMethod(
+                "OnDomainReload",
+                BindingFlags.NonPublic | BindingFlags.Static);
+            Assert.IsNotNull(cachedAssemblyPathsField);
+            Assert.IsNotNull(cachedCodeDomAssemblyPathsField);
+            Assert.IsNotNull(onDomainReload);
+
+            try
+            {
+                onDomainReload.Invoke(null, null);
+                var assemblyName = "McpCodeDomCache" + Guid.NewGuid().ToString("N");
+                var olderPath = CompileVersionedAssembly(tempRoot, assemblyName, "1.0.0.0");
+                var newerPath = CompileVersionedAssembly(tempRoot, assemblyName, "2.0.0.0");
+                var cachedAssemblyPaths = new[] { olderPath, newerPath };
+                cachedAssemblyPathsField.SetValue(null, cachedAssemblyPaths);
+
+                var first = ExecuteCode.FilterAssemblyPathsForCodeDom(cachedAssemblyPaths);
+                Assert.AreEqual(1, first.Length);
+
+                File.WriteAllText(olderPath, "invalidated");
+                File.WriteAllText(newerPath, "invalidated");
+                var second = ExecuteCode.FilterAssemblyPathsForCodeDom(cachedAssemblyPaths);
+                Assert.AreSame(first, second);
+
+                onDomainReload.Invoke(null, null);
+                cachedAssemblyPathsField.SetValue(null, cachedAssemblyPaths);
+                var afterReload = ExecuteCode.FilterAssemblyPathsForCodeDom(cachedAssemblyPaths);
+                Assert.AreNotSame(first, afterReload);
+                Assert.AreEqual(2, afterReload.Length);
+            }
+            finally
+            {
+                onDomainReload.Invoke(null, null);
+                Directory.Delete(tempRoot, true);
+            }
+        }
+
         // ──────────────────── Helpers ────────────────────
 
         private static string CreateTempDirectory()