MassTransit#5. Fixing dependencies on SqlLockStatementProvider

src/SampleBatch.Components/CustomSqlLockStatementFormatter.cs

@@ -0,0 +1,30 @@
+using MassTransit.EntityFrameworkCoreIntegration;
+using System.Text;
+
+namespace SampleBatch.Components
+{
+    internal class CustomSqlLockStatementFormatter : ILockStatementFormatter
+    {
+        public void Create(StringBuilder sb, string schema, string table)
+        {
+            sb.AppendFormat("SELECT * FROM [{0}].{1} WITH (UPDLOCK, ROWLOCK, SERIALIZABLE) WHERE ", schema, table);
+        }
+
+        public void AppendColumn(StringBuilder sb, int index, string columnName)
+        {
+            if (index == 0)
+                sb.AppendFormat("{0} = @p0", columnName);
+            else
+                sb.AppendFormat(" AND {0} = @p{1}", columnName, index);
+        }
+
+        public void Complete(StringBuilder sb)
+        {
+        }
+
+        public void CreateOutboxStatement(StringBuilder sb, string schema, string table, string columnName)
+        {
+            sb.AppendFormat(@"SELECT TOP 1 * FROM [{0}].{1} WITH (UPDLOCK, ROWLOCK, READPAST) ORDER BY {2}", schema, table, columnName);
+        }
+    }
+}

src/SampleBatch.Components/CustomSqlLockStatementProvider.cs

@@ -7,8 +7,8 @@ public class CustomSqlLockStatementProvider :
     {
         const string DefaultSchemaName = "dbo";
 
-        public CustomSqlLockStatementProvider(string lockStatement)
-            : base(DefaultSchemaName, lockStatement)
+        public CustomSqlLockStatementProvider()
+            : base(DefaultSchemaName, new CustomSqlLockStatementFormatter())
         {
         }
     }

src/SampleBatch.Tests/Integration/BatchStateMachineTests.cs

@@ -47,7 +47,7 @@ public BatchStateMachineTests()
                 db.Database.EnsureCreated();
             }
 
-            _sagaRepository = EntityFrameworkSagaRepository<BatchState>.CreatePessimistic(_dbContextFactory, new CustomSqlLockStatementProvider("select * from {0}.{1} WITH (UPDLOCK, ROWLOCK) WHERE BatchId = @p0"));
+            _sagaRepository = EntityFrameworkSagaRepository<BatchState>.CreatePessimistic(_dbContextFactory, new CustomSqlLockStatementProvider());
             _stateMachine = new BatchStateMachine();
 
             _inMemoryTestHarness = new InMemoryTestHarness();