fix example

Author: KirillKurdyukov

examples/src/EntityFrameworkCore.Ydb.QuickStart/Program.cs

@@ -1,5 +1,6 @@
 using EntityFrameworkCore.Ydb.Extensions;
 using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Design;
 
 await using var db = new BloggingContext();
 
@@ -26,14 +27,35 @@
 db.Remove(blog);
 await db.SaveChangesAsync();
 
+internal class BloggingContextFactory : IDesignTimeDbContextFactory<BloggingContext>
+{
+    public BloggingContext CreateDbContext(string[] args)
+    {
+        var optionsBuilder = new DbContextOptionsBuilder<BloggingContext>();
+
+        return new BloggingContext(
+            optionsBuilder.UseYdb("Host=localhost;Port=2136;Database=/local",
+                builder => builder.DisableRetryOnFailure()
+            ).Options
+        );
+    }
+}
+
 internal class BloggingContext : DbContext
 {
+    internal BloggingContext()
+    {
+    }
+
+    internal BloggingContext(DbContextOptions<BloggingContext> options) : base(options)
+    {
+    }
+
     public DbSet<Blog> Blogs { get; set; }
     public DbSet<Post> Posts { get; set; }
 
-    protected override void OnConfiguring(DbContextOptionsBuilder options)
-        => options.UseYdb("Host=localhost;Port=2136;Database=/local")
-            .LogTo(Console.WriteLine);
+    protected override void OnConfiguring(DbContextOptionsBuilder options) =>
+        options.UseYdb("Host=localhost;Port=2136;Database=/local");
 }
 
 internal class Blog

src/EFCore.Ydb/src/Infrastructure/Internal/YdbOptionsExtension.cs

@@ -13,6 +13,8 @@ public class YdbOptionsExtension : RelationalOptionsExtension
 
     public X509Certificate2Collection? ServerCertificates { get; private set; }
 
+    public bool DisableRetryExecutionStrategy { get; private set; }
+
     private DbContextOptionsExtensionInfo? _info;
 
     public YdbOptionsExtension()
@@ -23,11 +25,13 @@ private YdbOptionsExtension(YdbOptionsExtension copyFrom) : base(copyFrom)
     {
         CredentialsProvider = copyFrom.CredentialsProvider;
         ServerCertificates = copyFrom.ServerCertificates;
+        DisableRetryExecutionStrategy = copyFrom.DisableRetryExecutionStrategy;
     }
 
     protected override RelationalOptionsExtension Clone() => new YdbOptionsExtension(this);
 
-    public override void ApplyServices(IServiceCollection services) => services.AddEntityFrameworkYdb();
+    public override void ApplyServices(IServiceCollection services) => 
+        services.AddEntityFrameworkYdb(!DisableRetryExecutionStrategy);
 
     public override DbContextOptionsExtensionInfo Info => _info ??= new ExtensionInfo(this);
 
@@ -49,6 +53,15 @@ public YdbOptionsExtension WithServerCertificates(X509Certificate2Collection? se
         return clone;
     }
 
+    public YdbOptionsExtension DisableRetryOnFailure()
+    {
+        var clone = (YdbOptionsExtension)Clone();
+        
+        clone.DisableRetryExecutionStrategy = true;
+
+        return clone;
+    }
+
     private sealed class ExtensionInfo(YdbOptionsExtension extension) : RelationalExtensionInfo(extension)
     {
         public override bool IsDatabaseProvider => true;

src/EFCore.Ydb/src/Infrastructure/YdbDbContextOptionsBuilder.cs

@@ -14,4 +14,7 @@ public YdbDbContextOptionsBuilder WithCredentialsProvider(ICredentialsProvider?
 
     public YdbDbContextOptionsBuilder WithServerCertificates(X509Certificate2Collection? serverCertificates) =>
         WithOption(optionsBuilder => optionsBuilder.WithServerCertificates(serverCertificates));
+    
+    public YdbDbContextOptionsBuilder DisableRetryOnFailure() =>
+        WithOption(optionsBuilder => optionsBuilder.DisableRetryOnFailure());
 }

src/EFCore.Ydb/src/Storage/Internal/YdbRelationalConnection.cs

@@ -13,6 +13,7 @@ public class YdbRelationalConnection : RelationalConnection, IYdbRelationalConne
 {
     private readonly ICredentialsProvider? _credentialsProvider;
     private readonly X509Certificate2Collection? _serverCertificates;
+    private readonly bool _disableRetryExecuteStrategy;
 
     public YdbRelationalConnection(RelationalConnectionDependencies dependencies) : base(dependencies)
     {
@@ -21,14 +22,15 @@ public YdbRelationalConnection(RelationalConnectionDependencies dependencies) :
 
         _credentialsProvider = ydbOptionsExtension.CredentialsProvider;
         _serverCertificates = ydbOptionsExtension.ServerCertificates;
+        _disableRetryExecuteStrategy = ydbOptionsExtension.DisableRetryExecutionStrategy;
     }
 
     protected override DbConnection CreateDbConnection()
     {
         var ydbConnectionStringBuilder = new YdbConnectionStringBuilder(GetValidatedConnectionString())
         {
             CredentialsProvider = _credentialsProvider,
-            ServerCertificates = _serverCertificates
+            ServerCertificates = _serverCertificates,
         };
 
         return new YdbConnection(ydbConnectionStringBuilder);
@@ -40,6 +42,10 @@ public IYdbRelationalConnection Clone()
         {
             builder.WithCredentialsProvider(_credentialsProvider);
             builder.WithServerCertificates(_serverCertificates);
+            if (_disableRetryExecuteStrategy)
+            {
+                builder.DisableRetryOnFailure();
+            }
         }).Options;
 
         return new YdbRelationalConnection(Dependencies with { ContextOptions = options });