Added CreateFunction to SqlGenerator for SqlServerFunctionStore

Author: stidsborg

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

@@ -105,6 +105,58 @@ DELETE FROM {tablePrefix}_effects
         return stringBuilder.ToString();
     }
 
+    private string? _createFunctionSql;
+
+    public string CreateFunction(
+        StoredId storedId,
+        FlowInstance humanInstanceId,
+        byte[]? param,
+        long leaseExpiration,
+        long? postponeUntil,
+        long timestamp,
+        StoredId? parent,
+        SqlCommand command,
+        string? paramPrefix)
+    {
+        _createFunctionSql ??= @$"
+                INSERT INTO {tablePrefix}(
+                    FlowType, FlowInstance, 
+                    ParamJson, 
+                    Status,
+                    Epoch, 
+                    Expires,
+                    Timestamp,
+                    HumanInstanceId,
+                    Parent
+                )
+                VALUES
+                (
+                    @FlowType, @flowInstance, 
+                    @ParamJson,   
+                    @Status,
+                    0,
+                    @Expires,
+                    @Timestamp,
+                    @HumanInstanceId,
+                    @Parent
+                )";
+
+        var sql = _createFunctionSql;
+        if (paramPrefix != null)
+            sql = sql.Replace("@", $"@{paramPrefix}");
+
+        command.Parameters.AddWithValue($"@{paramPrefix}FlowType", storedId.Type.Value);
+        command.Parameters.AddWithValue($"@{paramPrefix}FlowInstance", storedId.Instance.Value);
+        command.Parameters.AddWithValue($"@{paramPrefix}Status", (int)(postponeUntil == null ? Status.Executing : Status.Postponed));
+        command.Parameters.AddWithValue($"@{paramPrefix}ParamJson", param == null ? SqlBinary.Null : param);
+        command.Parameters.AddWithValue($"@{paramPrefix}Expires", postponeUntil ?? leaseExpiration);
+        command.Parameters.AddWithValue($"@{paramPrefix}HumanInstanceId", humanInstanceId.Value);
+        command.Parameters.AddWithValue($"@{paramPrefix}Timestamp", timestamp);
+        command.Parameters.AddWithValue($"@{paramPrefix}Parent", parent?.Serialize() ?? (object)DBNull.Value);
+
+        return sql;
+    }
+
     private string? _succeedFunctionSql;
     public string SucceedFunction(
         StoredId storedId, byte[]? result, long timestamp, int expectedEpoch,

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

@@ -146,41 +146,21 @@ public async Task<bool> CreateFunction(
         await using var conn = await _connFunc();
 
         try
-        {
-            _createFunctionSql ??= @$"
-                INSERT INTO {_tableName}(
-                    FlowType, FlowInstance, 
-                    ParamJson, 
-                    Status,
-                    Epoch, 
-                    Expires,
-                    Timestamp,
-                    HumanInstanceId,
-                    Parent
-                )
-                VALUES
-                (
-                    @FlowType, @flowInstance, 
-                    @ParamJson,   
-                    @Status,
-                    0,
-                    @Expires,
-                    @Timestamp,
-                    @HumanInstanceId,
-                    @Parent
-                )";
-
-            await using var command = new SqlCommand(_createFunctionSql, conn);
+        { 
+            await using var command = new SqlCommand();
+            command.Connection = conn;
+            command.CommandText = _sqlGenerator.CreateFunction(
+                storedId,
+                humanInstanceId,
+                param,
+                leaseExpiration,
+                postponeUntil,
+                timestamp,
+                parent,
+                command,
+                paramPrefix: null
+            );
 
-            command.Parameters.AddWithValue("@FlowType", storedId.Type.Value);
-            command.Parameters.AddWithValue("@FlowInstance", storedId.Instance.Value);
-            command.Parameters.AddWithValue("@Status", (int) (postponeUntil == null ? Status.Executing : Status.Postponed));
-            command.Parameters.AddWithValue("@ParamJson", param == null ? SqlBinary.Null : param);
-            command.Parameters.AddWithValue("@Expires", postponeUntil ?? leaseExpiration);
-            command.Parameters.AddWithValue("@HumanInstanceId", humanInstanceId.Value);
-            command.Parameters.AddWithValue("@Timestamp", timestamp);
-            command.Parameters.AddWithValue("@Parent", parent?.Serialize() ?? (object) DBNull.Value);
-
             await command.ExecuteNonQueryAsync();
         }
         catch (SqlException sqlException) when (sqlException.Number == SqlError.UNIQUENESS_VIOLATION)