diff --git a/src/EFCore.Ydb/src/Extensions/YdbContextOptionsBuilderExtensions.cs b/src/EFCore.Ydb/src/Extensions/YdbContextOptionsBuilderExtensions.cs index 54378384..b020a0ee 100644 --- a/src/EFCore.Ydb/src/Extensions/YdbContextOptionsBuilderExtensions.cs +++ b/src/EFCore.Ydb/src/Extensions/YdbContextOptionsBuilderExtensions.cs @@ -39,6 +39,22 @@ public static DbContextOptionsBuilder UseYdb( return optionsBuilder; } + public static DbContextOptionsBuilder UseYdb( + this DbContextOptionsBuilder optionsBuilder, + string? connectionString, + Action? ydbOptionsAction = null) + where TContext : DbContext + => (DbContextOptionsBuilder)UseYdb( + (DbContextOptionsBuilder)optionsBuilder, connectionString, ydbOptionsAction); + + public static DbContextOptionsBuilder UseYdb( + this DbContextOptionsBuilder optionsBuilder, + DbConnection connection, + Action? ydbOptionsAction = null) + where TContext : DbContext + => (DbContextOptionsBuilder)UseYdb( + (DbContextOptionsBuilder)optionsBuilder, connection, ydbOptionsAction); + // TODO: Right now there are no arguments for constructor, so probably it's ok private static YdbOptionsExtension GetOrCreateExtension(DbContextOptionsBuilder options) => options.Options.FindExtension() ?? new YdbOptionsExtension();