Move FlowsManager thread notification into Effect.RunParallelle

Author: stidsborg

Core/Cleipnir.ResilientFunctions/CoreRuntime/Invocation/Workflow.cs

@@ -108,15 +108,7 @@ public Task<T> Message<T>(Func<T, bool> filter, TimeSpan? maxWait = null) where
 
     public Task AppendMessage(object msg, string? idempotencyKey = null) => MessageWriter.AppendMessage(msg, idempotencyKey);
 
-    public Task<T> Parallelle<T>(Func<Task<T>> work)
-    {
-        _flowsManager.StartThread(StoredId);
-        return Effect.RunParallelle(work).ContinueWith(t =>
-        {
-            _flowsManager.CompleteThread(StoredId);
-            return t.GetAwaiter().GetResult();
-        });
-    }
+    public Task<T> Parallelle<T>(Func<Task<T>> work) => Effect.RunParallelle(work);
 
     public string ExecutionTree()
     {

Core/Cleipnir.ResilientFunctions/Domain/Effect.cs

@@ -276,5 +276,14 @@ public async Task<TSeed> AggregateEach<T, TSeed>(
 
     internal string ExecutionTree() => EffectPrinter.Print(effectResults);
 
-    public Task<T> RunParallelle<T>(Func<Task<T>> work) => Capture(() => Task.Run(work));
+    public Task<T> RunParallelle<T>(Func<Task<T>> work)
+    {
+        flowsManager.StartThread(storedId);
+        var task = Capture(() => Task.Run(work));
+        return task.ContinueWith(t =>
+        {
+            flowsManager.CompleteThread(storedId);
+            return t.GetAwaiter().GetResult();
+        });
+    }
 }