diff --git a/src/modules/Elsa.Persistence.EntityFrameworkCore.Sqlite/Extensions/DbContextOptionsBuilderExtensions.cs b/src/modules/Elsa.Persistence.EntityFrameworkCore.Sqlite/Extensions/DbContextOptionsBuilderExtensions.cs index 87876aadc..20de7b974 100644 --- a/src/modules/Elsa.Persistence.EntityFrameworkCore.Sqlite/Extensions/DbContextOptionsBuilderExtensions.cs +++ b/src/modules/Elsa.Persistence.EntityFrameworkCore.Sqlite/Extensions/DbContextOptionsBuilderExtensions.cs @@ -1,12 +1,18 @@ using Elsa.Persistence.EntityFrameworkCore.Common; using Microsoft.EntityFrameworkCore; +using Microsoft.EntityFrameworkCore.Infrastructure; namespace Elsa.Persistence.EntityFrameworkCore.Sqlite.Extensions; public static class DbContextOptionsBuilderExtensions { - public static DbContextOptionsBuilder UseElsaSqlite(this DbContextOptionsBuilder builder, string connectionString = Constants.DefaultConnectionString) => - builder.UseSqlite(connectionString, db => db - .MigrationsAssembly(typeof(DbContextOptionsBuilderExtensions).Assembly.GetName().Name) - .MigrationsHistoryTable(ElsaDbContextBase.MigrationsHistoryTable, ElsaDbContextBase.ElsaSchema)); + public static DbContextOptionsBuilder UseElsaSqlite(this DbContextOptionsBuilder builder, string connectionString = Constants.DefaultConnectionString, Action? configure = default) => + builder.UseSqlite(connectionString, db => + { + db + .MigrationsAssembly(typeof(DbContextOptionsBuilderExtensions).Assembly.GetName().Name) + .MigrationsHistoryTable(ElsaDbContextBase.MigrationsHistoryTable, ElsaDbContextBase.ElsaSchema); + + configure?.Invoke(db); + }); } \ No newline at end of file