Add retry to GetSession in SessionPool.cs

ydb-platform · Sep 28, 2023 · c389d3e · c389d3e
1 parent 78d5353
commit c389d3e
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/Ydb.Sdk/src/Services/Table/SessionPool.cs b/src/Ydb.Sdk/src/Services/Table/SessionPool.cs
@@ -91,6 +91,20 @@ public SessionPool(Driver driver, SessionPoolConfig config)
     }
 
     public async Task<GetSessionResponse> GetSession()
+    {
+        const int maxAttempts = 100;
+
+        GetSessionResponse getSessionResponse = null!;
+        for (var attempt = 0; attempt < maxAttempts; attempt++)
+        {
+            getSessionResponse = await GetSessionAttempt();
+        }
+
+        _logger.LogError($"Failed to get session from pool or create it (attempts: {maxAttempts})");
+        return getSessionResponse;
+    }
+
+    private async Task<GetSessionResponse> GetSessionAttempt()
     {
         lock (_lock)
         {