IRecurringMessageScheduler 测试使用，以及bug发现

2024-10-17 21:43:39 +08:00
parent 81b9c57de7
commit 8760ad3904
8 changed files with 93 additions and 135 deletions
@@ -0,0 +1,66 @@
+using Microsoft.Extensions.Hosting;
+using System.Threading;
+using System;
+using System.Threading.Tasks;
+using MassTransit;
+using IRaCIS.Core.Domain.Models;
+using IRaCIS.Core.Infra.EFCore;
+using Microsoft.Extensions.Logging;
+using Hangfire;
+using IRaCIS.Core.Application.Helper;
+using IRaCIS.Core.Application.MassTransit.Consumer;
+using IRaCIS.Core.Domain.Share;
+using MassTransit.Scheduling;
+using Hangfire.Storage;
+using System.Linq;
+using Microsoft.EntityFrameworkCore;
+
+namespace IRaCIS.Core.API.HostService;
+
+public class HangfireHostService(IRecurringMessageScheduler _recurringMessageScheduler, 
+    IRepository<TrialEmailNoticeConfig> _trialEmailNoticeConfigRepository,
+    ILogger<HangfireHostService> _logger) : IHostedService
+{
+  
+
+    public async Task StartAsync(CancellationToken cancellationToken)
+    {
+        _logger.LogInformation("项目启动 hangfire 任务初始化 执行开始~");
+
+
+        //创建邮件定时任务
+        //项目定时任务都在default 队列
+        var dbJobIdList = JobStorage.Current.GetConnection().GetRecurringJobs().Where(t => t.Queue == "default").Select(t => t.Id).ToList();
+
+        foreach (var jobId in dbJobIdList)
+        {
+            HangfireJobHelper.RemoveCronJob(jobId);
+        }
+
+
+        var taskInfoList = await _trialEmailNoticeConfigRepository.Where(t => t.Trial.TrialStatusStr == StaticData.TrialState.TrialOngoing && t.EmailCron != string.Empty && t.IsAutoSend)
+              .Select(t => new { t.Id, t.Code, TrialCode = t.Trial.TrialCode, t.EmailCron, t.BusinessScenarioEnum, t.TrialId })
+              .ToListAsync();
+
+        foreach (var task in taskInfoList)
+        {
+            //利用主键作为任务Id
+            var jobId = $"{task.TrialId}({task.TrialCode})_({task.BusinessScenarioEnum})";
+
+            var trialId = task.TrialId;
+
+            HangfireJobHelper.AddOrUpdateTrialCronJob(jobId, trialId, task.BusinessScenarioEnum, task.EmailCron);
+        }
+
+
+        await _recurringMessageScheduler.ScheduleRecurringPublish(new QCImageQuestionSchedule() { CronExpression = "0/3 * * * * ? " }, new MasstransiTestCommand { value = "message at " + DateTime.Now.ToString() });
+
+
+
+        _logger.LogInformation("项目启动 hangfire 任务初始化 执行结束");
+    }
+
+    public Task StopAsync(CancellationToken cancellationToken) => Task.CompletedTask;
+
+
+}