add TotalMemoryMB metric for query history

Author: rferraton

src/PlanViewer.App/Dialogs/QueryStoreHistoryWindow.axaml

@@ -59,6 +59,7 @@
                         <ComboBoxItem Content="Total Reads" Tag="TotalLogicalReads"/>
                         <ComboBoxItem Content="Total Writes" Tag="TotalLogicalWrites"/>
                         <ComboBoxItem Content="Total Physical Reads" Tag="TotalPhysicalReads"/>
+                        <ComboBoxItem Content="Total Memory (MB)" Tag="TotalMemoryMb"/>
                         <ComboBoxItem Content="Executions" Tag="CountExecutions"/>
                     </ComboBox>
                     <StackPanel Orientation="Horizontal" Spacing="4" VerticalAlignment="Center">
@@ -142,6 +143,7 @@
                 <DataGridTextColumn Header="Total Reads" Binding="{ReflectionBinding TotalLogicalReadsDisplay}" Width="100"/>
                 <DataGridTextColumn Header="Total Writes" Binding="{ReflectionBinding TotalLogicalWritesDisplay}" Width="100"/>
                 <DataGridTextColumn Header="Total Phys Reads" Binding="{ReflectionBinding TotalPhysicalReadsDisplay}" Width="120"/>
+                <DataGridTextColumn Header="Total Memory (MB)" Binding="{ReflectionBinding TotalMemoryMbDisplay}" Width="120"/>
                 <DataGridTextColumn Header="Min DOP" Binding="{ReflectionBinding MinDop}" Width="70"/>
                 <DataGridTextColumn Header="Max DOP" Binding="{ReflectionBinding MaxDop}" Width="70"/>
                 <DataGridTextColumn Header="Last Execution" Binding="{ReflectionBinding LastExecutionLocal}" Width="140"/>

src/PlanViewer.App/Dialogs/QueryStoreHistoryWindow.axaml.cs

@@ -84,7 +84,7 @@ public partial class QueryStoreHistoryWindow : Window
         ["avg-writes"]       = "AvgLogicalWrites",
         ["physical-reads"]   = "TotalPhysicalReads",
         ["avg-physical-reads"] = "AvgPhysicalReads",
-        ["memory"]           = "TotalCpuMs",
+        ["memory"]           = "TotalMemoryMb",
         ["avg-memory"]       = "AvgMemoryMb",
         ["executions"]       = "CountExecutions",
     };
@@ -877,7 +877,8 @@ private void OnChartPointerMoved(object? sender, PointerEventArgs e)
         "TotalLogicalReads"  => row.TotalLogicalReads,
         "TotalLogicalWrites" => row.TotalLogicalWrites,
         "TotalPhysicalReads" => row.TotalPhysicalReads,
-        "CountExecutions"    => row.CountExecutions,
+        "TotalMemoryMb"      => row.TotalMemoryMb,
+		"CountExecutions"    => row.CountExecutions,
         _                    => row.AvgCpuMs,
     };
 

src/PlanViewer.Core/Models/QueryStoreHistoryRow.cs

@@ -23,8 +23,8 @@ public class QueryStoreHistoryRow
     public double TotalLogicalReads { get; set; }
     public double TotalLogicalWrites { get; set; }
     public double TotalPhysicalReads { get; set; }
-
-    public int MinDop { get; set; }
+    public double TotalMemoryMb { get; set; }
+	public int MinDop { get; set; }
     public int MaxDop { get; set; }
     public DateTime? LastExecutionUtc { get; set; }
 
@@ -41,7 +41,8 @@ public class QueryStoreHistoryRow
     public string TotalLogicalReadsDisplay => TotalLogicalReads.ToString("N2");
     public string TotalLogicalWritesDisplay => TotalLogicalWrites.ToString("N2");
     public string TotalPhysicalReadsDisplay => TotalPhysicalReads.ToString("N2");
+    public string TotalMemoryMbDisplay => TotalMemoryMb.ToString("N2");
 
-    public string IntervalStartLocal => TimeDisplayHelper.FormatForDisplay(IntervalStartUtc);
+	public string IntervalStartLocal => TimeDisplayHelper.FormatForDisplay(IntervalStartUtc);
     public string LastExecutionLocal => LastExecutionUtc.HasValue ? TimeDisplayHelper.FormatForDisplay(LastExecutionUtc.Value) : "";
 }

src/PlanViewer.Core/Services/QueryStoreService.cs

@@ -608,7 +608,8 @@ THEN SUM(rs.avg_rowcount * rs.count_executions) / SUM(rs.count_executions)
     SUM(rs.avg_physical_io_reads * rs.count_executions),
     MIN(rs.min_dop),
     MAX(rs.max_dop),
-    MAX(rs.last_execution_time)
+    MAX(rs.last_execution_time),
+    SUM(rs.avg_query_max_used_memory * rs.count_executions)
 FROM sys.query_store_runtime_stats rs
 JOIN sys.query_store_runtime_stats_interval rsi
     ON rs.runtime_stats_interval_id = rsi.runtime_stats_interval_id
@@ -652,7 +653,8 @@ JOIN sys.query_store_query q
                 MinDop = (int)reader.GetInt64(15),
                 MaxDop = (int)reader.GetInt64(16),
                 LastExecutionUtc = reader.IsDBNull(17) ? null : ((DateTimeOffset)reader.GetValue(17)).UtcDateTime,
-            });
+                TotalMemoryMb = reader.GetDouble(18),
+			});
         }
 
         return rows;