From c5f2c47f8a18b9c9cdec11f518adf0a9527f205d Mon Sep 17 00:00:00 2001
From: Maxim Samsonov <m.samsonov@computer.org>
Date: Fri, 17 May 2024 21:07:38 +0300
Subject: [PATCH] Fixed wait state reporting

---
 dkgServiceNode/Controllers/NodesController.cs | 2 ++
 dkgServiceNode/Data/DbEnsure.cs               | 1 +
 2 files changed, 3 insertions(+)

diff --git a/dkgServiceNode/Controllers/NodesController.cs b/dkgServiceNode/Controllers/NodesController.cs
index ce8c6bb..3ab5bc3 100644
--- a/dkgServiceNode/Controllers/NodesController.cs
+++ b/dkgServiceNode/Controllers/NodesController.cs
@@ -172,6 +172,7 @@ internal async Task<ObjectResult> TrToRunningStepTwoConditional(Round? round, No
             }
             
             runner.SetStepTwoWaitingTime(round);
+            await UpdateNodeState(node, (short)stReport.Status, round?.Id);
 
             if (stReport.Data.Length != 0)
             {
@@ -192,6 +193,7 @@ internal async Task<ObjectResult> TrToRunningStepThreeConditional(Round? round,
             }
 
             runner.SetStepThreeWaitingTime(round);
+            await UpdateNodeState(node, (short)stReport.Status, round?.Id);
 
             if (stReport.Data.Length != 0)
             {
diff --git a/dkgServiceNode/Data/DbEnsure.cs b/dkgServiceNode/Data/DbEnsure.cs
index d77cc15..c23ad8a 100644
--- a/dkgServiceNode/Data/DbEnsure.cs
+++ b/dkgServiceNode/Data/DbEnsure.cs
@@ -251,6 +251,7 @@ public static void Ensure(string connectionString)
                 PuVersionUpdate("0.4.2", connection);
                 EnsureVersion("0.4.3", sqlScript_0_4_3, connection);
                 EnsureVersion("0.5.0", sqlScript_0_5_0, connection);
+                PuVersionUpdate("0.5.1", connection);
             }
         }
     }