fixed DB consistency

PragmaticFlow · Sep 17, 2024 · 36f9c5e · 36f9c5e
1 parent 6750369
commit 36f9c5e
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 4 deletions.
diff --git a/src/NBomber.Sinks.Timescale/NBomber.Sinks.Timescale.csproj b/src/NBomber.Sinks.Timescale/NBomber.Sinks.Timescale.csproj
@@ -21,7 +21,7 @@
 
     <ItemGroup>      
       <PackageReference Include="NBomber.Contracts" Version="5.5.0" />
-      <PackageReference Include="Npgsql" Version="8.0.3" />
+      <PackageReference Include="Npgsql" Version="8.0.4" />
       <PackageReference Include="RepoDb.PostgreSql" Version="1.13.1" />
       <PackageReference Include="RepoDb.PostgreSql.BulkOperations" Version="1.13.1" />
     </ItemGroup>

diff --git a/src/NBomber.Sinks.Timescale/TimescaleDbSink.cs b/src/NBomber.Sinks.Timescale/TimescaleDbSink.cs
@@ -24,6 +24,7 @@ public class TimescaleDbSink : IReportingSink
     private IBaseContext _context;
     private NpgsqlConnection _mainConnection;
     private string _connectionString = "";
+    private bool _dbError = false;
 
     public string SinkName => "NBomber.Sinks.TimescaleDb";
 
@@ -83,8 +84,31 @@ public async Task Start()
                 TestName = testInfo.TestName,
                 NodeInfo = Json.serialize(nodeInfo)
             };
-
-            await _mainConnection.BinaryBulkInsertAsync(SqlQueries.SessionsTable, new [] { record });
+
+            var text = @$"INSERT INTO {SqlQueries.SessionsTable} 
+                        (""{ColumnNames.Time}"",
+                        ""{ColumnNames.SessionId}"",
+                        ""{ColumnNames.CurrentOperation}"",
+                        ""{ColumnNames.TestSuite}"",
+                        ""{ColumnNames.TestName}"",
+                        ""{ColumnNames.NodeInfo}"")
+                        VALUES 
+                        ('{record.Time}',
+                        '{record.SessionId}',
+                        '{record.CurrentOperation}',
+                        '{record.TestSuite}',
+                        '{record.TestName}',
+                        '{record.NodeInfo}'::jsonb)";
+            try
+            {
+                await _mainConnection.ExecuteNonQueryAsync(text);
+            }
+            catch (Exception ex) 
+            {
+                _dbError = true;
+                _logger.Error(ex, ex.Message);
+                throw ex;
+            }
         }
     }
 
@@ -102,7 +126,7 @@ public void Dispose()
 
     private async Task SaveScenarioStats(ScenarioStats[] stats, bool isFinal = false)
     {
-        if (_mainConnection != null)
+        if (_mainConnection != null && !_dbError)
         {
             var currentTime = DateTime.UtcNow;