Adds resolver compiler source generator. (#7137)

ChilliCream · Jun 4, 2024 · 983faa0 · 983faa0
1 parent 08453b5
commit 983faa0
Show file tree

Hide file tree

Showing 86 changed files with 2,156 additions and 439 deletions.
diff --git a/...olate/Core/src/Execution/DependencyInjection/InternalSchemaServiceCollectionExtensions.cs b/...olate/Core/src/Execution/DependencyInjection/InternalSchemaServiceCollectionExtensions.cs
@@ -39,7 +39,7 @@ internal static IServiceCollection TryAddDiagnosticEvents(
         });
         return services;
     }
-    
+
     public static T GetApplicationService<T>(this IServiceProvider services) where T : notnull
         => services.GetApplicationServices().GetRequiredService<T>();
 

diff --git a/...late/Core/src/Execution/DependencyInjection/RequestExecutorServiceCollectionExtensions.cs b/...late/Core/src/Execution/DependencyInjection/RequestExecutorServiceCollectionExtensions.cs
@@ -9,6 +9,7 @@
 using HotChocolate.Execution.Options;
 using HotChocolate.Execution.Processing;
 using HotChocolate.Fetching;
+using HotChocolate.Internal;
 using HotChocolate.Language;
 using HotChocolate.Resolvers;
 using Microsoft.Extensions.DependencyInjection.Extensions;
@@ -38,6 +39,7 @@ public static IServiceCollection AddGraphQLCore(this IServiceCollection services
         services.TryAddSingleton<DefaultRequestContextAccessor>();
         services.TryAddSingleton<IRequestContextAccessor>(sp => sp.GetRequiredService<DefaultRequestContextAccessor>());
         services.TryAddSingleton<AggregateServiceScopeInitializer>();
+        services.TryAddSingleton<IParameterBindingResolver, DefaultParameterBindingResolver>();
 
         services.TryAddSingleton<ObjectPool<StringBuilder>>(sp =>
         {

diff --git a/src/HotChocolate/Core/src/Types.Analyzers/Generators/DataLoaderSyntaxGenerator.cs b/src/HotChocolate/Core/src/Types.Analyzers/Generators/DataLoaderSyntaxGenerator.cs
@@ -18,7 +18,7 @@ public DataLoaderSyntaxGenerator()
         _writer = new CodeWriter(_sb);
     }
 
-    public void WriterHeader()
+    public void WriteHeader()
     {
         _writer.WriteFileHeader();
         _writer.WriteIndentedLine("using Microsoft.Extensions.DependencyInjection;");
@@ -91,7 +91,7 @@ public void WriteBeginDataLoaderClass(
                     key.ToFullyQualified(),
                     value.ToFullyQualified());
                 break;
-            
+
             case DataLoaderKind.Group:
                 _writer.WriteIndentedLine(
                     ": global::GreenDonut.GroupedDataLoader<{0}, {1}>",
@@ -298,4 +298,4 @@ public void Dispose()
         _writer = default!;
         _disposed = true;
     }
-}
+}
diff --git a/src/HotChocolate/Core/src/Types.Analyzers/Generators/ModuleSyntaxGenerator.cs b/src/HotChocolate/Core/src/Types.Analyzers/Generators/ModuleSyntaxGenerator.cs
@@ -20,7 +20,7 @@ public ModuleSyntaxGenerator(string moduleName, string ns)
         _writer = new CodeWriter(_sb);
     }
 
-    public void WriterHeader()
+    public void WriteHeader()
     {
         _writer.WriteFileHeader();
         _writer.WriteLine();

diff --git a/src/HotChocolate/Core/src/Types.Analyzers/Generators/ObjectTypeExtensionSyntaxGenerator.cs b/src/HotChocolate/Core/src/Types.Analyzers/Generators/ObjectTypeExtensionSyntaxGenerator.cs
@@ -1,29 +1,24 @@
 using System.Text;
 using HotChocolate.Types.Analyzers.Helpers;
 using HotChocolate.Types.Analyzers.Inspectors;
+using Microsoft.CodeAnalysis;
 
 namespace HotChocolate.Types.Analyzers.Generators;
 
-public sealed class ObjectTypeExtensionSyntaxGenerator
+public sealed class ObjectTypeExtensionSyntaxGenerator(StringBuilder sb, string ns)
 {
-    private readonly string _ns;
-    private readonly CodeWriter _writer;
+    private readonly CodeWriter _writer = new(sb);
 
-    public ObjectTypeExtensionSyntaxGenerator(StringBuilder sb, string ns)
-    {
-        _ns = ns;
-        _writer = new(sb);
-    }
-
-    public void WriterHeader()
+    public void WriteHeader()
     {
         _writer.WriteFileHeader();
+        _writer.WriteIndentedLine("using Microsoft.Extensions.DependencyInjection;");
         _writer.WriteLine();
     }
 
     public void WriteBeginNamespace()
     {
-        _writer.WriteIndentedLine("namespace {0}", _ns);
+        _writer.WriteIndentedLine("namespace {0}", ns);
         _writer.WriteIndentedLine("{");
         _writer.IncreaseIndent();
     }
@@ -58,14 +53,17 @@ public void WriteInitializeMethod(ObjectTypeExtensionInfo objectTypeExtension)
 
         using (_writer.IncreaseIndent())
         {
-            _writer.WriteIndentedLine("const global::System.Reflection.BindingFlags bindingFlags =");
-            using (_writer.IncreaseIndent())
+            if (objectTypeExtension.Members.Length > 0)
             {
-                _writer.WriteIndentedLine("global::System.Reflection.BindingFlags.Public |");
+                _writer.WriteIndentedLine("const global::{0} bindingFlags =", WellKnownTypes.BindingFlags);
                 using (_writer.IncreaseIndent())
                 {
-                    _writer.WriteIndentedLine("System.Reflection.BindingFlags.NonPublic |");
-                    _writer.WriteIndentedLine("System.Reflection.BindingFlags.Static;");
+                    _writer.WriteIndentedLine("global::{0}.Public", WellKnownTypes.BindingFlags);
+                    using (_writer.IncreaseIndent())
+                    {
+                        _writer.WriteIndentedLine("| global::{0}.NonPublic", WellKnownTypes.BindingFlags);
+                        _writer.WriteIndentedLine("| global::{0}.Static;", WellKnownTypes.BindingFlags);
+                    }
                 }
             }
 
@@ -89,17 +87,62 @@ public void WriteInitializeMethod(ObjectTypeExtensionInfo objectTypeExtension)
 
             if (objectTypeExtension.Members.Length > 0)
             {
-                _writer.WriteLine();
                 foreach (var member in objectTypeExtension.Members)
                 {
-                    _writer.WriteIndentedLine(
-                        "descriptor.Field(thisType.GetMember(\"{0}\", bindingFlags)[0]);",
-                        member.Name);
+                    _writer.WriteLine();
+                    _writer.WriteIndentedLine("descriptor");
+
+                    using (_writer.IncreaseIndent())
+                    {
+                        _writer.WriteIndentedLine(
+                            ".Field(thisType.GetMember(\"{0}\", bindingFlags)[0])",
+                            member.Name);
+
+                        if (member is IMethodSymbol method &&
+                            method.GetResultKind() is not ResolverResultKind.Pure)
+                        {
+                            _writer.WriteIndentedLine(
+                                ".Extend().Definition.Resolver = {0}Resolvers.{1}_{2};",
+                                objectTypeExtension.Type.ToDisplayString(),
+                                objectTypeExtension.Type.Name,
+                                member.Name);
+                        }
+                        else
+                        {
+                            _writer.WriteIndentedLine(
+                                ".Extend().Definition.PureResolver = {0}Resolvers.{1}_{2};",
+                                objectTypeExtension.Type.ToDisplayString(),
+                                objectTypeExtension.Type.Name,
+                                member.Name);
+                        }
+                    }
                 }
             }
 
             _writer.WriteLine();
             _writer.WriteIndentedLine("Configure(descriptor);");
+
+            if (objectTypeExtension.Members.Length > 0)
+            {
+                _writer.WriteLine();
+                _writer.WriteIndentedLine("descriptor.Extend().Context.OnSchemaCreated(");
+                using (_writer.IncreaseIndent())
+                {
+                    _writer.WriteIndentedLine("schema =>");
+                    _writer.WriteIndentedLine("{");
+                    using (_writer.IncreaseIndent())
+                    {
+                        _writer.WriteIndentedLine("var services = schema.Services.GetApplicationServices();");
+                        _writer.WriteIndentedLine(
+                            "var bindingResolver = services.GetRequiredService<global::{0}>();",
+                            WellKnownTypes.ParameterBindingResolver);
+                        _writer.WriteIndentedLine(
+                            "global::{0}Resolvers.InitializeBindings(bindingResolver);",
+                            objectTypeExtension.Type.ToDisplayString());
+                    }
+                    _writer.WriteIndentedLine("});");
+                }
+            }
         }
 
         _writer.WriteIndentedLine("}");

diff --git a/src/HotChocolate/Core/src/Types.Analyzers/Generators/OperationFieldSyntaxGenerator.cs b/src/HotChocolate/Core/src/Types.Analyzers/Generators/OperationFieldSyntaxGenerator.cs
@@ -18,7 +18,7 @@ public OperationFieldSyntaxGenerator()
         _writer = new CodeWriter(_sb);
     }
 
-    public void WriterHeader()
+    public void WriteHeader()
     {
         _writer.WriteFileHeader();
         _writer.WriteLine();

diff --git a/src/HotChocolate/Core/src/Types.Analyzers/Generators/RequestMiddlewareSyntaxGenerator.cs b/src/HotChocolate/Core/src/Types.Analyzers/Generators/RequestMiddlewareSyntaxGenerator.cs
@@ -25,7 +25,7 @@ public RequestMiddlewareSyntaxGenerator(string moduleName, string ns)
         _writer = new(_sb);
     }
 
-    public void WriterHeader()
+    public void WriteHeader()
     {
         _writer.WriteFileHeader();
         _writer.WriteIndentedLine("using Microsoft.Extensions.DependencyInjection;");

diff --git a/src/HotChocolate/Core/src/Types.Analyzers/Generators/ResolverName.cs b/src/HotChocolate/Core/src/Types.Analyzers/Generators/ResolverName.cs
@@ -0,0 +1,23 @@
+using Microsoft.CodeAnalysis;
+
+namespace HotChocolate.Types.Analyzers.Generators;
+
+public readonly struct ResolverName(string typeName, string memberName)
+{
+    public readonly string TypeName = typeName;
+
+    public readonly string MemberName = memberName;
+}
+
+public readonly struct ResolverInfo(ResolverName resolverName, IMethodSymbol? methodSymbol)
+{
+    public readonly ResolverName Name = resolverName;
+
+    public readonly IMethodSymbol? Method = methodSymbol;
+
+    public readonly int ParameterCount = methodSymbol?.Parameters.Length ?? 0;
+
+    public readonly bool Skip =>
+        ParameterCount == 0 ||
+        (ParameterCount == 1 && (Method?.Parameters[0]?.IsParent() ?? false));
+}
diff --git a/src/HotChocolate/Core/src/Types.Analyzers/Generators/ResolverResultKind.cs b/src/HotChocolate/Core/src/Types.Analyzers/Generators/ResolverResultKind.cs
@@ -0,0 +1,12 @@
+namespace HotChocolate.Types.Analyzers.Generators;
+
+public enum ResolverResultKind
+{
+    Task,
+    Executable,
+    Queryable,
+    AsyncEnumerable,
+    TaskAsyncEnumerable,
+    Pure,
+    Invalid
+}