cKey
4 years ago
16 changed files with 334 additions and 37 deletions
@ -0,0 +1,10 @@ |
|||||
|
using Quartz; |
||||
|
|
||||
|
namespace LINGYUN.Abp.BackgroundTasks.Quartz; |
||||
|
|
||||
|
public interface IQuartzKeyBuilder |
||||
|
{ |
||||
|
JobKey CreateJobKey(JobInfo jobInfo); |
||||
|
|
||||
|
TriggerKey CreateTriggerKey(JobInfo jobInfo); |
||||
|
} |
||||
@ -0,0 +1,35 @@ |
|||||
|
using Quartz; |
||||
|
using Volo.Abp.DependencyInjection; |
||||
|
using Volo.Abp.MultiTenancy; |
||||
|
|
||||
|
namespace LINGYUN.Abp.BackgroundTasks.Quartz; |
||||
|
|
||||
|
public class QuartzKeyBuilder : IQuartzKeyBuilder, ISingletonDependency |
||||
|
{ |
||||
|
protected ICurrentTenant CurrentTenant { get; } |
||||
|
|
||||
|
public QuartzKeyBuilder(ICurrentTenant currentTenant) |
||||
|
{ |
||||
|
CurrentTenant = currentTenant; |
||||
|
} |
||||
|
|
||||
|
public JobKey CreateJobKey(JobInfo jobInfo) |
||||
|
{ |
||||
|
var name = jobInfo.Id.ToString(); |
||||
|
var group = CurrentTenant.IsAvailable |
||||
|
? $"{CurrentTenant.Id}:{jobInfo.Group}" |
||||
|
: $"Default:{jobInfo.Group}"; |
||||
|
|
||||
|
return new JobKey(name, group); |
||||
|
} |
||||
|
|
||||
|
public TriggerKey CreateTriggerKey(JobInfo jobInfo) |
||||
|
{ |
||||
|
var name = jobInfo.Id.ToString(); |
||||
|
var group = CurrentTenant.IsAvailable |
||||
|
? $"{CurrentTenant.Id}:{jobInfo.Group}" |
||||
|
: $"Default:{jobInfo.Group}"; |
||||
|
|
||||
|
return new TriggerKey(name, group); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,140 @@ |
|||||
|
using LINGYUN.Abp.BackgroundTasks; |
||||
|
using LINGYUN.Abp.BackgroundTasks.Internal; |
||||
|
using LINGYUN.Abp.Data.DbMigrator; |
||||
|
using LINGYUN.Abp.MultiTenancy; |
||||
|
using LY.MicroService.TaskManagement.EntityFrameworkCore; |
||||
|
using Microsoft.EntityFrameworkCore; |
||||
|
using Microsoft.Extensions.Logging; |
||||
|
using Microsoft.Extensions.Options; |
||||
|
using System; |
||||
|
using System.Collections.Generic; |
||||
|
using System.Threading.Tasks; |
||||
|
using Volo.Abp.DependencyInjection; |
||||
|
using Volo.Abp.Domain.Entities.Events.Distributed; |
||||
|
using Volo.Abp.EventBus.Distributed; |
||||
|
using Volo.Abp.MultiTenancy; |
||||
|
using Volo.Abp.TenantManagement; |
||||
|
using Volo.Abp.Uow; |
||||
|
|
||||
|
namespace LY.MicroService.TaskManagement.EventBus.Handlers |
||||
|
{ |
||||
|
public class TenantSynchronizer : |
||||
|
IDistributedEventHandler<CreateEventData>, |
||||
|
IDistributedEventHandler<EntityDeletedEto<TenantEto>>, |
||||
|
ITransientDependency |
||||
|
{ |
||||
|
protected ILogger<TenantSynchronizer> Logger { get; } |
||||
|
|
||||
|
protected ICurrentTenant CurrentTenant { get; } |
||||
|
protected IUnitOfWorkManager UnitOfWorkManager { get; } |
||||
|
protected IDbSchemaMigrator DbSchemaMigrator { get; } |
||||
|
protected AbpBackgroundTasksOptions Options { get; } |
||||
|
protected IJobScheduler JobScheduler { get; } |
||||
|
public TenantSynchronizer( |
||||
|
ICurrentTenant currentTenant, |
||||
|
IUnitOfWorkManager unitOfWorkManager, |
||||
|
IDbSchemaMigrator dbSchemaMigrator, |
||||
|
IOptions<AbpBackgroundTasksOptions> options, |
||||
|
IJobScheduler jobScheduler, |
||||
|
ILogger<TenantSynchronizer> logger) |
||||
|
{ |
||||
|
CurrentTenant = currentTenant; |
||||
|
UnitOfWorkManager = unitOfWorkManager; |
||||
|
DbSchemaMigrator = dbSchemaMigrator; |
||||
|
JobScheduler = jobScheduler; |
||||
|
Options = options.Value; |
||||
|
|
||||
|
Logger = logger; |
||||
|
} |
||||
|
|
||||
|
public async Task HandleEventAsync(EntityDeletedEto<TenantEto> eventData) |
||||
|
{ |
||||
|
// 租户删除时移除轮询作业
|
||||
|
var pollingJob = BuildPollingJobInfo(eventData.Entity.Id, eventData.Entity.Name); |
||||
|
await JobScheduler.RemoveAsync(pollingJob); |
||||
|
|
||||
|
var cleaningJob = BuildCleaningJobInfo(eventData.Entity.Id, eventData.Entity.Name); |
||||
|
await JobScheduler.RemoveAsync(cleaningJob); |
||||
|
} |
||||
|
|
||||
|
public async Task HandleEventAsync(CreateEventData eventData) |
||||
|
{ |
||||
|
await MigrateAsync(eventData); |
||||
|
|
||||
|
// 持久层介入之后提供对于租户的后台工作者轮询作业
|
||||
|
await QueueBackgroundJobAsync(eventData); |
||||
|
} |
||||
|
|
||||
|
private async Task QueueBackgroundJobAsync(CreateEventData eventData) |
||||
|
{ |
||||
|
var pollingJob = BuildPollingJobInfo(eventData.Id, eventData.Name); |
||||
|
await JobScheduler.QueueAsync(pollingJob); |
||||
|
|
||||
|
var cleaningJob = BuildCleaningJobInfo(eventData.Id, eventData.Name); |
||||
|
await JobScheduler.QueueAsync(cleaningJob); |
||||
|
} |
||||
|
|
||||
|
private async Task MigrateAsync(CreateEventData eventData) |
||||
|
{ |
||||
|
using (var unitOfWork = UnitOfWorkManager.Begin()) |
||||
|
{ |
||||
|
using (CurrentTenant.Change(eventData.Id, eventData.Name)) |
||||
|
{ |
||||
|
Logger.LogInformation("Migrating the new tenant database with localization.."); |
||||
|
// 迁移租户数据
|
||||
|
await DbSchemaMigrator.MigrateAsync<TaskManagementMigrationsDbContext>( |
||||
|
(connectionString, builder) => |
||||
|
{ |
||||
|
builder.UseMySql(connectionString, ServerVersion.AutoDetect(connectionString)); |
||||
|
|
||||
|
return new TaskManagementMigrationsDbContext(builder.Options); |
||||
|
}); |
||||
|
await unitOfWork.SaveChangesAsync(); |
||||
|
|
||||
|
Logger.LogInformation("Migrated the new tenant database with localization."); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private JobInfo BuildPollingJobInfo(Guid tenantId, string tenantName) |
||||
|
{ |
||||
|
return new JobInfo |
||||
|
{ |
||||
|
Id = tenantId, |
||||
|
Name = nameof(BackgroundPollingJob), |
||||
|
Group = "Polling", |
||||
|
Description = "Polling tasks to be executed", |
||||
|
Args = new Dictionary<string, object>() { { nameof(JobInfo.TenantId), tenantId } }, |
||||
|
Status = JobStatus.Running, |
||||
|
BeginTime = DateTime.Now, |
||||
|
CreationTime = DateTime.Now, |
||||
|
Cron = Options.JobFetchCronExpression, |
||||
|
JobType = JobType.Period, |
||||
|
Priority = JobPriority.High, |
||||
|
LockTimeOut = Options.JobFetchLockTimeOut, |
||||
|
TenantId = tenantId, |
||||
|
Type = typeof(BackgroundPollingJob).AssemblyQualifiedName, |
||||
|
}; |
||||
|
} |
||||
|
|
||||
|
private JobInfo BuildCleaningJobInfo(Guid tenantId, string tenantName) |
||||
|
{ |
||||
|
return new JobInfo |
||||
|
{ |
||||
|
Id = tenantId, |
||||
|
Name = nameof(BackgroundCleaningJob), |
||||
|
Group = "Cleaning", |
||||
|
Description = "Cleaning tasks to be executed", |
||||
|
Args = new Dictionary<string, object>() { { nameof(JobInfo.TenantId), tenantId } }, |
||||
|
Status = JobStatus.Running, |
||||
|
BeginTime = DateTime.Now, |
||||
|
CreationTime = DateTime.Now, |
||||
|
Cron = Options.JobCleanCronExpression, |
||||
|
JobType = JobType.Period, |
||||
|
Priority = JobPriority.High, |
||||
|
TenantId = tenantId, |
||||
|
Type = typeof(BackgroundCleaningJob).AssemblyQualifiedName, |
||||
|
}; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue