Added Operation Plan Formatter (#8475)

Author: michaelstaib

src/HotChocolate/Core/src/Execution.Abstractions/Execution/ExecutionResult.cs

@@ -1,3 +1,5 @@
+using HotChocolate.Features;
+
 namespace HotChocolate.Execution;
 
 /// <summary>
@@ -23,6 +25,9 @@ protected ExecutionResult(Func<ValueTask>[] cleanupTasks)
     /// <inheritdoc cref="IExecutionResult" />
     public abstract IReadOnlyDictionary<string, object?>? ContextData { get; }
 
+    /// <inheritdoc cref="IFeatureProvider" />
+    public IFeatureCollection Features { get; } = new FeatureCollection();
+
     private protected Func<ValueTask>[] CleanupTasks => _cleanupTasks;
 
     /// <inheritdoc cref="IExecutionResult" />

src/HotChocolate/Core/src/Execution.Abstractions/Execution/IExecutionResult.cs

@@ -1,3 +1,5 @@
+using HotChocolate.Features;
+
 namespace HotChocolate.Execution;
 
 /// <summary>
@@ -8,7 +10,7 @@ namespace HotChocolate.Execution;
 /// them allows it to give back its used memory to the execution
 /// engine result pools.
 /// </remarks>
-public interface IExecutionResult : IAsyncDisposable
+public interface IExecutionResult : IFeatureProvider, IAsyncDisposable
 {
     /// <summary>
     /// Gets the result kind.

src/HotChocolate/Core/src/Features/FeatureCollection.cs

@@ -179,8 +179,8 @@ public IEnumerator<KeyValuePair<Type, object>> GetEnumerator()
 
     private sealed class KeyComparer : IEqualityComparer<KeyValuePair<Type, object>>
     {
-        public bool Equals(KeyValuePair<Type, object> x, KeyValuePair<Type, object> y) =>
-            x.Key.Equals(y.Key);
+        public bool Equals(KeyValuePair<Type, object> x, KeyValuePair<Type, object> y)
+            => x.Key.Equals(y.Key);
 
         public int GetHashCode(KeyValuePair<Type, object> obj) =>
             obj.Key.GetHashCode();

src/HotChocolate/Fusion-vnext/src/Fusion.AspNetCore/DependencyInjection/FusionServerServiceCollectionExtensions.cs

@@ -65,6 +65,35 @@ private static IFusionGatewayBuilder AddGraphQLGatewayServerCore(
         return builder;
     }
 
+    public static IFusionGatewayBuilder AddHttpRequestInterceptor<T>(
+        this IFusionGatewayBuilder builder)
+        where T : IHttpRequestInterceptor, new()
+    {
+        ArgumentNullException.ThrowIfNull(builder);
+
+        return builder.ConfigureSchemaServices(
+            (_, s) =>
+            {
+                s.RemoveAll<IHttpRequestInterceptor>();
+                s.AddSingleton<IHttpRequestInterceptor>(new T());
+            });
+    }
+
+    public static IFusionGatewayBuilder AddHttpRequestInterceptor(
+        this IFusionGatewayBuilder builder,
+        Func<IServiceProvider, IHttpRequestInterceptor> factory)
+    {
+        ArgumentNullException.ThrowIfNull(builder);
+        ArgumentNullException.ThrowIfNull(factory);
+
+        return builder.ConfigureSchemaServices(
+            (_, s) =>
+            {
+                s.RemoveAll<IHttpRequestInterceptor>();
+                s.AddSingleton(factory);
+            });
+    }
+
     private static IFusionGatewayBuilder AddDefaultHttpRequestInterceptor(
         this IFusionGatewayBuilder builder)
         => builder.ConfigureSchemaServices(

src/HotChocolate/Fusion-vnext/src/Fusion.Execution/Execution/Extensions/FusionRequestContextExtensions.cs

@@ -19,7 +19,7 @@ public static class FusionRequestContextExtensions
     /// The request context.
     /// </param>
     /// <returns>
-    /// The <see cref="OperationExecutionPlan"/> if it exists, otherwise <c>null</c>.
+    /// The <see cref="OperationPlan"/> if it exists, otherwise <c>null</c>.
     /// </returns>
     public static string GetOperationId(
         this RequestContext context)
@@ -37,15 +37,15 @@ public static string GetOperationId(
     }
 
     /// <summary>
-    /// Gets the <see cref="OperationExecutionPlan"/> from the request context.
+    /// Gets the <see cref="OperationPlan"/> from the request context.
     /// </summary>
     /// <param name="context">
     /// The request context.
     /// </param>
     /// <returns>
-    /// The <see cref="OperationExecutionPlan"/> if it exists, otherwise <c>null</c>.
+    /// The <see cref="OperationPlan"/> if it exists, otherwise <c>null</c>.
     /// </returns>
-    public static OperationExecutionPlan? GetOperationPlan(
+    public static OperationPlan? GetOperationPlan(
         this RequestContext context)
     {
         ArgumentNullException.ThrowIfNull(context);
@@ -73,7 +73,7 @@ public static void SetOperationId(
     }
 
     /// <summary>
-    /// Sets the <see cref="OperationExecutionPlan"/> on the request context.
+    /// Sets the <see cref="OperationPlan"/> on the request context.
     /// </summary>
     /// <param name="context">
     /// The request context.
@@ -83,7 +83,7 @@ public static void SetOperationId(
     /// </param>
     public static void SetOperationPlan(
         this RequestContext context,
-        OperationExecutionPlan plan)
+        OperationPlan plan)
     {
         ArgumentNullException.ThrowIfNull(context);
         ArgumentNullException.ThrowIfNull(plan);

src/HotChocolate/Fusion-vnext/src/Fusion.Execution/Execution/FusionOperationInfo.cs

@@ -7,7 +7,7 @@ internal sealed class FusionOperationInfo : RequestFeature
 {
     public string? OperationId { get; set; }
 
-    public OperationExecutionPlan? OperationPlan { get; set; }
+    public OperationPlan? OperationPlan { get; set; }
 
     protected override void Reset()
     {

src/HotChocolate/Fusion-vnext/src/Fusion.Execution/Execution/FusionRequestExecutorManager.cs

@@ -300,7 +300,7 @@ private static void AddOperationPlanner(IServiceCollection services)
             static sp =>
             {
                 var options = sp.GetRequiredService<ISchemaDefinition>().GetRequestOptions();
-                return new Cache<OperationExecutionPlan>(
+                return new Cache<OperationPlan>(
                     options.OperationExecutionPlanCacheSize,
                     options.OperationExecutionPlanCacheDiagnostics);
             });

src/HotChocolate/Fusion-vnext/src/Fusion.Execution/Execution/FusionRequestOptions.cs

@@ -10,6 +10,7 @@ public sealed class FusionRequestOptions : ICloneable
     private CacheDiagnostics? _operationExecutionPlanCacheDiagnostics;
     private int _operationDocumentCacheSize = 256;
     private int _sourceSchemaOperationCacheSize = 256;
+    private bool _collectOperationPlanTelemetry;
     private bool _isReadOnly;
 
     /// <summary>
@@ -99,6 +100,20 @@ public int SourceSchemaOperationCacheSize
         }
     }
 
+    public bool CollectOperationPlanTelemetry
+    {
+        get => _collectOperationPlanTelemetry;
+        set
+        {
+            if (_isReadOnly)
+            {
+                throw new InvalidOperationException("The request options are read-only.");
+            }
+
+            _collectOperationPlanTelemetry = value;
+        }
+    }
+
     /// <summary>
     /// Clones the request options into a new mutable instance.
     /// </summary>
@@ -111,6 +126,9 @@ public FusionRequestOptions Clone()
         clone._executionTimeout = _executionTimeout;
         clone._operationExecutionPlanCacheSize = _operationExecutionPlanCacheSize;
         clone._operationExecutionPlanCacheDiagnostics = _operationExecutionPlanCacheDiagnostics;
+        clone._operationDocumentCacheSize = _operationDocumentCacheSize;
+        clone._sourceSchemaOperationCacheSize = _sourceSchemaOperationCacheSize;
+        clone._collectOperationPlanTelemetry = _collectOperationPlanTelemetry;
         return clone;
     }
 

src/HotChocolate/Fusion-vnext/src/Fusion.Execution/Execution/Nodes/ExecutionNodeResult.cs

@@ -1,3 +1,10 @@
+using System.Diagnostics;
+
 namespace HotChocolate.Fusion.Execution.Nodes;
 
-public record ExecutionNodeResult(int Id, ExecutionStatus Status, TimeSpan Duration);
+public record ExecutionNodeResult(
+    int Id,
+    Activity? Activity,
+    ExecutionStatus Status,
+    TimeSpan Duration,
+    Exception? Exception = null);

src/HotChocolate/Fusion-vnext/src/Fusion.Execution/Execution/Nodes/IntrospectionExecutionNode.cs

@@ -9,12 +9,23 @@ public sealed class IntrospectionExecutionNode : ExecutionNode
 
     public IntrospectionExecutionNode(int id, Selection[] selections)
     {
+        ArgumentNullException.ThrowIfNull(selections);
+
+        if (selections.Length == 0)
+        {
+            throw new ArgumentException(
+                "There must be at least one introspection selection.",
+                nameof(selections));
+        }
+
         Id = id;
         _selections = selections;
     }
 
     public override int Id { get; }
 
+    public ReadOnlySpan<Selection> Selections => _selections;
+
     public override ReadOnlySpan<ExecutionNode> Dependencies => default;
 
     public override Task<ExecutionNodeResult> ExecuteAsync(
@@ -43,7 +54,12 @@ public override Task<ExecutionNodeResult> ExecuteAsync(
         ExecuteSelections(context, backlog);
         context.AddPartialResults(root, _selections);
 
-        return Task.FromResult(new ExecutionNodeResult(Id, ExecutionStatus.Success, Stopwatch.GetElapsedTime(start)));
+        return Task.FromResult(
+            new ExecutionNodeResult(
+                Id,
+                Activity.Current,
+                ExecutionStatus.Success,
+                Stopwatch.GetElapsedTime(start)));
     }
 
     private static void ExecuteSelections(