做了定时任务同步功能

Deali-Axy · Jan 24, 2022 · d0549a4 · d0549a4
1 parent 3140d47
commit d0549a4
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/CrawlCenter.Web/Controllers/RecurringTaskController.cs b/CrawlCenter.Web/Controllers/RecurringTaskController.cs
@@ -5,7 +5,9 @@
 using CrawlCenter.Contrib.WebMessages;
 using CrawlCenter.Data.Models;
 using CrawlCenter.Data.Repositories;
+using CrawlCenter.Web.Tasks;
 using CrawlCenter.Web.ViewModels.RecurringTasks;
+using Hangfire;
 using Microsoft.AspNetCore.Http;
 using Microsoft.AspNetCore.Mvc;
 using Microsoft.AspNetCore.Mvc.Rendering;
@@ -93,12 +95,31 @@ public IActionResult Edit(RecurringTaskEditViewModel viewModel) {
 
         [HttpPost]
         public IActionResult Delete([FromForm] Guid id) {
+            var job = _recurringTaskRepo.GetById(id);
+            if (job == null) return NotFound();
+            RecurringJob.RemoveIfExists(job.Id.ToString());
+
             var affectRows = _recurringTaskRepo.Delete(id);
             if (affectRows > 0)
                 _messages.Success($"删除定时任务 {id} 成功！");
             else
                 _messages.Error($"删除定时任务 {id} 失败！");
             return RedirectToAction(nameof(Index));
         }
+
+        [HttpGet]
+        public IActionResult SyncData() {
+            var jobs = _recurringTaskRepo.GetAll().ToList();
+            foreach (var job in jobs) {
+                RecurringJob.RemoveIfExists(job.Id.ToString());
+                RecurringJob.AddOrUpdate(job.Id.ToString(),
+                    () => new RunCrawl().Run(job.CrawlTask),
+                    () => job.Cron);
+            }
+
+            _messages.Info($"已同步{jobs.Count}个定时任务！");
+
+            return RedirectToAction(nameof(Index));
+        }
     }
 }
diff --git a/CrawlCenter.Web/Views/RecurringTask/Index.cshtml b/CrawlCenter.Web/Views/RecurringTask/Index.cshtml
@@ -18,6 +18,10 @@
                 <span data-feather="plus"></span>
                 新增
             </a>
+            <a class="btn btn-outline-success" asp-controller="RecurringTask" asp-action="SyncData">
+                <span data-feather="refresh-cw"></span>
+                同步
+            </a>
         </div>
     </div>
 </div>