Use Output-clause in SqlServerFunctionStore's RestartExecution-method

Author: stidsborg

Stores/SqlServer/Cleipnir.ResilientFunctions.SqlServer/SqlGenerator.cs

@@ -367,20 +367,17 @@ public StoreCommand RestartExecution(StoredId storedId, int expectedEpoch, long
                 Status = {(int)Status.Executing}, 
                 Expires = @LeaseExpiration,
                 Interrupted = 0
-            WHERE FlowType = @FlowType AND FlowInstance = @FlowInstance AND Epoch = @ExpectedEpoch;
-
-            SELECT ParamJson,                
-                   Status,
-                   ResultJson, 
-                   ExceptionJson,                   
-                   Expires,
-                   Epoch,
-                   Interrupted,
-                   Timestamp,
-                   HumanInstanceId,
-                   Parent
-            FROM {tablePrefix}
-            WHERE FlowType = @FlowType AND FlowInstance = @FlowInstance";
+            OUTPUT inserted.ParamJson,                
+                   inserted.Status,
+                   inserted.ResultJson, 
+                   inserted.ExceptionJson,                   
+                   inserted.Expires,
+                   inserted.Epoch,
+                   inserted.Interrupted,
+                   inserted.Timestamp,
+                   inserted.HumanInstanceId,
+                   inserted.Parent
+            WHERE FlowType = @FlowType AND FlowInstance = @FlowInstance AND Epoch = @ExpectedEpoch;";
 
         var storeCommand = StoreCommand.Create(_restartExecutionSql);
         storeCommand.AddParameter("@LeaseExpiration", leaseExpiration);

Stores/SqlServer/Cleipnir.ResilientFunctions.SqlServer/SqlServerFunctionStore.cs

@@ -259,9 +259,6 @@ WHEN NOT MATCHED THEN
             .ToSqlCommand(conn);
 
         await using var reader = await command.ExecuteReaderAsync();
-        if (reader.RecordsAffected != 1)
-            return null;
-        
         var sf = _sqlGenerator.ReadToStoredFlow(storedId, reader);
         if (sf?.Epoch != expectedEpoch + 1)
             return null;