Merge pull request #12 from PragmaticFlow/11-add-new-column-scenariot…

…imestamp added new column scenarion timestamp
PragmaticFlow · Aug 22, 2024 · 149ce12 · 149ce12
2 parents 5c4ec66 + 2db49c1
commit 149ce12
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 2 deletions.
diff --git a/examples/Demo/Program.cs b/examples/Demo/Program.cs
@@ -43,8 +43,9 @@ public void Run()
             .LoadInfraConfig("infra-config.json")
             .WithReportingInterval(TimeSpan.FromSeconds(5))
             .WithReportingSinks(_timescaleDbSink)
-            .WithTestSuite("reporting")
+            .WithTestSuite("reporting test")
             .WithTestName("timescale_db_demo")
+            .WithSessionId("dasfvxcz214141-wfva1-219")
             .Run();
     }
 }
diff --git a/src/NBomber.Sinks.Timescale/Contracts/ColumnNames.cs b/src/NBomber.Sinks.Timescale/Contracts/ColumnNames.cs
@@ -3,6 +3,7 @@
 public static class ColumnNames
 {
     public const string Time = "time";
+    public const string ScenarioTimestamp = "scenario_timestamp";
     public const string SessionId = "session_id";
     public const string CurrentOperation = "current_operation";
 

diff --git a/src/NBomber.Sinks.Timescale/Contracts/PointDbRecord.cs b/src/NBomber.Sinks.Timescale/Contracts/PointDbRecord.cs
@@ -8,6 +8,7 @@ namespace NBomber.Sinks.Timescale.Contracts;
 public class PointDbRecord
 {
     [Column(ColumnNames.Time)] public DateTime Time { get; set; }
+    [Column(ColumnNames.ScenarioTimestamp)] public TimeSpan ScenarioTimestamp { get; set; }
     [Column(ColumnNames.SessionId)] public string SessionId { get; set; }
     [Column(ColumnNames.CurrentOperation)] public OperationType CurrentOperation { get; set; }
 
@@ -59,7 +60,7 @@ public class PointDbRecord
     [Column(ColumnNames.FailDataP99)] public long FailDataP99 { get; set; }
     [Column(ColumnNames.FailStatusCodes)] public string FailStatusCodes { get; set; }
     [Column(ColumnNames.FailLatencyCount)] public string FailLatencyCount { get; set; }
-    
+
     [Column(ColumnNames.SimulationValue)] public int SimulationValue { get; set; }
 }
 

diff --git a/src/NBomber.Sinks.Timescale/SqlQueries.cs b/src/NBomber.Sinks.Timescale/SqlQueries.cs
@@ -13,6 +13,7 @@ public static class SqlQueries
         CREATE TABLE IF NOT EXISTS ""{StepStatsTable}""
         (
             ""{ColumnNames.Time}"" TIMESTAMPTZ NOT NULL,
+            ""{ColumnNames.ScenarioTimestamp}"" TIME WITHOUT TIME ZONE NOT NULL,
             ""{ColumnNames.SessionId}"" TEXT NOT NULL,
             ""{ColumnNames.CurrentOperation}"" TEXT,            
             

diff --git a/src/NBomber.Sinks.Timescale/TimescaleDbSink.cs b/src/NBomber.Sinks.Timescale/TimescaleDbSink.cs
@@ -170,6 +170,7 @@ private PointDbRecord[] MapToPoint(ScenarioStats scnStats, DateTime currentTime)
             .Select(step => new PointDbRecord
             {
                 Time = currentTime,
+                ScenarioTimestamp = scnStats.Duration,
                 SessionId = testInfo.SessionId,
                 CurrentOperation = nodeInfo.CurrentOperation,
                 Scenario = scnStats.ScenarioName,