Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adds resolver compiler source generator. #7137

Merged
merged 8 commits into from
Jun 4, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -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>();

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down Expand Up @@ -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 =>
{
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ public DataLoaderSyntaxGenerator()
_writer = new CodeWriter(_sb);
}

public void WriterHeader()
public void WriteHeader()
{
_writer.WriteFileHeader();
_writer.WriteIndentedLine("using Microsoft.Extensions.DependencyInjection;");
Expand Down Expand Up @@ -91,7 +91,7 @@ public void WriteBeginDataLoaderClass(
key.ToFullyQualified(),
value.ToFullyQualified());
break;

case DataLoaderKind.Group:
_writer.WriteIndentedLine(
": global::GreenDonut.GroupedDataLoader<{0}, {1}>",
Expand Down Expand Up @@ -298,4 +298,4 @@ public void Dispose()
_writer = default!;
_disposed = true;
}
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ public ModuleSyntaxGenerator(string moduleName, string ns)
_writer = new CodeWriter(_sb);
}

public void WriterHeader()
public void WriteHeader()
{
_writer.WriteFileHeader();
_writer.WriteLine();
Expand Down
Original file line number Diff line number Diff line change
@@ -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();
}
Expand Down Expand Up @@ -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);
}
}
}

Expand All @@ -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("}");
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ public OperationFieldSyntaxGenerator()
_writer = new CodeWriter(_sb);
}

public void WriterHeader()
public void WriteHeader()
{
_writer.WriteFileHeader();
_writer.WriteLine();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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;");
Expand Down
Original file line number Diff line number Diff line change
@@ -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));
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
namespace HotChocolate.Types.Analyzers.Generators;

public enum ResolverResultKind
{
Task,
Executable,
Queryable,
AsyncEnumerable,
TaskAsyncEnumerable,
Pure,
Invalid
}
Loading
Loading