Made Cost Metrics Class Public

src/HotChocolate/CostAnalysis/src/CostAnalysis/CostMetrics.cs

@@ -1,7 +1,7 @@
 namespace HotChocolate.CostAnalysis;
 
 /// <summary>https://ibm.github.io/graphql-specs/cost-spec.html#sec-__cost</summary>
-internal sealed class CostMetrics
+public sealed class CostMetrics
 {
     /// <summary>https://ibm.github.io/graphql-specs/cost-spec.html#sec-Field-Cost</summary>
     public double FieldCost { get; set; } = 0;

src/HotChocolate/CostAnalysis/src/CostAnalysis/Extensions/CostAnalyzerRequestContextExtensions.cs

@@ -2,9 +2,12 @@
 
 namespace HotChocolate.Execution;
 
-internal static class CostAnalyzerRequestContextExtensions
+/// <summary>
+/// Cost Analyzer extensions for the <see cref="IRequestContext"/>.
+/// </summary>
+public static class CostAnalyzerRequestContextExtensions
 {
-    public static IRequestContext AddCostMetrics(
+    internal static IRequestContext AddCostMetrics(
         this IRequestContext context,
         CostMetrics costMetrics)
     {
@@ -22,7 +25,36 @@ public static IRequestContext AddCostMetrics(
         return context;
     }
 
-    public static CostAnalyzerMode GetCostAnalyzerMode(
+    /// <summary>
+    /// Gets the cost metrics from the context.
+    /// </summary>
+    /// <param name="context">
+    /// The request context.
+    /// </param>
+    /// <returns>
+    /// Returns the cost metrics.
+    /// </returns>
+    /// <exception cref="ArgumentNullException">
+    /// <paramref name="context"/> is <c>null</c>.
+    /// </exception>
+    public static CostMetrics GetCostMetrics(
+        this IRequestContext context)
+    {
+        if (context is null)
+        {
+            throw new ArgumentNullException(nameof(context));
+        }
+
+        if (context.ContextData.TryGetValue(WellKnownContextData.CostMetrics, out var value) &&
+            value is CostMetrics costMetrics)
+        {
+            return costMetrics;
+        }
+
+        return new CostMetrics();
+    }
+
+    internal static CostAnalyzerMode GetCostAnalyzerMode(
         this IRequestContext context,
         CostOptions options)
     {