mirror of
https://github.com/GeWuYou/GFramework.git
synced 2026-05-09 01:54:30 +08:00
aebf1e974d
- 新增 stream pipeline 契约、dispatcher executor 缓存与 generated invoker 兼容路径 - 补充 Architecture 与 IOC 的流式管道注册入口及对应回归测试 - 更新 CQRS 文档和 cqrs-rewrite 的 active tracking/trace
45 lines
1.7 KiB
C#
45 lines
1.7 KiB
C#
// Copyright (c) 2025-2026 GeWuYou
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
using System.Threading;
|
|
using GFramework.Cqrs.Abstractions.Cqrs;
|
|
|
|
namespace GFramework.Core.Tests.Architectures;
|
|
|
|
/// <summary>
|
|
/// 记录流式请求通过管道次数的测试行为。
|
|
/// </summary>
|
|
/// <typeparam name="TRequest">流式请求类型。</typeparam>
|
|
/// <typeparam name="TResponse">流式响应元素类型。</typeparam>
|
|
public sealed class TrackingStreamPipelineBehavior<TRequest, TResponse> : IStreamPipelineBehavior<TRequest, TResponse>
|
|
where TRequest : IStreamRequest<TResponse>
|
|
{
|
|
private static int _invocationCount;
|
|
|
|
/// <summary>
|
|
/// 获取当前测试进程中该流式请求类型对应的行为触发次数。
|
|
/// 该计数器是按泛型闭包共享的静态状态，测试需要在每次运行前显式重置。
|
|
/// </summary>
|
|
public static int InvocationCount
|
|
{
|
|
get => Volatile.Read(ref _invocationCount);
|
|
set => Volatile.Write(ref _invocationCount, value);
|
|
}
|
|
|
|
/// <summary>
|
|
/// 以线程安全方式记录一次行为执行，然后继续执行下一个处理阶段。
|
|
/// </summary>
|
|
/// <param name="message">当前流式请求消息。</param>
|
|
/// <param name="next">下一个处理阶段。</param>
|
|
/// <param name="cancellationToken">取消令牌。</param>
|
|
/// <returns>下游处理阶段返回的异步流。</returns>
|
|
public IAsyncEnumerable<TResponse> Handle(
|
|
TRequest message,
|
|
StreamMessageHandlerDelegate<TRequest, TResponse> next,
|
|
CancellationToken cancellationToken)
|
|
{
|
|
Interlocked.Increment(ref _invocationCount);
|
|
return next(message, cancellationToken);
|
|
}
|
|
}
|