修复循环依赖StackOverflow问题

Author: yihango

src/Riven.Modular/Modular/ModuleDescriptor.cs

@@ -29,5 +29,10 @@ public ModuleDescriptor(Type moduleType, params IModuleDescriptor[] dependencies
             this.ModuleType = moduleType;
             this.Dependencies = dependencies ?? new ModuleDescriptor[0];
         }
+
+        public override string ToString()
+        {
+            return ModuleType.FullName;
+        }
     }
 }

src/Riven.Modular/Modular/ModuleManager.cs

@@ -162,37 +162,54 @@ public virtual List<IModuleDescriptor> ModuleSort<TModule>(List<IModuleDescripto
         /// </summary>
         /// <param name="moduleType"></param>
         /// <returns></returns>
-        protected virtual List<IModuleDescriptor> VisitModule(Type moduleType)
+        protected virtual List<IModuleDescriptor> VisitModule(Type moduleType, Type parentModuleType = null, Dictionary<Type, bool> parentDependModuleType = null)
         {
             var moduleDescriptors = new List<IModuleDescriptor>();
             if (!moduleType.IsModule())
             {
-                return moduleDescriptors;
+                goto end;
+            }
+            if (parentDependModuleType == null)
+            {
+                parentDependModuleType = new Dictionary<Type, bool>();
             }
 
-            // 依赖标记
+            // 获取依赖模块信息
             var dependModulesAttribute = moduleType.GetCustomAttribute<DependsOnAttribute>(false);
 
-            // 依赖属性为空
+            // 创建模块信息
+            // 依赖模块为空
             if (dependModulesAttribute == null)
             {
-                moduleDescriptors.Add(new ModuleDescriptor(moduleType));
+                moduleDescriptors.Add(
+                    new ModuleDescriptor(moduleType)
+                    );
+                goto end;
             }
-            else
+
+            // 依赖属性不为空,递归获取依赖
+            var dependModuleDescriptors = new List<IModuleDescriptor>();
+            foreach (var dependModuleType in dependModulesAttribute.DependModuleTypes)
             {
-                // 依赖属性不为空,递归获取依赖
-                var dependModuleDescriptors = new List<IModuleDescriptor>();
-                foreach (var dependModuleType in dependModulesAttribute.DependModuleTypes)
+                if (parentDependModuleType.TryGetValue(dependModuleType, out var isEx) || dependModuleType == parentModuleType)
                 {
-                    dependModuleDescriptors.AddRange(
-                        this.VisitModule(dependModuleType)
-                    );
+                    throw new InvalidOperationException($"There are cyclic dependencies！\r\n{moduleType.FullName} <--> {dependModuleType.FullName}");
                 }
 
-                // 创建模块描述信息,内容为模块类型和依赖类型
-                moduleDescriptors.Add(new ModuleDescriptor(moduleType, dependModuleDescriptors.ToArray()));
+                parentDependModuleType.Add(dependModuleType, true);
+                dependModuleDescriptors.AddRange(
+                    this.VisitModule(dependModuleType, moduleType, parentDependModuleType)
+                );
             }
 
+            parentDependModuleType.Clear();
+            // 创建模块信息,内容为模块类型和依赖信息
+            moduleDescriptors.Add(
+                new ModuleDescriptor(moduleType, dependModuleDescriptors.ToArray())
+                );
+
+
+        end: // 返回结果
             return moduleDescriptors;
         }
 

src/Riven.Modular/Modular/Topological.cs

@@ -51,30 +51,32 @@ static void Visit<T>(T item, Func<T, IEnumerable<T>> getDependencies, List<T> so
                 // 如果处理的为当前节点，则说明存在循环引用
                 if (inProcess)
                 {
-                    throw new ArgumentException("topological module cyclic dependency found.");
+                    throw new InvalidOperationException($"There are cyclic dependencies！\r\n{item.ToString()}");
                 }
+
+                return;
             }
-            else
-            {
-                // 正在处理当前顶点
-                visited[item] = true;
 
-                // 获得所有依赖项
-                var dependencies = getDependencies(item);
-                // 如果依赖项集合不为空，遍历访问其依赖节点
-                if (dependencies != null)
+
+
+            // 正在处理当前顶点
+            visited[item] = true;
+
+            // 获得所有依赖项
+            var dependencies = getDependencies(item);
+            // 如果依赖项集合不为空，遍历访问其依赖节点
+            if (dependencies != null)
+            {
+                foreach (var dependency in dependencies)
                 {
-                    foreach (var dependency in dependencies)
-                    {
-                        // 递归遍历访问
-                        Visit(dependency, getDependencies, sorted, visited);
-                    }
+                    // 递归遍历访问
+                    Visit(dependency, getDependencies, sorted, visited);
                 }
-
-                // 处理完成置为 false
-                visited[item] = false;
-                sorted.Add(item);
             }
+
+            // 处理完成置为 false
+            visited[item] = false;
+            sorted.Add(item);
         }
     }
 }