26 lines
843 B
C#
26 lines
843 B
C#
using System.Threading.Tasks;
|
|
using Tapeti.Config;
|
|
|
|
namespace Tapeti.Flow.Default
|
|
{
|
|
public class FlowCleanupMiddleware : ICleanupMiddleware
|
|
{
|
|
public async Task Handle(IMessageContext context, HandlingResult handlingResult)
|
|
{
|
|
if (!context.Items.TryGetValue(ContextItems.FlowContext, out var flowContextObj))
|
|
return;
|
|
var flowContext = (FlowContext)flowContextObj;
|
|
|
|
if (flowContext?.FlowStateLock != null)
|
|
{
|
|
if (handlingResult.ConsumeResponse == ConsumeResponse.Nack
|
|
|| handlingResult.MessageAction == MessageAction.ErrorLog)
|
|
{
|
|
await flowContext.FlowStateLock.DeleteFlowState();
|
|
}
|
|
flowContext.FlowStateLock.Dispose();
|
|
}
|
|
}
|
|
}
|
|
}
|