| | | 1 | | using Elsa.Common.Entities; |
| | | 2 | | using Elsa.Common.Multitenancy; |
| | | 3 | | using Elsa.Extensions; |
| | | 4 | | using Microsoft.EntityFrameworkCore; |
| | | 5 | | using Microsoft.EntityFrameworkCore.ChangeTracking; |
| | | 6 | | using Microsoft.Extensions.DependencyInjection; |
| | | 7 | | |
| | | 8 | | namespace Elsa.Persistence.EFCore; |
| | | 9 | | |
| | | 10 | | /// <summary> |
| | | 11 | | /// An optional base class to implement with some opinions on certain converters to install for certain DB providers. |
| | | 12 | | /// </summary> |
| | | 13 | | public abstract class ElsaDbContextBase : DbContext, IElsaDbContextSchema |
| | | 14 | | { |
| | 1 | 15 | | private static readonly ISet<EntityState> ModifiedEntityStates = new HashSet<EntityState> |
| | 1 | 16 | | { |
| | 1 | 17 | | EntityState.Added, |
| | 1 | 18 | | EntityState.Modified, |
| | 1 | 19 | | }; |
| | | 20 | | |
| | 925 | 21 | | protected IServiceProvider ServiceProvider { get; } |
| | | 22 | | private readonly ElsaDbContextOptions? _elsaDbContextOptions; |
| | 262472 | 23 | | public string? TenantId { get; set; } |
| | | 24 | | |
| | | 25 | | /// <summary> |
| | | 26 | | /// The default schema used by Elsa. |
| | | 27 | | /// </summary> |
| | 49903 | 28 | | public static string ElsaSchema { get; set; } = "Elsa"; |
| | | 29 | | |
| | | 30 | | /// <inheritdoc/> |
| | 163 | 31 | | public string Schema { get; } |
| | | 32 | | |
| | | 33 | | /// <summary> |
| | | 34 | | /// The table used to store the migrations history. |
| | | 35 | | /// </summary> |
| | 3052 | 36 | | public static string MigrationsHistoryTable { get; set; } = "__EFMigrationsHistory"; |
| | | 37 | | |
| | | 38 | | /// <summary> |
| | | 39 | | /// Initializes a new instance of the <see cref="ElsaDbContextBase"/> class. |
| | | 40 | | /// </summary> |
| | 46851 | 41 | | protected ElsaDbContextBase(DbContextOptions options, IServiceProvider serviceProvider) : base(options) |
| | | 42 | | { |
| | 46851 | 43 | | ServiceProvider = serviceProvider; |
| | 46851 | 44 | | _elsaDbContextOptions = options.FindExtension<ElsaDbContextOptionsExtension>()?.Options; |
| | | 45 | | |
| | | 46 | | // ReSharper disable once VirtualMemberCallInConstructor |
| | 46851 | 47 | | Schema = !string.IsNullOrWhiteSpace(_elsaDbContextOptions?.SchemaName) ? _elsaDbContextOptions.SchemaName : Elsa |
| | | 48 | | |
| | 46851 | 49 | | var tenantAccessor = serviceProvider.GetService<ITenantAccessor>(); |
| | 46851 | 50 | | var tenantId = (tenantAccessor?.TenantId).NormalizeTenantId(); |
| | 46851 | 51 | | TenantId ??= tenantId; |
| | 46851 | 52 | | } |
| | | 53 | | |
| | | 54 | | /// <inheritdoc/> |
| | | 55 | | public override async Task<int> SaveChangesAsync(CancellationToken cancellationToken = default) |
| | | 56 | | { |
| | 921 | 57 | | await OnBeforeSavingAsync(cancellationToken); |
| | 921 | 58 | | return await base.SaveChangesAsync(cancellationToken); |
| | 920 | 59 | | } |
| | | 60 | | |
| | | 61 | | /// <inheritdoc /> |
| | | 62 | | protected override void OnModelCreating(ModelBuilder modelBuilder) |
| | | 63 | | { |
| | 4 | 64 | | if (!string.IsNullOrWhiteSpace(Schema)) |
| | 4 | 65 | | modelBuilder.HasDefaultSchema(Schema); |
| | | 66 | | |
| | 4 | 67 | | var additionalConfigurations = _elsaDbContextOptions?.GetModelConfigurations(this); |
| | | 68 | | |
| | 4 | 69 | | additionalConfigurations?.Invoke(modelBuilder); |
| | | 70 | | |
| | 4 | 71 | | using var scope = ServiceProvider.CreateScope(); |
| | 4 | 72 | | var entityTypeHandlers = scope.ServiceProvider.GetServices<IEntityModelCreatingHandler>().ToList(); |
| | | 73 | | |
| | 66 | 74 | | foreach (var entityType in modelBuilder.Model.GetEntityTypes().ToList()) |
| | | 75 | | { |
| | 464 | 76 | | foreach (var handler in entityTypeHandlers) |
| | 203 | 77 | | handler.Handle(this, modelBuilder, entityType); |
| | | 78 | | } |
| | 4 | 79 | | } |
| | | 80 | | |
| | | 81 | | private async Task OnBeforeSavingAsync(CancellationToken cancellationToken) |
| | | 82 | | { |
| | 921 | 83 | | using var scope = ServiceProvider.CreateScope(); |
| | 921 | 84 | | var handlers = scope.ServiceProvider.GetServices<IEntitySavingHandler>().ToList(); |
| | 13478 | 85 | | foreach (var entry in ChangeTracker.Entries().Where(IsModifiedEntity)) |
| | | 86 | | { |
| | 81452 | 87 | | foreach (var handler in handlers) |
| | 34908 | 88 | | await handler.HandleAsync(this, entry, cancellationToken); |
| | 5818 | 89 | | } |
| | 921 | 90 | | } |
| | | 91 | | |
| | | 92 | | /// <summary> |
| | | 93 | | /// Determine if an entity was modified. |
| | | 94 | | /// </summary> |
| | | 95 | | private bool IsModifiedEntity(EntityEntry entityEntry) |
| | | 96 | | { |
| | 5818 | 97 | | return ModifiedEntityStates.Contains(entityEntry.State) && entityEntry.Entity is Entity; |
| | | 98 | | } |
| | | 99 | | } |