RDB-136 Flow tabel wordt niet meer opgeruimd Bij een exceptie in Flow

Flow ruimt nu zijn locks goed op en de weggeschreven state ingeval van exceptions tijdens het afhandelen van de message

Tapeti.Flow/Default/FlowCleanupMiddleware.cs

@@ -0,0 +1,29 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Tapeti.Config;
+
+namespace Tapeti.Flow.Default
+{
+    public class FlowCleanupMiddleware : ICleanupMiddleware
+    {
+        public async Task Handle(IMessageContext context, ConsumeResponse response)
+        {
+            object flowContextObj;
+            if (!context.Items.TryGetValue(ContextItems.FlowContext, out flowContextObj))
+                return;
+            var flowContext = (FlowContext)flowContextObj;
+
+            if (flowContext.FlowStateLock != null)
+            {
+                if (response == ConsumeResponse.Nack)
+                {
+                    await flowContext.FlowStateLock.DeleteFlowState();
+                }
+                flowContext.FlowStateLock.Dispose();
+            }
+        }
+    }
+}

Tapeti.Flow/FlowMiddleware.cs

@@ -24,7 +24,8 @@ namespace Tapeti.Flow
 
         public IEnumerable<object> GetMiddleware(IDependencyResolver dependencyResolver)
         {
-            return new[] { new FlowBindingMiddleware() };
+            yield return new FlowBindingMiddleware();
+            yield return new FlowCleanupMiddleware();
         }
     }
 }

Tapeti.Flow/Tapeti.Flow.csproj

@@ -54,6 +54,7 @@
     <Compile Include="Annotations\ContinuationAttribute.cs" />
     <Compile Include="Annotations\StartAttribute.cs" />
     <Compile Include="ContextItems.cs" />
+    <Compile Include="Default\FlowCleanupMiddleware.cs" />
     <Compile Include="Default\FlowMessageFilterMiddleware.cs" />
     <Compile Include="Default\FlowBindingMiddleware.cs" />
     <Compile Include="Default\FlowContext.cs" />

Tapeti/Connection/TapetiConsumer.cs

@@ -129,6 +129,8 @@ namespace Tapeti.Connection
             if (message == null)
                 throw new ArgumentException("Empty message");
 
+            context.Message = message;
+
             var validMessageType = false;
 
             foreach (var binding in bindings)