From 50891e5a6b1948d98da172b6d49973cb5cc781de Mon Sep 17 00:00:00 2001
From: Chris Midgley <midgleyc@gmail.com>
Date: Fri, 21 Jan 2022 22:32:38 +0000
Subject: [PATCH] Add overload `EnableRetryOnFailure`. (#2206)

Allow specifying `errorCodesToAdd` without count or delay.

Fixes #2048.
---
 .../Infrastructure/NpgsqlDbContextOptionsBuilder.cs  | 10 ++++++++++
 src/EFCore.PG/NpgsqlRetryingExecutionStrategy.cs     | 12 ++++++++++++
 2 files changed, 22 insertions(+)

diff --git a/src/EFCore.PG/Infrastructure/NpgsqlDbContextOptionsBuilder.cs b/src/EFCore.PG/Infrastructure/NpgsqlDbContextOptionsBuilder.cs
index 3f9526d9..394eb81a 100644
--- a/src/EFCore.PG/Infrastructure/NpgsqlDbContextOptionsBuilder.cs
+++ b/src/EFCore.PG/Infrastructure/NpgsqlDbContextOptionsBuilder.cs
@@ -151,6 +151,16 @@ public virtual NpgsqlDbContextOptionsBuilder EnableRetryOnFailure()
     public virtual NpgsqlDbContextOptionsBuilder EnableRetryOnFailure(int maxRetryCount)
         => ExecutionStrategy(c => new NpgsqlRetryingExecutionStrategy(c, maxRetryCount));
 
+    /// <summary>
+    /// Configures the context to use the default retrying <see cref="IExecutionStrategy" />.
+    /// </summary>
+    /// <param name="errorCodesToAdd">Additional error codes that should be considered transient.</param>
+    /// <returns>
+    /// An instance of <see cref="NpgsqlDbContextOptionsBuilder"/> with the specified parameters.
+    /// </returns>
+    public virtual NpgsqlDbContextOptionsBuilder EnableRetryOnFailure(ICollection<string>? errorCodesToAdd)
+        => ExecutionStrategy(c => new NpgsqlRetryingExecutionStrategy(c, errorCodesToAdd));
+
     /// <summary>
     /// Configures the context to use the default retrying <see cref="IExecutionStrategy" />.
     /// </summary>
diff --git a/src/EFCore.PG/NpgsqlRetryingExecutionStrategy.cs b/src/EFCore.PG/NpgsqlRetryingExecutionStrategy.cs
index e33909f9..0e2d9dc0 100644
--- a/src/EFCore.PG/NpgsqlRetryingExecutionStrategy.cs
+++ b/src/EFCore.PG/NpgsqlRetryingExecutionStrategy.cs
@@ -53,6 +53,18 @@ public class NpgsqlRetryingExecutionStrategy : ExecutionStrategy
     {
     }
 
+    /// <summary>
+    ///     Creates a new instance of <see cref="NpgsqlRetryingExecutionStrategy" />.
+    /// </summary>
+    /// <param name="dependencies"> Parameter object containing service dependencies. </param>
+    /// <param name="errorCodesToAdd"> Additional error codes that should be considered transient. </param>
+    public NpgsqlRetryingExecutionStrategy(
+        ExecutionStrategyDependencies dependencies,
+        ICollection<string>? errorCodesToAdd)
+        : this(dependencies, DefaultMaxRetryCount, DefaultMaxDelay, errorCodesToAdd)
+    {
+    }
+
     /// <summary>
     ///     Creates a new instance of <see cref="NpgsqlRetryingExecutionStrategy" />.
     /// </summary>
-- 
GitLab