Skip to content

Feat/Add stream invoker provider support to CQRS handler registry generator #307

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.

Sign up for GitHub

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

Merged
GeWuYou merged 14 commits into from
May 1, 2026
Merged

Feat/Add stream invoker provider support to CQRS handler registry generator #307

Show file tree
Hide file tree
Changes from 12 commits
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
9847706
docs(cqrs): 补充生成式 stream invoker 文档语义
GeWuYou Apr 30, 2026
f17f9f3
test(cqrs): 补充 stream invoker provider 生成断言
GeWuYou Apr 30, 2026
ea0b937
feat(cqrs): 补充生成式 stream invoker 接缝
GeWuYou Apr 30, 2026
172c081
test(cqrs): 补充 hidden implementation provider 元数据断言
GeWuYou Apr 30, 2026
eb30388
feat(cqrs): 扩大生成式 invoker 发射范围
GeWuYou Apr 30, 2026
5a77e2f
test(cqrs-tests): 补充 hidden implementation generated invoker 回归
GeWuYou Apr 30, 2026
6b5c5d9
docs(cqrs): 更新生成式 invoker 恢复点
GeWuYou Apr 30, 2026
dc21188
test(cqrs): 锁定 precise reflected provider 边界
GeWuYou Apr 30, 2026
502f652
test(cqrs): 补充 provider gate 合同回归
GeWuYou Apr 30, 2026
1091594
fix(cqrs): 收敛 generated invoker 异常语义
GeWuYou Apr 30, 2026
8b36626
test(cqrs): 补充 provider fallback 回归
GeWuYou Apr 30, 2026
8352874
fix(cqrs): 收敛生成调用描述符与PR评审回归
GeWuYou Apr 30, 2026
9296def
test(cqrs): 补齐 stream invoker gate 回归
GeWuYou Apr 30, 2026
26314db
docs(cqrs-rewrite): 收敛恢复入口文档
GeWuYou Apr 30, 2026
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
26 changes: 22 additions & 4 deletions GFramework.Cqrs.SourceGenerators/Cqrs/CqrsHandlerRegistryGenerator.Models.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,16 +9,23 @@ private readonly record struct RequestInvokerRegistrationSpec(
string RequestTypeDisplayName,
string ResponseTypeDisplayName);

private readonly record struct StreamInvokerRegistrationSpec(
string RequestTypeDisplayName,
string ResponseTypeDisplayName);

private readonly record struct HandlerRegistrationSpec(
string HandlerInterfaceDisplayName,
string ImplementationTypeDisplayName,
string HandlerInterfaceLogName,
string ImplementationLogName,
RequestInvokerRegistrationSpec? RequestInvokerRegistration);
RequestInvokerRegistrationSpec? RequestInvokerRegistration,
StreamInvokerRegistrationSpec? StreamInvokerRegistration);

private readonly record struct ReflectedImplementationRegistrationSpec(
string HandlerInterfaceDisplayName,
string HandlerInterfaceLogName);
string HandlerInterfaceLogName,
RequestInvokerRegistrationSpec? RequestInvokerRegistration,
StreamInvokerRegistrationSpec? StreamInvokerRegistration);

private readonly record struct OrderedRegistrationSpec(
string HandlerInterfaceLogName,
Expand All @@ -31,14 +38,18 @@ private readonly record struct GeneratedRegistrySourceShape(
bool HasReflectionTypeLookups,
bool HasExternalAssemblyTypeLookups,
bool SupportsRequestInvokerProvider,
ImmutableArray<RequestInvokerEmissionSpec> RequestInvokerEmissions)
ImmutableArray<RequestInvokerEmissionSpec> RequestInvokerEmissions,
bool SupportsStreamInvokerProvider,
ImmutableArray<StreamInvokerEmissionSpec> StreamInvokerEmissions)
{
public bool RequiresRegistryAssemblyVariable =>
HasReflectedImplementationRegistrations ||
HasPreciseReflectedRegistrations ||
HasReflectionTypeLookups;

public bool HasRequestInvokerProvider => SupportsRequestInvokerProvider && !RequestInvokerEmissions.IsDefaultOrEmpty;

public bool HasStreamInvokerProvider => SupportsStreamInvokerProvider && !StreamInvokerEmissions.IsDefaultOrEmpty;
}

private readonly record struct RequestInvokerEmissionSpec(
Expand All @@ -47,6 +58,12 @@ private readonly record struct RequestInvokerEmissionSpec(
string HandlerInterfaceDisplayName,
int MethodIndex);

private readonly record struct StreamInvokerEmissionSpec(
string RequestTypeDisplayName,
string ResponseTypeDisplayName,
string HandlerInterfaceDisplayName,
int MethodIndex);

/// <summary>
/// 标记某条 handler 注册语句在生成阶段采用的表达策略。
/// </summary>
Expand Down Expand Up @@ -328,5 +345,6 @@ private readonly record struct GenerationEnvironment(
bool SupportsNamedReflectionFallbackTypes,
bool SupportsDirectReflectionFallbackTypes,
bool SupportsMultipleReflectionFallbackAttributes,
bool SupportsRequestInvokerProvider);
bool SupportsRequestInvokerProvider,
bool SupportsStreamInvokerProvider);
}
209 changes: 204 additions & 5 deletions GFramework.Cqrs.SourceGenerators/Cqrs/CqrsHandlerRegistryGenerator.SourceEmission.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -57,26 +57,32 @@ private static GeneratedRegistrySourceShape CreateGeneratedRegistrySourceShape(
var requestInvokerEmissions = CreateRequestInvokerEmissions(
generationEnvironment.SupportsRequestInvokerProvider,
registrations);
var streamInvokerEmissions = CreateStreamInvokerEmissions(
generationEnvironment.SupportsStreamInvokerProvider,
registrations);

return new GeneratedRegistrySourceShape(
hasReflectedImplementationRegistrations,
hasPreciseReflectedRegistrations,
hasReflectionTypeLookups,
hasExternalAssemblyTypeLookups,
generationEnvironment.SupportsRequestInvokerProvider,
requestInvokerEmissions);
requestInvokerEmissions,
generationEnvironment.SupportsStreamInvokerProvider,
streamInvokerEmissions);
}

/// <summary>
/// 从 direct handler 注册描述中提取 request invoker 发射计划。
/// 从可直接表达 handler 接口的注册描述中提取 request invoker 发射计划。
/// </summary>
/// <param name="supportsRequestInvokerProvider">
/// 指示当前 runtime 是否同时暴露 <c>ICqrsRequestInvokerProvider</c> 与
/// <c>IEnumeratesCqrsRequestInvokerDescriptors</c> 契约；若不支持，则本方法必须返回空结果并让后续发射路径整体跳过。
/// </param>
/// <param name="registrations">已按稳定顺序整理完成的 handler 注册描述。</param>
/// <returns>
/// 由 <c>directRegistration.RequestInvokerRegistration</c> 派生出的 <see cref="RequestInvokerEmissionSpec" /> 集合。
/// 由 direct registration 或 reflected-implementation registration 上的
/// <c>RequestInvokerRegistration</c> 派生出的 <see cref="RequestInvokerEmissionSpec" /> 集合。
/// <c>methodIndex</c> 按 <paramref name="registrations" /> 与其 direct registration 的遍历顺序单调递增，
/// 因而只要上游排序稳定，生成的 invoker 方法名与描述符顺序就跨运行保持稳定。
/// </returns>
Expand Down Expand Up @@ -106,6 +112,71 @@ private static ImmutableArray<RequestInvokerEmissionSpec> CreateRequestInvokerEm
directRegistration.HandlerInterfaceDisplayName,
methodIndex++));
}

foreach (var reflectedRegistration in registration.ReflectedImplementationRegistrations)
{
if (reflectedRegistration.RequestInvokerRegistration is not { } requestInvokerRegistration)
continue;

builder.Add(new RequestInvokerEmissionSpec(
requestInvokerRegistration.RequestTypeDisplayName,
requestInvokerRegistration.ResponseTypeDisplayName,
reflectedRegistration.HandlerInterfaceDisplayName,
methodIndex++));
}
}

return builder.ToImmutable();
}

/// <summary>
/// 从可直接表达 handler 接口的注册描述中提取 stream invoker 发射计划。
/// </summary>
/// <param name="supportsStreamInvokerProvider">
/// 指示当前 runtime 是否同时暴露 <c>ICqrsStreamInvokerProvider</c> 与
/// <c>IEnumeratesCqrsStreamInvokerDescriptors</c> 契约；若不支持，则本方法必须返回空结果并让后续发射路径整体跳过。
/// </param>
/// <param name="registrations">已按稳定顺序整理完成的 handler 注册描述。</param>
/// <returns>
/// 由 direct registration 或 reflected-implementation registration 上的
/// <c>StreamInvokerRegistration</c> 派生出的 <see cref="StreamInvokerEmissionSpec" /> 集合。
/// <c>methodIndex</c> 按 <paramref name="registrations" /> 与其 direct registration 的遍历顺序单调递增，
/// 因而只要上游排序稳定，生成的 invoker 方法名与描述符顺序就跨运行保持稳定。
/// </returns>
private static ImmutableArray<StreamInvokerEmissionSpec> CreateStreamInvokerEmissions(
bool supportsStreamInvokerProvider,
IReadOnlyList<ImplementationRegistrationSpec> registrations)
{
if (!supportsStreamInvokerProvider)
return ImmutableArray<StreamInvokerEmissionSpec>.Empty;

var builder = ImmutableArray.CreateBuilder<StreamInvokerEmissionSpec>();
var methodIndex = 0;
foreach (var registration in registrations)
{
foreach (var directRegistration in registration.DirectRegistrations)
{
if (directRegistration.StreamInvokerRegistration is not { } streamInvokerRegistration)
continue;

builder.Add(new StreamInvokerEmissionSpec(
streamInvokerRegistration.RequestTypeDisplayName,
streamInvokerRegistration.ResponseTypeDisplayName,
directRegistration.HandlerInterfaceDisplayName,
methodIndex++));
}

foreach (var reflectedRegistration in registration.ReflectedImplementationRegistrations)
{
if (reflectedRegistration.StreamInvokerRegistration is not { } streamInvokerRegistration)
continue;

builder.Add(new StreamInvokerEmissionSpec(
streamInvokerRegistration.RequestTypeDisplayName,
streamInvokerRegistration.ResponseTypeDisplayName,
reflectedRegistration.HandlerInterfaceDisplayName,
methodIndex++));
}
}

return builder.ToImmutable();
Expand Down Expand Up @@ -221,6 +292,15 @@ private static void AppendGeneratedRegistryType(
builder.Append(".IEnumeratesCqrsRequestInvokerDescriptors");
}

if (sourceShape.HasStreamInvokerProvider)
{
builder.Append(", global::");
builder.Append(CqrsRuntimeNamespace);
builder.Append(".ICqrsStreamInvokerProvider, global::");
builder.Append(CqrsRuntimeNamespace);
builder.Append(".IEnumeratesCqrsStreamInvokerDescriptors");
}

builder.AppendLine();
builder.AppendLine("{");
AppendRegisterMethod(builder, registrations, sourceShape);
Expand All @@ -231,6 +311,12 @@ private static void AppendGeneratedRegistryType(
AppendRequestInvokerProviderMembers(builder, sourceShape.RequestInvokerEmissions);
}

if (sourceShape.HasStreamInvokerProvider)
{
builder.AppendLine();
AppendStreamInvokerProviderMembers(builder, sourceShape.StreamInvokerEmissions);
}

if (sourceShape.HasExternalAssemblyTypeLookups)
{
builder.AppendLine();
Expand Down Expand Up @@ -366,9 +452,11 @@ private static void AppendRequestInvokerDescriptorArray(
/// </remarks>
private static void AppendRequestInvokerProviderMethods(StringBuilder builder)
{
builder.Append(" public global::System.Collections.Generic.IReadOnlyList<global::");
builder.Append(" global::System.Collections.Generic.IReadOnlyList<global::");
builder.Append(CqrsRuntimeNamespace);
builder.Append(".CqrsRequestInvokerDescriptorEntry> global::");
builder.Append(CqrsRuntimeNamespace);
builder.AppendLine(".CqrsRequestInvokerDescriptorEntry> GetDescriptors()");
builder.AppendLine(".IEnumeratesCqrsRequestInvokerDescriptors.GetDescriptors()");
builder.AppendLine(" {");
builder.AppendLine(" return RequestInvokerDescriptors;");
builder.AppendLine(" }");
Expand Down Expand Up @@ -424,6 +512,117 @@ private static void AppendRequestInvokerMethod(StringBuilder builder, RequestInv
builder.AppendLine(" }");
}

/// <summary>
/// 发射 generated registry 的 stream invoker provider 成员。
/// </summary>
/// <param name="builder">生成源码构造器。</param>
/// <param name="streamInvokerEmissions">当前要输出的 stream invoker 发射计划。</param>
private static void AppendStreamInvokerProviderMembers(
StringBuilder builder,
ImmutableArray<StreamInvokerEmissionSpec> streamInvokerEmissions)
{
AppendStreamInvokerDescriptorArray(builder, streamInvokerEmissions);
builder.AppendLine();
AppendStreamInvokerProviderMethods(builder);

for (var index = 0; index < streamInvokerEmissions.Length; index++)
{
builder.AppendLine();
AppendStreamInvokerMethod(builder, streamInvokerEmissions[index]);
}
}

/// <summary>
/// 发射 generated registry 的 stream invoker 描述符数组。
/// </summary>
private static void AppendStreamInvokerDescriptorArray(
StringBuilder builder,
ImmutableArray<StreamInvokerEmissionSpec> streamInvokerEmissions)
{
builder.AppendLine(" private static readonly global::GFramework.Cqrs.CqrsStreamInvokerDescriptorEntry[] StreamInvokerDescriptors =");
builder.AppendLine(" [");

for (var index = 0; index < streamInvokerEmissions.Length; index++)
{
var emission = streamInvokerEmissions[index];
builder.Append(" new global::");
builder.Append(CqrsRuntimeNamespace);
builder.Append(".CqrsStreamInvokerDescriptorEntry(typeof(");
builder.Append(emission.RequestTypeDisplayName);
builder.Append("), typeof(");
builder.Append(emission.ResponseTypeDisplayName);
builder.Append("), new global::");
builder.Append(CqrsRuntimeNamespace);
builder.Append(".CqrsStreamInvokerDescriptor(typeof(");
builder.Append(emission.HandlerInterfaceDisplayName);
builder.Append("), typeof(");
builder.Append(GeneratedTypeName);
builder.Append(").GetMethod(nameof(InvokeStreamHandler");
builder.Append(emission.MethodIndex);
builder.Append("), global::System.Reflection.BindingFlags.NonPublic | global::System.Reflection.BindingFlags.Static)!))");
builder.AppendLine(index == streamInvokerEmissions.Length - 1 ? string.Empty : ",");
}

builder.AppendLine(" ];");
}

/// <summary>
/// 发射 generated registry 对 stream invoker provider 契约的实现方法。
/// </summary>
private static void AppendStreamInvokerProviderMethods(StringBuilder builder)
{
builder.Append(" global::System.Collections.Generic.IReadOnlyList<global::");
builder.Append(CqrsRuntimeNamespace);
builder.Append(".CqrsStreamInvokerDescriptorEntry> global::");
builder.Append(CqrsRuntimeNamespace);
builder.AppendLine(".IEnumeratesCqrsStreamInvokerDescriptors.GetDescriptors()");
builder.AppendLine(" {");
builder.AppendLine(" return StreamInvokerDescriptors;");
builder.AppendLine(" }");
builder.AppendLine();
builder.Append(" public bool TryGetDescriptor(global::System.Type requestType, global::System.Type responseType, out global::");
builder.Append(CqrsRuntimeNamespace);
builder.AppendLine(".CqrsStreamInvokerDescriptor? descriptor)");
builder.AppendLine(" {");
builder.AppendLine(" if (requestType is null)");
builder.AppendLine(" throw new global::System.ArgumentNullException(nameof(requestType));");
builder.AppendLine(" if (responseType is null)");
builder.AppendLine(" throw new global::System.ArgumentNullException(nameof(responseType));");
builder.AppendLine();
builder.AppendLine(" foreach (var entry in StreamInvokerDescriptors)");
builder.AppendLine(" {");
builder.AppendLine(" if (entry.RequestType == requestType && entry.ResponseType == responseType)");
builder.AppendLine(" {");
builder.AppendLine(" descriptor = entry.Descriptor;");
builder.AppendLine(" return true;");
builder.AppendLine(" }");
builder.AppendLine(" }");
builder.AppendLine();
builder.AppendLine(" descriptor = null;");
builder.AppendLine(" return false;");
builder.AppendLine(" }");
}

/// <summary>
/// 为单个 stream invoker 描述符发射对应的静态强类型桥接方法。
/// </summary>
private static void AppendStreamInvokerMethod(StringBuilder builder, StreamInvokerEmissionSpec emission)
{
builder.Append(" private static object InvokeStreamHandler");
builder.Append(emission.MethodIndex);
builder.Append("(object handler, object request, global::System.Threading.CancellationToken cancellationToken)");
builder.AppendLine();
builder.AppendLine(" {");
builder.Append(" var typedHandler = (");
builder.Append(emission.HandlerInterfaceDisplayName);
builder.AppendLine(")handler;");
builder.Append(" var typedRequest = (");
builder.Append(emission.RequestTypeDisplayName);
builder.AppendLine(")request;");
builder.AppendLine(" return typedHandler.Handle(typedRequest, cancellationToken);");
builder.AppendLine(" }");
}

private static void AppendDirectRegistrations(
StringBuilder builder,
ImplementationRegistrationSpec registration)
Expand Down
Loading
Loading