Tagging activities in internal node agent work with "wolverine_node_a…

…ssignments"
JasperFx · Aug 19, 2024 · 16d6e62 · GeeSuth · Aug 21, 2024 · 16d6e62
1 parent 85122c2
commit 16d6e62
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/src/Wolverine/Runtime/Agents/NodeAgentController.EvaluateAssignments.cs b/src/Wolverine/Runtime/Agents/NodeAgentController.EvaluateAssignments.cs
@@ -9,6 +9,8 @@ public partial class NodeAgentController
     // Tested w/ integration tests all the way
     public async Task<AgentCommands> EvaluateAssignmentsAsync(IReadOnlyList<WolverineNode> nodes)
     {
+        using var activity = WolverineTracing.ActivitySource.StartActivity("wolverine_node_assignments");
+
         var grid = new AssignmentGrid();
 
         var capabilities = nodes.SelectMany(x => x.Capabilities).Distinct().ToArray();

diff --git a/src/Wolverine/Runtime/Agents/NodeAgentController.HeartBeat.cs b/src/Wolverine/Runtime/Agents/NodeAgentController.HeartBeat.cs
@@ -30,6 +30,8 @@ public async Task<AgentCommands> DoHealthChecksAsync()
         {
             return AgentCommands.Empty;
         }
+
+        using var activity = WolverineTracing.ActivitySource.StartActivity("wolverine_node_assignments");
 
         // write health check regardless
         await _persistence.MarkHealthCheckAsync(_runtime.Options.UniqueNodeId);

diff --git a/src/Wolverine/Runtime/Agents/NodeAgentController.StartLocally.cs b/src/Wolverine/Runtime/Agents/NodeAgentController.StartLocally.cs
@@ -8,6 +8,8 @@ public partial class NodeAgentController
 
     public async Task StartSoloModeAsync()
     {
+        using var activity = WolverineTracing.ActivitySource.StartActivity("wolverine_node_assignments");
+
         await _runtime.Storage.Nodes.ClearAllAsync(_cancellation.Token);
         await _runtime.Storage.Admin.ReleaseAllOwnershipAsync();