添加任务依赖

Author: XiaoHeitu

src/SmartCode.Generator/GeneratorProjectBuilder.cs

@@ -1,4 +1,5 @@
 using System.Collections.Generic;
+using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.Extensions.Logging;
@@ -11,7 +12,6 @@ public class GeneratorProjectBuilder : IProjectBuilder
         private readonly Project _project;
         private readonly IPluginManager _pluginManager;
         private readonly ILogger<GeneratorProjectBuilder> _logger;
-        private CountdownEvent countdown = new CountdownEvent(1);
 
         public GeneratorProjectBuilder(
             Project project
@@ -23,29 +23,69 @@ Project project
             _logger = logger;
         }
 
+        CountdownEvent countdown = new CountdownEvent(1);
 
         public async Task Build()
         {
             var dataSource = _pluginManager.Resolve<IDataSource>(_project.DataSource.Name);
             await dataSource.InitData();
 
-            this.countdown.Reset();
-            foreach (var buildKV in _project.BuildTasks)
+            //foreach (var buildKV in _project.BuildTasks)
+            //{
+            //    _logger.LogInformation($"-------- BuildTask:{buildKV.Key} Start! ---------");
+            //    var output = buildKV.Value.Output;
+            //    var buildContext = new BuildContext
+            //    {
+            //        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! ---------");
+            //}
+            BuildContext[] contexts = _project.BuildTasks.Select(d => new BuildContext
             {
-                _logger.LogInformation($"-------- BuildTask:{buildKV.Key} Start! ---------");
-                var output = buildKV.Value.Output;
-                var buildContext = new BuildContext
+                PluginManager = _pluginManager,
+                Project = _project,
+                DataSource = dataSource,
+                BuildKey = d.Key,
+                Build = d.Value,
+                Output = d.Value.Output?.Copy(),
+            }).ToArray();
+            foreach (var context in contexts)
+            {
+                context.DependOn = contexts.Where(d => d.Build.DependOn.Contains(d.BuildKey)).ToArray();
+            }
+            countdown.Reset();
+            foreach (var context in contexts)
+            {
+                context.BuildTask = Task.Factory.StartNew(this.BuildTask, null, TaskCreationOptions.LongRunning);
+            }
+
+            countdown.Signal();
+        }
+        private async void BuildTask(object obj)
+        {
+            countdown.Wait();
+            var context = (BuildContext)obj;
+            _logger.LogInformation($"-------- BuildTask:{context.BuildKey} Wait! ---------");
+            //等待依赖任务
+            if (context.DependOn != null)
+            {
+                foreach (var dcontext in context.DependOn)
                 {
-                    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! ---------");
+                    await dcontext.BuildTask;
+                }
             }
+
+            _logger.LogInformation($"-------- BuildTask:{context.BuildKey} Start! ---------");
+            //执行自身任务
+            await _pluginManager.Resolve<IBuildTask>(context.Build.Type).Build(context);
+
+            _logger.LogInformation($"-------- BuildTask:{context.BuildKey} End! ---------");
         }
     }
 }

src/SmartCode/BuildContext.cs

@@ -1,6 +1,7 @@
 using System;
 using System.Collections.Generic;
 using System.Text;
+using System.Threading.Tasks;
 using SmartCode.Configuration;
 
 namespace SmartCode
@@ -26,5 +27,9 @@ public void SetItem(string key, object item)
         {
             Items[key] = item;
         }
+
+        public IEnumerable<BuildContext> DependOn { get; set; }
+        public IEnumerable<BuildContext> Next { get; set; }
+        public Task BuildTask { get; set; }
     }
 }

src/SmartCode/Configuration/Build.cs

@@ -13,14 +13,18 @@ public class Build
         /// </summary>
         public String Type { get; set; }
         public String Module { get; set; }
-        public TemplateEngine TemplateEngine { get; set; } 
+        public TemplateEngine TemplateEngine { get; set; }
         public Output Output { get; set; }
         public IEnumerable<String> IncludeTables { get; set; }
         public IEnumerable<String> IgnoreTables { get; set; }
         public bool? IgnoreNoPKTable { get; set; }
         public bool? IgnoreView { get; set; }
         public NamingConverter NamingConverter { get; set; }
         /// <summary>
+        /// 依赖于
+        /// </summary>
+        public IEnumerable<String> DependOn { get; set; }
+        /// <summary>
         /// 自定义构建参数
         /// </summary>
         public IDictionary<String, object> Parameters { get; set; }