SemianiakaVY/elsa-core
1
0
Fork 0
Code Issues Pull requests Actions 1 Packages Projects Releases Wiki Activity

Added DeleteVariablesAsync method for the workflow context

Browse source
This commit is contained in:
Marius Vasile Vușcan 2025-01-13 16:48:24 +02:00
parent b2331f73db
commit e365d44637
No known key found for this signature in database
GPG key ID: E0EC75631D902E06
2 changed files with 33 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/modules/Elsa.Workflows.Core/Contracts/IVariablePersistenceManager.cs
View file

@ -19,7 +19,12 @@ public interface IVariablePersistenceManager
Task SaveVariablesAsync(WorkflowExecutionContext context);
/// <summary>
/// Deletes the specified variables from the <see cref="WorkflowExecutionContext"/>.
/// Deletes the specified variables from the <see cref="ActivityExecutionContext"/>.
/// </summary>
Task DeleteVariablesAsync(ActivityExecutionContext context);
/// <summary>
/// Deletes the specified variables from the <see cref="WorkflowExecutionContext"/>.
/// </summary>
Task DeleteVariablesAsync(WorkflowExecutionContext context);
}

27
src/modules/Elsa.Workflows.Core/Services/VariablePersistenceManager.cs
View file

@ -120,6 +120,33 @@ public class VariablePersistenceManager(IStorageDriverManager storageDriverManag
}
}
/// <inheritdoc />
public async Task DeleteVariablesAsync(WorkflowExecutionContext context)
{
var cancellationToken = context.CancellationTokens.ApplicationCancellationToken;
var activityContexts = context.ActivityExecutionContexts.ToList();
foreach (var activityContext in activityContexts)
{
var variables = GetLocalVariables(activityContext).ToList();
foreach (var variable in variables)
{
var block = variable.GetBlock(activityContext.ExpressionExecutionContext);
var metadata = (VariableBlockMetadata)block.Metadata!;
var driver = _storageDriverManager.Get(metadata.StorageDriverType!);
if (driver == null)
continue;
var id = GetStateId(variable);
var storageDriverContext = new StorageDriverContext(activityContext, variable, cancellationToken);
await driver.DeleteAsync(id, storageDriverContext);
}
}
}
private IEnumerable<Variable> GetLocalVariables(IExecutionContext context) => context.Variables;
private MemoryBlock EnsureBlock(MemoryRegister register, Variable variable)