Add MiniProfiler Fields Middleware

Author: rasmusjp

src/Our.Umbraco.GraphQL/Instrumentation/MiniProfilerFieldsMiddleware.cs

@@ -0,0 +1,18 @@
+using GraphQL.Instrumentation;
+using GraphQL.Types;
+using StackExchange.Profiling;
+using System.Threading.Tasks;
+
+namespace Our.Umbraco.GraphQL.Instrumentation
+{
+    public class MiniProfilerFieldsMiddleware
+    {
+        public Task<object> Resolve(ResolveFieldContext context, FieldMiddlewareDelegate next)
+        {
+            using (MiniProfiler.Current.Step($"[GraphQL] Resolving {string.Join(".", context.Path)}", ProfileLevel.Info))
+            {
+                return next(context);
+            }
+        }
+    }
+}

src/Our.Umbraco.GraphQL/Our.Umbraco.GraphQL.csproj

@@ -251,6 +251,7 @@
     <Compile Include="Filters\NotFilter.cs" />
     <Compile Include="Filters\OrFilter.cs" />
     <Compile Include="Filters\StartsWithFilter.cs" />
+    <Compile Include="Instrumentation\MiniProfilerFieldsMiddleware.cs" />
     <Compile Include="Schema\Conventions.cs" />
     <Compile Include="Types\ConnectionExtensions.cs" />
     <Compile Include="Types\ConnectionWithFilterBuilder.cs" />

src/Our.Umbraco.GraphQL/Web/GraphQLMiddleware.cs

@@ -8,6 +8,10 @@
 using GraphQL.Instrumentation;
 using GraphQL.Utilities;
 using Microsoft.Owin;
+using Newtonsoft.Json;
+using Newtonsoft.Json.Linq;
+using Newtonsoft.Json.Serialization;
+using Our.Umbraco.GraphQL.Instrumentation;
 using Our.Umbraco.GraphQL.Schema;
 using StackExchange.Profiling;
 using Umbraco.Core;
@@ -75,7 +79,7 @@ public override async Task Invoke(IOwinContext context)
                         Inputs variables = requestParams.Variables;
 
                         var start = DateTime.Now;
-
+                        MiniProfiler.Start();
                         var result = await _documentExecutor
                             .ExecuteAsync(x =>
                             {
@@ -86,6 +90,7 @@ public override async Task Invoke(IOwinContext context)
                                 {
                                     x.EnableMetrics = true;
                                     x.FieldMiddleware.Use<InstrumentFieldsMiddleware>();
+                                    x.FieldMiddleware.Use<MiniProfilerFieldsMiddleware>();
                                 }
                                 x.FieldNameConverter = new DefaultFieldNameConverter();
                                 x.Inputs = variables;
@@ -104,7 +109,14 @@ public override async Task Invoke(IOwinContext context)
                         if (_options.EnableMetrics && result.Errors == null)
                         {
                             result.EnrichWithApolloTracing(start);
+
+                            if (result.Extensions == null)
+                            {
+                                result.Extensions = new Dictionary<string, object>();
+                            }
+                            result.Extensions["miniProfiler"] = JObject.FromObject(MiniProfiler.Current, new JsonSerializer { ContractResolver = new CamelCasePropertyNamesContractResolver() });
                         }
+                        MiniProfiler.Stop();
 
                         return result;
                     });