Increase SQL DB command timeouts to 3000 seconds everywhere in the co…

…debase.
giorgi1324 · Nov 15, 2024 · 1a0e937 · 1a0e937
1 parent 3041587
commit 1a0e937
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/src/shared/Dapper/BaseDapperStorage.cs b/src/shared/Dapper/BaseDapperStorage.cs
@@ -2,11 +2,20 @@ namespace SharedContracts.Dapper;
 
 public abstract class BaseDapperStorage(ILogger<IDataStorage> logger, PostgresConfiguration configuration)
 {
+    private NpgsqlConnection CreateNpgsqlConnection()
+    {
+        var connectionStringBuilder = new NpgsqlConnectionStringBuilder(configuration.StorageConnectionString)
+        {
+            CommandTimeout = 3000,
+        };
+
+        return new(connectionStringBuilder.ConnectionString);
+    }
     protected async Task ExecuteCommandAsync(Func<NpgsqlConnection, Task> operation, string errorMessage, CancellationToken cancellationToken = default)
     {
         try
         {
-            await using var connection = new NpgsqlConnection(configuration.StorageConnectionString);
+            await using var connection = CreateNpgsqlConnection();
             await connection.OpenAsync(cancellationToken);
 
             await operation(connection);
@@ -21,7 +30,7 @@ protected async Task<TResult> ExecuteCommandAsync<TResult>(Func<NpgsqlConnection
     {
         try
         {
-            await using var connection = new NpgsqlConnection(configuration.StorageConnectionString);
+            await using var connection = CreateNpgsqlConnection();
             await connection.OpenAsync(cancellationToken);
 
             var result = await operation(connection);
@@ -38,7 +47,7 @@ protected async Task<DapperResult<TResult, TFailure>> ExecuteCommandAsync<TResul
     {
         try
         {
-            await using var connection = new NpgsqlConnection(configuration.StorageConnectionString);
+            await using var connection = CreateNpgsqlConnection();
             await connection.OpenAsync(cancellationToken);
 
             var result = await operation(connection);

diff --git a/src/tissue/Features/DataProcessing/DapperDataStorage.cs b/src/tissue/Features/DataProcessing/DapperDataStorage.cs
@@ -37,7 +37,12 @@ public async Task DeleteTorrentsByInfoHashes(IReadOnlyCollection<string> infoHas
 
     private async Task<NpgsqlConnection> CreateAndOpenConnection(CancellationToken cancellationToken = default)
     {
-        var connection = new NpgsqlConnection(configuration.StorageConnectionString);
+        var connectionStringBuilder = new NpgsqlConnectionStringBuilder(configuration.StorageConnectionString)
+        {
+            CommandTimeout = 3000,
+        };
+
+        var connection = new NpgsqlConnection(connectionStringBuilder.ConnectionString);
         await connection.OpenAsync(cancellationToken);
         return connection;
     }