Fix spans not hidden on expand/collapse (#6394)

JamesNK · web-flow · commit 029a0f745700 · 2024-10-21T21:13:15.000+08:00
diff --git a/src/Aspire.Dashboard/Components/Pages/TraceDetail.razor.cs b/src/Aspire.Dashboard/Components/Pages/TraceDetail.razor.cs
@@ -261,7 +261,7 @@ private string GetRowClass(SpanWaterfallViewModel viewModel)
 
     public SpanDetailsViewModel? SelectedSpan { get; set; }
 
-    private void OnToggleCollapse(SpanWaterfallViewModel viewModel)
+    private async Task OnToggleCollapse(SpanWaterfallViewModel viewModel)
     {
         // View model data is recreated if the trace updates.
         // Persist the collapsed state in a separate list.
@@ -275,6 +275,8 @@ private void OnToggleCollapse(SpanWaterfallViewModel viewModel)
             viewModel.IsCollapsed = true;
             _collapsedSpanIds.Add(viewModel.Span.SpanId);
         }
+
+        await _dataGrid.SafeRefreshDataAsync();
     }
 
     private async Task OnShowPropertiesAsync(SpanWaterfallViewModel viewModel, string? buttonId)

Original file line number	Diff line number	Diff line change
`@@ -261,7 +261,7 @@ private string GetRowClass(SpanWaterfallViewModel viewModel)`
`261`	`261`
`262`	`262`	`public SpanDetailsViewModel? SelectedSpan { get; set; }`
`263`	`263`
`264`		`- private void OnToggleCollapse(SpanWaterfallViewModel viewModel)`
	`264`	`+ private async Task OnToggleCollapse(SpanWaterfallViewModel viewModel)`
`265`	`265`	`{`
`266`	`266`	`// View model data is recreated if the trace updates.`
`267`	`267`	`// Persist the collapsed state in a separate list.`
`@@ -275,6 +275,8 @@ private void OnToggleCollapse(SpanWaterfallViewModel viewModel)`
`275`	`275`	`viewModel.IsCollapsed = true;`
`276`	`276`	`_collapsedSpanIds.Add(viewModel.Span.SpanId);`
`277`	`277`	`}`
	`278`	`+`
	`279`	`+ await _dataGrid.SafeRefreshDataAsync();`
`278`	`280`	`}`
`279`	`281`
`280`	`282`	`private async Task OnShowPropertiesAsync(SpanWaterfallViewModel viewModel, string? buttonId)`