单个任务内并行生成

Author: XiaoHeitu

src/SmartCode.Generator/BuildTasks/TableBuildTask.cs

@@ -28,16 +28,33 @@ public override async Task Build(BuildContext context)
             var filterTables = FilterTable(context.GetDataSource<ITableSource>().Tables, context.BuildKey,
                 context.Build);
             context.SetCurrentAllTable(filterTables);
-            foreach (var table in filterTables)
+            var tasks = filterTables.Select(table =>
             {
-                _logger.LogInformation($"BuildTable:{table.Name} Start!");
-                context.SetCurrentTable(table);
-                _pluginManager.Resolve<INamingConverter>().Convert(context);
-                context.Result = await _pluginManager.Resolve<ITemplateEngine>(context.Build.TemplateEngine.Name)
-                    .Render(context);
-                await _pluginManager.Resolve<IOutput>(context.Build.Output.Type).Output(context);
-                _logger.LogInformation($"BuildTable:{table.Name} End!");
-            }
+                BuildContext newContext = new BuildContext();
+                newContext.Project = context.Project;
+                newContext.DataSource = context.DataSource;
+                newContext.PluginManager = context.PluginManager;
+                newContext.BuildKey = context.BuildKey;
+                newContext.Build = context.Build;
+                newContext.Result = context.Result;
+                newContext.Output = context.Output;
+                newContext.SetCurrentTable(table);
+
+                return BuildTable(newContext);
+            }).ToArray();
+
+            await Task.WhenAll(tasks);
+        }
+
+        private async Task BuildTable(BuildContext context)
+        {
+            var table = context.GetCurrentTable();
+            _logger.LogInformation($"BuildTable:{table.Name} Start!");
+            _pluginManager.Resolve<INamingConverter>().Convert(context);
+            context.Result = await _pluginManager.Resolve<ITemplateEngine>(context.Build.TemplateEngine.Name)
+                .Render(context);
+            await _pluginManager.Resolve<IOutput>(context.Build.Output.Type).Output(context);
+            _logger.LogInformation($"BuildTable:{table.Name} End!");
         }
 
         public override void Initialize(IDictionary<string, object> parameters)

src/SmartCode.Generator/GeneratorProjectBuilder.cs

@@ -32,43 +32,20 @@ public async Task Build()
             this.countdown.Reset();
             foreach (var buildKV in _project.BuildTasks)
             {
-                if (!buildKV.Value.WaitPre)
+                _logger.LogInformation($"-------- BuildTask:{buildKV.Key} Start! ---------");
+                var output = buildKV.Value.Output;
+                var buildContext = new BuildContext
                 {
-                    this.countdown.AddCount();
-                    Thread t = new Thread((obj) =>
-                    {
-                        var p = ((KeyValuePair<string, Build> buildKV, IDataSource dataSource))obj;
-                        BuildTask(buildKV, dataSource);
-                        this.countdown.Signal();
-                    });
-
-                    t.Start((buildKV, dataSource));
-                    continue;
-                }
-
-                this.countdown.Signal();
-                this.countdown.Wait();
-                this.countdown.Reset();
-
-                BuildTask(buildKV, dataSource);
+                    PluginManager = _pluginManager,
+                    Project = _project,
+                    DataSource = dataSource,
+                    BuildKey = buildKV.Key,
+                    Build = buildKV.Value,
+                    Output = output?.Copy()
+                };
+                await _pluginManager.Resolve<IBuildTask>(buildKV.Value.Type).Build(buildContext);
+                _logger.LogInformation($"-------- BuildTask:{buildKV.Key} End! ---------");
             }
         }
-
-        private void BuildTask(KeyValuePair<string, Build> buildKV, IDataSource dataSource)
-        {
-            _logger.LogInformation($"-------- BuildTask:{buildKV.Key} Start! ---------");
-            var output = buildKV.Value.Output;
-            BuildContext buildContext = new BuildContext
-            {
-                PluginManager = _pluginManager,
-                Project = _project,
-                DataSource = dataSource,
-                BuildKey = buildKV.Key,
-                Build = buildKV.Value,
-                Output = output?.Copy()
-            };
-            _pluginManager.Resolve<IBuildTask>(buildKV.Value.Type).Build(buildContext).Wait();
-            _logger.LogInformation($"-------- BuildTask:{buildKV.Key} End! ---------");
-        }
     }
 }

src/SmartCode/Configuration/Build.cs

@@ -21,10 +21,6 @@ public class Build
         public bool? IgnoreView { get; set; }
         public NamingConverter NamingConverter { get; set; }
         /// <summary>
-        /// 是否等待前序任务
-        /// </summary>
-        public bool WaitPre { get; set; } = true;
-        /// <summary>
         /// 自定义构建参数
         /// </summary>
         public IDictionary<String, object> Parameters { get; set; }