init commit

Author: KirillKurdyukov

src/Ydb.Sdk/src/Ado/RetryPolicy/YdbRetryPolicyConfig.cs

@@ -15,4 +15,11 @@ public class YdbRetryPolicyConfig
     public int SlowCapBackoffMs { get; init; } = 5_000;
 
     public bool EnableRetryIdempotence { get; init; } = false;
+
+    public override string ToString() => $"MaxAttempt={MaxAttempt};" +
+                                         $"FastBackoffBaseMs={FastBackoffBaseMs};" +
+                                         $"SlowBackoffBaseMs={SlowBackoffBaseMs};" +
+                                         $"FastCapBackoffMs={FastCapBackoffMs};" +
+                                         $"SlowCapBackoffMs={SlowCapBackoffMs};" +
+                                         $"EnableRetryIdempotence={EnableRetryIdempotence}";
 }

src/Ydb.Sdk/src/Ado/RetryPolicy/YdbRetryPolicyExecutor.cs

@@ -1,10 +1,10 @@
 namespace Ydb.Sdk.Ado.RetryPolicy;
 
-internal class YdbRetryPolicyExecutor
+internal sealed class YdbRetryPolicyExecutor
 {
-    private readonly YdbRetryPolicy _retryPolicy;
+    private readonly IRetryPolicy _retryPolicy;
 
-    public YdbRetryPolicyExecutor(YdbRetryPolicy retryPolicy)
+    public YdbRetryPolicyExecutor(IRetryPolicy retryPolicy)
     {
         _retryPolicy = retryPolicy;
     }
@@ -25,15 +25,19 @@ public YdbRetryPolicyExecutor(YdbRetryPolicy retryPolicy)
     ///     first time or after retrying transient failures). If the task fails with a non-transient error or
     ///     the retry limit is reached, the returned task will become faulted and the exception must be observed.
     /// </returns>
-    public virtual Task<TResult> ExecuteAsync<TResult>(Func<CancellationToken, Task<TResult>> operation,
-        CancellationToken cancellationToken = default) => ExecuteImplementationAsync(operation, cancellationToken);
+    public Task<TResult> ExecuteAsync<TResult>(
+        Func<CancellationToken, Task<TResult>> operation,
+        CancellationToken cancellationToken = default
+    ) => ExecuteImplementationAsync(operation, cancellationToken);
 
-    public async Task ExecuteAsync(Func<CancellationToken, Task> operation, CancellationToken cancellationToken = new())
-        => await ExecuteImplementationAsync(async ct =>
-        {
-            await operation(ct).ConfigureAwait(false);
-            return 0;
-        }, cancellationToken).ConfigureAwait(false);
+    public async Task ExecuteAsync(
+        Func<CancellationToken, Task> operation,
+        CancellationToken cancellationToken = default
+    ) => await ExecuteImplementationAsync(async ct =>
+    {
+        await operation(ct).ConfigureAwait(false);
+        return 0;
+    }, cancellationToken).ConfigureAwait(false);
 
     private async Task<TResult> ExecuteImplementationAsync<TResult>(
         Func<CancellationToken, Task<TResult>> operation,
@@ -44,7 +48,7 @@ CancellationToken cancellationToken
         while (true)
         {
             cancellationToken.ThrowIfCancellationRequested();
-            
+
             try
             {
                 return await operation(cancellationToken).ConfigureAwait(false);
@@ -54,7 +58,7 @@ CancellationToken cancellationToken
                 var delay = _retryPolicy.GetNextDelay(e, attempt++);
                 if (delay == null)
                     throw;
-                
+
                 await Task.Delay(delay.Value, cancellationToken).ConfigureAwait(false);
             }
         }

src/Ydb.Sdk/src/Ado/Transaction/TransactionExtensions.cs

@@ -0,0 +1,35 @@
+using Ydb.Query;
+
+namespace Ydb.Sdk.Ado.Transaction;
+
+internal static class TransactionExtensions
+{
+    private static readonly TransactionSettings SerializableRw = new()
+        { SerializableReadWrite = new SerializableModeSettings() };
+
+    private static readonly TransactionSettings SnapshotRo = new()
+        { SnapshotReadOnly = new SnapshotModeSettings() };
+
+    private static readonly TransactionSettings StaleRo = new()
+        { StaleReadOnly = new StaleModeSettings() };
+
+    private static readonly TransactionSettings OnlineRo = new()
+        { OnlineReadOnly = new OnlineModeSettings { AllowInconsistentReads = false } };
+
+    private static readonly TransactionSettings OnlineInconsistentRo = new()
+        { OnlineReadOnly = new OnlineModeSettings { AllowInconsistentReads = true } };
+
+    internal static TransactionSettings TransactionSettings(this TransactionMode mode) =>
+        mode switch
+        {
+            TransactionMode.SerializableRw => SerializableRw,
+            TransactionMode.SnapshotRo => SnapshotRo,
+            TransactionMode.StaleRo => StaleRo,
+            TransactionMode.OnlineRo => OnlineRo,
+            TransactionMode.OnlineInconsistentRo => OnlineInconsistentRo,
+            _ => throw new ArgumentOutOfRangeException(nameof(mode), mode, null)
+        };
+
+    internal static TransactionControl TransactionControl(this TransactionMode mode, bool commit = true) =>
+        new() { BeginTx = mode.TransactionSettings(), CommitTx = commit };
+}

src/Ydb.Sdk/src/Ado/Transaction/TransactionMode.cs

@@ -0,0 +1,12 @@
+// ReSharper disable once CheckNamespace
+namespace Ydb.Sdk.Ado;
+
+public enum TransactionMode
+{
+    SerializableRw,
+    SnapshotRo,
+    StaleRo,
+
+    OnlineRo,
+    OnlineInconsistentRo
+}

src/Ydb.Sdk/src/Ado/YdbConnection.cs

@@ -3,7 +3,7 @@
 using System.Diagnostics.CodeAnalysis;
 using Ydb.Sdk.Ado.BulkUpsert;
 using Ydb.Sdk.Ado.Session;
-using Ydb.Sdk.Services.Query;
+using Ydb.Sdk.Ado.Transaction;
 using static System.Data.IsolationLevel;
 
 namespace Ydb.Sdk.Ado;
@@ -59,14 +59,14 @@ protected override YdbTransaction BeginDbTransaction(IsolationLevel isolationLev
 
         return BeginTransaction(isolationLevel switch
         {
-            Serializable or Unspecified => TxMode.SerializableRw,
+            Serializable or Unspecified => TransactionMode.SerializableRw,
             _ => throw new ArgumentException("Unsupported isolationLevel: " + isolationLevel)
         });
     }
 
     public new YdbTransaction BeginTransaction(IsolationLevel isolationLevel) => BeginDbTransaction(isolationLevel);
 
-    public YdbTransaction BeginTransaction(TxMode txMode = TxMode.SerializableRw)
+    public YdbTransaction BeginTransaction(TransactionMode transactionMode = TransactionMode.SerializableRw)
     {
         ThrowIfConnectionClosed();
 
@@ -77,7 +77,7 @@ public YdbTransaction BeginTransaction(TxMode txMode = TxMode.SerializableRw)
             );
         }
 
-        CurrentTransaction = new YdbTransaction(this, txMode);
+        CurrentTransaction = new YdbTransaction(this, transactionMode);
 
         return CurrentTransaction;
     }

src/Ydb.Sdk/src/Ado/YdbConnectionStringBuilder.cs

@@ -3,6 +3,7 @@
 using System.Security.Cryptography.X509Certificates;
 using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Logging.Abstractions;
+using Ydb.Sdk.Ado.RetryPolicy;
 using Ydb.Sdk.Auth;
 using Ydb.Sdk.Transport;
 
@@ -320,6 +321,10 @@ public int CreateSessionTimeout
 
     public X509Certificate2Collection? ServerCertificates { get; init; }
 
+    public IRetryPolicy RetryPolicy { get; init; } = YdbRetryPolicy.Default;
+
+    internal YdbRetryPolicyExecutor YdbRetryPolicyExecutor => new(RetryPolicy);
+
     private void SaveValue(string propertyName, object? value)
     {
         if (value == null)