Kaynağa Gözat

fix: 补回 d68cccf 调整去掉的脚本创建代码的逻辑

许俊杰 3 yıl önce
ebeveyn
işleme
520b37f2b6

+ 38 - 7
Admin.NET/Admin.NET.Core/Service/Job/DbJobPersistence.cs

@@ -11,11 +11,11 @@ public class DbJobPersistence : IJobPersistence
     private readonly SqlSugarRepository<SysJobTrigger> _triggerRepository;
     private readonly DynamicJobCompiler _dynamicJobCompiler;
 
-    public DbJobPersistence(IServiceProvider serviceProvider, DynamicJobCompiler dynamicJobCompiler)
+    public DbJobPersistence(IServiceProvider serviceProvider)
     {
         _serviceScope = serviceProvider.CreateScope();
         var services = _serviceScope.ServiceProvider;
-        _dynamicJobCompiler = dynamicJobCompiler;
+        _dynamicJobCompiler = services.GetService<DynamicJobCompiler>();
         _sqlSugarClient = services.GetService<ISqlSugarClient>();
         _jobRepository = services.GetService<SqlSugarRepository<SysJobDetail>>();
         _triggerRepository = services.GetService<SqlSugarRepository<SysJobTrigger>>();
@@ -27,9 +27,8 @@ public class DbJobPersistence : IJobPersistence
     /// <returns></returns>
     public IEnumerable<SchedulerBuilder> Preload()
     {
-
         // 获取所有定义的作业
-        var allJobs = App.EffectiveTypes.ScanToBuilders();
+        var allJobs = App.EffectiveTypes.ScanToBuilders().ToList();
         // 若数据库不存在任何作业,则直接返回
         if (!_jobRepository.IsAny(u => true)) return allJobs;
 
@@ -46,21 +45,53 @@ public class DbJobPersistence : IJobPersistence
             // 同步数据库数据
             jobBuilder.LoadFrom(dbDetail);
 
+            // 获取作业的所有数据库的触发器
+            var dbTriggers = _triggerRepository.GetList(u => u.JobId == jobBuilder.JobId).ToArray();
             // 遍历所有作业触发器
             foreach (var (_, triggerBuilder) in schedulerBuilder.GetEnumerable())
             {
                 // 加载数据库数据
-                var dbTrigger = _triggerRepository.GetFirst(u => u.JobId == jobBuilder.JobId && u.TriggerId == triggerBuilder.TriggerId);
+                var dbTrigger = dbTriggers.FirstOrDefault(u => u.JobId == jobBuilder.JobId && u.TriggerId == triggerBuilder.TriggerId);
                 if (dbTrigger == null) continue;
 
                 triggerBuilder.LoadFrom(dbTrigger)
                               .Updated();   // 标记更新
             }
+            // 遍历所有非编译时定义的触发器加入到作业中
+            foreach (var dbTrigger in dbTriggers)
+            {
+                if (schedulerBuilder.GetTriggerBuilder(dbTrigger.TriggerId)?.JobId == jobBuilder.JobId) continue;
+                var triggerBuilder = TriggerBuilder.Create(dbTrigger.TriggerId).LoadFrom(dbTrigger);
+                schedulerBuilder.AddTriggerBuilder(triggerBuilder); // 先添加
+                triggerBuilder.Updated(); // 再标记更新
+            }
 
             // 标记更新
             schedulerBuilder.Updated();
         }
 
+        // 获取数据库所有通过脚本创建的作业
+        var allDbScriptJobs = _jobRepository.GetList(u => u.CreateFromScript);
+        foreach (var dbDetail in allDbScriptJobs)
+        {
+            // 动态创建作业
+            var jobType = _dynamicJobCompiler.BuildJob(dbDetail.ScriptCode);
+            var jobBuilder = JobBuilder.Create(jobType).LoadFrom(dbDetail);
+
+            // 强行设置为不扫描 IJob 实现类 [Trigger] 特性触发器,否则 SchedulerBuilder.Create 会再次扫描,导致重复添加同名触发器
+            jobBuilder.SetIncludeAnnotations(false);
+
+            // 获取作业的所有数据库的触发器加入到作业中
+            var dbTriggers = _triggerRepository.GetList(u => u.JobId == jobBuilder.JobId).ToArray();
+            var triggerBuilders = dbTriggers.Select(u => TriggerBuilder.Create(u.TriggerId).LoadFrom(u).Updated());
+            var schedulerBuilder = SchedulerBuilder.Create(jobBuilder, triggerBuilders.ToArray());
+
+            // 标记更新
+            schedulerBuilder.Updated();
+
+            allJobs.Add(schedulerBuilder);
+        }
+
         return allJobs;
     }
 
@@ -87,7 +118,7 @@ public class DbJobPersistence : IJobPersistence
         }
         else if (context.Behavior == PersistenceBehavior.Updated)
         {
-            _sqlSugarClient.Updateable(jobDetail).WhereColumns(u => new { u.JobId }).IgnoreColumns(u => new { u.Id }).ExecuteCommand();
+            _sqlSugarClient.Updateable(jobDetail).WhereColumns(u => new { u.JobId }).IgnoreColumns(u => new { u.Id, u.CreateFromScript, u.ScriptCode }).ExecuteCommand();
         }
         else if (context.Behavior == PersistenceBehavior.Removed)
         {
@@ -115,7 +146,7 @@ public class DbJobPersistence : IJobPersistence
             _sqlSugarClient.Deleteable<SysJobTrigger>().Where(u => u.TriggerId == jobTrigger.TriggerId && u.JobId == jobTrigger.JobId).ExecuteCommand();
         }
     }
-   
+
     public void Dispose()
     {
         _serviceScope?.Dispose();

+ 1 - 1
Admin.NET/Admin.NET.Core/Service/Job/SysJobService.cs

@@ -81,7 +81,7 @@ public class SysJobService : IDynamicApiController, ITransient
             throw Oops.Oh(ErrorCodeEnum.D1703);
 
         _schedulerFactory.AddJob(
-            JobBuilder.Create(jobType).SetIncludeAnnotations(input.IncludeAnnotations)
+            JobBuilder.Create(jobType)
                 .LoadFrom(input.Adapt<SysJobDetail>()).SetJobType(jobType));
 
         // 延迟一下等待持久化写入,再执行其他字段的更新