From bf2232d03c07fb9e671c50c3af578c352d7c411c Mon Sep 17 00:00:00 2001
From: KirillKurdyukov <kurdyukov-kir@ydb.tech>
Date: Fri, 16 May 2025 17:32:45 +0400
Subject: [PATCH] feat: YdbContextOptionsBuilderExtensions.cs added
 UseYdb<Context> for PooledDbContextFactory

---
 .../YdbContextOptionsBuilderExtensions.cs        | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

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<TContext> UseYdb<TContext>(
+        this DbContextOptionsBuilder<TContext> optionsBuilder,
+        string? connectionString,
+        Action<YdbDbContextOptionsBuilder>? ydbOptionsAction = null)
+        where TContext : DbContext
+        => (DbContextOptionsBuilder<TContext>)UseYdb(
+            (DbContextOptionsBuilder)optionsBuilder, connectionString, ydbOptionsAction);
+
+    public static DbContextOptionsBuilder<TContext> UseYdb<TContext>(
+        this DbContextOptionsBuilder<TContext> optionsBuilder,
+        DbConnection connection,
+        Action<YdbDbContextOptionsBuilder>? ydbOptionsAction = null)
+        where TContext : DbContext
+        => (DbContextOptionsBuilder<TContext>)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<YdbOptionsExtension>() ?? new YdbOptionsExtension();