在 .NET 中优雅地合并并发请求


项目地址:https://github.com/eventhorizon-cli/RequestBatcher

NuGet:RequestBatcher

本文基于正式发布的 RequestBatcher v0.0.2 源码编写,支持 .NET 8 及以上版本。

前言

假设一个商品详情 API 会并发查询价格。最直接的实现,是每个请求都单独访问一次数据库:

请求 A -> SELECT 商品 101
请求 B -> SELECT 商品 101
请求 C -> SELECT 商品 205

请求 A 和 B 查询的是同一件商品,却仍然产生了两次数据库调用。流量集中到达时,数据库连接、网络往返、命令解析和查询执行都会被重复很多次。

要把同一时刻涌入的查询合起来,应用需要一个能一次处理多项请求的批处理方法。它可以先去重商品 ID,再执行一次批量查询:

请求 A -> 商品 101 --\
请求 B -> 商品 101 ----> 批量处理:去重为 [101, 205] -> 一次批量查询
请求 C -> 商品 205 --/

这个批处理方法可以先对当前批次中的商品 ID 执行 Distinct,再通过一次批量查询读取数据,最后把结果分发给每个调用方。这种合并只处理当前进程内短暂积压的请求,不是缓存,也不会为了凑批固定等待。这里的去重只发生在同一次批处理里;相同查询进入不同批次时,仍然会再次访问下游。

批量写入也是同一类问题。对于某个数据写入场景,多个调用方各自提交一条数据记录,批处理方法可以把当前批次一次写入下游。

当然,也可以让上游调用方先收集一个 List<T>,再统一交给数据库。问题是,请求通常来自彼此独立的 HTTP 调用、消息处理器或后台任务。它们并不知道同一时刻还有谁在做相同的事情,更不应该共同维护一套批次、并发、超时、异常和停止逻辑。

RequestBatcher 解决的正是这个协调问题:

它使用两个会反复出现的概念:分区是独立、按顺序处理请求的内存队列;Consumer 是从一个分区取出当前请求并调用批处理方法的处理循环。分区键只决定请求进入哪个分区,不负责去重。

  • 每个调用方仍然只提交自己的一个请求;
  • 同一分区中已经排队的请求,会被合并成最多 BatchSize 项的批次;
  • 批处理方法一次拿到一组请求,负责执行真正的下游批量操作;
  • 每个调用方仍然等待自己的 Task,并得到这个请求实际的成功、失败或取消结果;
  • 队列容量和批处理并发都可以被限制,避免流量突发直接压垮下游。

它不是持久化消息队列,也不是事务协调器。它更像应用调用路径中的一个进程内“请求汇合点”:上游保持单请求接口,下游获得批量处理机会。

公共接口

对应用来说,RequestBatcher 只有两个核心接口:IRequestBatcher<TRequest> 用于提交请求,IRequestBatchHandler<TRequest> 用于处理当前批次:

public interface IRequestBatcher<in TRequest>
{
    Task ProcessAsync(
        TRequest request,
        CancellationToken cancellationToken = default);

    Task ProcessAsync(
        IEnumerable<TRequest> requests,
        CancellationToken cancellationToken = default);
}

public interface IRequestBatchHandler<TRequest>
{
    ValueTask HandleAsync(
        IReadOnlyList<TRequest> requests,
        CancellationToken cancellationToken = default);
}

调用方通常使用第一个重载提交一项请求,并等待它实际处理完成;第二个重载用于调用方本来就持有一组请求时的一次性提交。Handler 接收的是当前批次的 IReadOnlyList<TRequest>,而不是某次提交的原始集合;一次显式提交仍可能被拆成多个 Handler 调用。

批量处理的收益来源

RequestBatcher 本身不会让一段逐条业务代码自动变快。收益来自 Handler 把多项请求真正变成更少的下游操作:

逐条处理:N 个请求 -> N 次数据库查询

批量处理:N 个请求 -> M 次 Handler -> M 次批量查询
                         通常 M < N

对于商品价格查询,示例 Handler 会先对当前批次的 ProductId 执行 Distinct,再用一次 ANY(@ProductIds) 查询价格,最后把结果写回每个 PriceQuery。因此,重复请求越集中在同一批次中,减少的数据库调用就越多。

下面这组端到端基准已经提交到 PostgreSqlPriceQueryBenchmarks.cs。工作负载是 1,000 次查询,但只有 10 个商品 ID,每个重复 100 次;这是适合批内去重的特定场景,不代表一般查询的平均结果。

直连路径和 RequestBatcher 都限制为 4 路并发、Npgsql 连接池固定为 4 个连接。容器启动、建表、填充数据和连接池预热不计入耗时,计时范围是提交请求到全部 Task 完成。每个场景预热 2 次、测量 8 次,并校验每个请求都得到了正确价格;RequestBatcher 路径还会验证 Handler 确实形成了多项批次。

场景 平均耗时 标准差 下游查询 处理情况
逐条查询,1,000 个请求 71.223 ms 2.793 ms 每个请求一次 SELECT 1,000 次单项查询
RequestBatcher,10 个商品 ID、每个重复 100 次 2.930 ms 0.430 ms 每个 Handler 批次一次 ANY(@product_ids) 批内 Distinct 后查询,再逐项回填结果

测试环境为 macOS 15.7.8、Apple M2 Max、.NET 8.0.25,以及 Testcontainers 启动的 PostgreSQL 17.6-alpine;BatchSize=100、MaxConcurrency=4。在这组高重复查询里,逐条查询的平均耗时约为 RequestBatcher 的 24.3 倍(71.223 / 2.930);RequestBatcher 约为直连路径的 4.1%。BenchmarkDotNet 也提示单次迭代不足 100 ms,因此这些数值只说明这个请求分布和这台机器上的相对结果,不应外推为所有查询场景的固定倍数。

这不是“相同商品永远只查一次”的承诺。相同 ProductId 仍然可能落入不同 Handler 批次;它只会在当前批次中通过 Distinct 合并。批次边界会受到请求到达时机、Consumer 调度、分区和 BatchSize 的共同影响。这组数据只说明:当大量重复查询确实在同一批次中汇合,并且 Handler 做了真正的批量查询时,收益会非常明显。

适用场景

RequestBatcher 适合彼此独立、可以在当前进程内短暂排队,并且下游更适合一次处理多项的请求。典型场景包括:

  • 数据库、缓存或下游 API 支持多项查询,希望把并发单项查询合并,并在批内去重相同 Key;
  • 数据库支持批量 INSERT、UPDATE 或 UPSERT,希望把并发单条写入合并成较少的批量操作;
  • 流量会在短时间内集中到达,需要限制排队请求数和下游并发量,对数据库或下游 API 形成并发限制与背压保护;
  • 相关请求需要保持分区内顺序,或者可以在当前批次中合并重复更新、去重相同查询;
  • 调用方取消时,只需要撤销尚未分发的请求;已经交给 Handler 的请求应继续执行并返回实际结果。

它尤其适合“上游接口必须保持单请求,只有下游知道如何批量处理”的调用链。调用方不需要感知批次,Handler 也不需要关心每个请求来自哪个调用方。

这里的限制由 MaxConcurrency 和 MaxPendingRequests 提供:前者限制同时执行的 Handler 数量,后者限制内存中尚未完成的请求数量。它们会把下游变慢的压力传回上游,但不按每秒请求数做固定速率限制;需要 QPS 限流时,应与专门的限流器配合使用。

不适用场景

下面这些需求已经超出进程内请求合并的职责:

  • 已接收的请求必须在进程崩溃后恢复。此时需要持久化存储或可靠消息队列;
  • 操作必须与调用方当前事务一起提交或回滚。把操作移出原事务会改变一致性边界;
  • 调用方必须直接从一次调用中得到业务结果,并且不能接受像下方示例那样由请求对象保存结果。RequestBatcher 只返回表示处理完成的 Task;
  • 下游操作依赖自动重试,或者副作用必须满足恰好一次(exactly-once)。RequestBatcher 不提供这些保证;
  • 业务必须凑满最小批量,或者依赖固定收集窗口。RequestBatcher 只处理当前已经排队的请求;
  • 单个调用方断开或取消后,正在执行的下游操作也必须立即停止。调用方的 CancellationToken 不会传给共享的 Handler 调用。

请求合并过程

多个独立请求合并为一个处理批次

RequestBatcher 的基本流程只有四步:

  1. 调用方通过 ProcessAsync 提交一个 TRequest。
  2. 请求按配置进入某个内存分区,并获得独立的完成状态。
  3. Consumer 可以继续处理该分区时,取出当前已经排队的最多 BatchSize 个请求,只调用一次 Handler。
  4. Handler 的执行结果再分别完成这一批请求对应的 Task。

这里最重要的词是“已经排队”。RequestBatcher 使用的是机会式合并,而不是固定时间窗口。

它不会在第一个请求到达后等待 10 ms、50 ms,试图凑满一个批次;Consumer 可以继续处理该分区时,会尽快取走当前已有的请求。因此,BatchSize 是单次 Handler 调用的上限,不是触发处理的最小数量。

一次可能的执行过程如下:

t0  请求 A 到达,分区空闲,Handler 开始处理 [A]
t1  A 仍在处理,请求 B、C、D 进入同一分区排队
t2  A 完成,Handler 下一次处理 [B, C, D]

具体批次边界取决于请求到达与 Consumer 调度的时机,RequestBatcher 不承诺 A 一定单独成批,也不承诺 B、C、D 一定在同一批。它只保证单批不超过 BatchSize,并在有并发积压时有机会形成更大的批次。

这个取舍很实用:低流量下不为了凑批而主动增加延迟,高并发下又可以利用排队请求减少下游调用次数。如果业务必须“至少凑够 100 条再处理”,或者必须使用固定收集窗口,RequestBatcher 并不适合。

快速接入示例

先安装 NuGet 包:

dotnet add package RequestBatcher --version 0.0.2

先定义查询请求和返回结果。ProcessAsync 只返回表示处理完成的 Task,因此请求对象需要保存自己的查询结果:

public sealed record ProductPrice(
    long ProductId,
    decimal Price);

public sealed class PriceQuery(long productId)
{
    public long ProductId { get; } = productId;

    public ProductPrice? Result { get; private set; }

    public void SetResult(ProductPrice? result) => Result = result;
}

public interface IProductPriceStore
{
    Task<IReadOnlyList<ProductPrice>> FindManyAsync(
        IReadOnlyList<long> productIds,
        CancellationToken cancellationToken);
}

Handler 对当前批次中的商品 ID 去重,执行一次批量查询,再把结果写回每个请求:

public sealed class PriceQueryHandler(IProductPriceStore store)
    : IRequestBatchHandler<PriceQuery>
{
    public async ValueTask HandleAsync(
        IReadOnlyList<PriceQuery> requests,
        CancellationToken cancellationToken = default)
    {
        var productIds = requests
            .Select(request => request.ProductId)
            .Distinct()
            .ToArray();

        var prices = await store.FindManyAsync(productIds, cancellationToken);
        var pricesByProductId = prices.ToDictionary(price => price.ProductId);

        foreach (var request in requests)
        {
            request.SetResult(
                pricesByProductId.GetValueOrDefault(request.ProductId));
        }
    }
}

然后把 Handler 和 RequestBatcher 一起注册到应用已有的 DI 容器:

builder.Services.AddRequestBatcher<PriceQuery, PriceQueryHandler>(
    ServiceLifetime.Scoped,
    options =>
    {
        options.BatchSize = 100;
        options.MaxConcurrency = 4;
        options.MaxPendingRequests = 10_000;
        options.FullMode = RequestBatchFullMode.Wait;
        // 可选:让相同商品路由到同一分区,增加它们在同一批次中去重的机会。
        // 分区键只决定路由,不会自动删除重复请求。
        options.UsePartitionKey(query => query.ProductId);
    });

最后,在应用服务中注入 IRequestBatcher<PriceQuery>。调用方仍然一次只查询一个商品:

public sealed class ProductPriceService(
    IRequestBatcher<PriceQuery> requestBatcher)
{
    public async Task<ProductPrice?> GetAsync(
        long productId,
        CancellationToken cancellationToken = default)
    {
        var query = new PriceQuery(productId);
        await requestBatcher.ProcessAsync(query, cancellationToken);
        return query.Result;
    }
}

GetAsync 会等待查询所在的 Handler 批次完成,再读取这个请求对应的结果。相同 ProductId 通过分区键进入同一分区,因此有机会在当前批次中被去重;如果它们落入不同批次,仍然会再次访问数据库,RequestBatcher 不会替代缓存。

如果 Handler 没有需要由 DI 解析的依赖,也可以通过 AddRequestBatcher<TRequest> 的委托重载直接注册一个 RequestBatchHandler<TRequest>,注册时仍需明确指定 Handler 生命周期。

Handler 生命周期

RequestBatcher Coordinator 本身是 Singleton,但 Handler 生命周期由注册时显式指定:

  • Scoped:每个 Handler 批次创建一个异步 Scope,并在批次结束后释放;
  • Transient:同样在每个批次的 Scope 中解析一次;
  • Singleton:所有批次复用同一个实例。

数据库上下文等 Scoped 依赖适合使用 Scoped Handler。选择 Singleton 且 MaxConcurrency > 1 时,多个分区可能同时调用同一个 Handler 实例,因此它必须是线程安全的。

请求完成与异常回传

请求合并最麻烦的部分,其实不是把多个对象放进一个数组,而是把批量执行结果重新对应回每个调用方。

RequestBatcher 的请求、队列、Handler 与完成路径

单个请求进入 RequestBatcher 后,会被包装成内部的 PendingBatchRequest<TRequest>。其中既保存请求本身,也保存状态、取消注册和完成源。

Handler 正常完成时,这一批请求对应的 Task 全部成功;Handler 抛出异常时,同一批调用方都会收到这个原始异常。失败不会触发自动重试,Consumer 会继续处理后续批次。

Caller A --\
Caller B ----> Handler([A, B, C]) 成功 ----> Task A/B/C 全部成功
Caller C --/

Caller D --\
Caller E ----> Handler([D, E]) 失败 ------> Task D/E 收到同一异常

Handler 返回 ValueTask,是因为同步完成的处理器可以避免额外创建 Task。这个 ValueTask 只会在 RequestBatcher 内部等待一次,不会直接暴露给调用方。

显式批量提交的语义

除了提交单个请求,RequestBatcher 也支持提交调用方已经持有的一组请求:

await requestBatcher.ProcessAsync(priceQueries, cancellationToken);

这个重载只表示调用方用一次 ProcessAsync 提交了多个请求,不表示 Handler 只会调用一次。RequestBatcher 会先枚举输入并创建快照,再让每一项独立路由;它们可能被 BatchSize 拆开,也可能进入不同分区并行执行。

行为 单个请求 显式提交一组请求
输入 接收一个 TRequest 枚举一次并保存快照;空序列立即完成
路由 路由到一个分区 每一项独立路由,一次提交可以跨分区
Handler 边界 可能与其他排队请求合并 可能按分区和 BatchSize 拆成多次 Handler 调用
完成 Task 等待这一项 一个 Task 等待组内所有项结束
失败 返回所在 Handler 批次的异常 任一相关项失败,整组 Task 失败;已经成功的操作不会回滚

显式提交的 Task 会等待所有项结束,再按下面的优先级决定最终结果:

  1. 只要有任何一项失败,整组 Task 就失败;错误可能来自 Handler,也可能来自入队、路由或 Consumer;
  2. 没有失败但至少一项取消,整组 Task 取消;
  3. 所有项都成功,整组 Task 成功。

一次提交可能跨多个 Handler 批次。如果多个批次分别失败,所有不同的异常实例仍可以从 Task.Exception.InnerExceptions 中取得;普通 await 则遵循 .NET 的 Task 语义抛出其中一个异常。

容量策略取决于 FullMode:Wait 可以随着容量释放逐步接收超大请求组,Fail 则要求整组立即获得容量。具体行为见下文的“背压与容量控制”。无论哪种模式,一次显式提交都不是事务边界、分区边界或 Handler 批次边界。

批次大小与处理并发

BatchSize 控制一次 Handler 最多处理多少项,MaxConcurrency 控制最多有多少次 Handler 调用并行执行。

它们不能互相替代:批次很大不代表并发很高,并发很高也不代表每批有很多请求。

BatchSize:单批上限

假设 BatchSize = 100:

  • 分区中只有 3 项排队时,Handler 可以收到 3 项;
  • 分区中有 250 项排队时,会被拆成最多 100 项的多个批次;
  • 不同分区的请求不会被放进同一个 Handler 批次。

批次大小应该根据下游能力决定,而不是越大越好。SQL 参数数量、请求体大小、事务持锁时间、缓存 Pipeline 长度和 API 限流,都可能成为新的边界。

MaxConcurrency:并发与分区

RequestBatcher 使用 MaxConcurrency 个内存分区,并为每个分区建立一个顺序消费循环。因此,这个配置同时决定 Handler 最大并发数和处理分区数。

配置 路由方式 顺序语义
MaxConcurrency = 1 所有请求进入唯一分区 保持全局处理顺序
MaxConcurrency > 1,无分区键 请求逐项轮询到不同分区 只保证各分区内部顺序
MaxConcurrency > 1,有分区键 相同键路由到同一分区 相同键进入分区后顺序处理,不同分区可以并行

并发调用方在请求真正进入分区之前没有额外的先后保证。提高 MaxConcurrency 后,也不再存在跨分区的全局 FIFO。

下游如果最多只允许 8 个并发连接,把 MaxConcurrency 配成 100 并不会带来免费性能,反而可能把压力转移到连接池和下游限流队列。它应该与下游的真实并发能力一起设置。

分区键与去重

快速接入中的价格查询使用 ProductId 作为分区键,目的是让相同商品的查询进入同一分区。真正的去重仍然来自 Handler 中的 Distinct;分区键本身不会删除任何请求。

分区键也可以用于需要顺序处理的写入。例如,同一个商品价格的多个版本可以批量更新,但不希望两个 Handler 同时更新同一商品。

这时可以配置分区键:

builder.Services.AddRequestBatcher<PriceUpdate, PriceUpdateHandler>(
    ServiceLifetime.Scoped,
    options =>
    {
        options.BatchSize = 100;
        options.MaxConcurrency = 4;
        options.MaxPendingRequests = 10_000;
        options.UsePartitionKey(update => update.ProductId);
    });

相同 ProductId 会进入同一分区,因而不会被两个 Handler 调用并发处理;其他商品仍然可以在不同分区并行执行。

Handler 还可以合并当前批次中的重复更新,只保留版本最高的一项:

public sealed record PriceUpdate(
    long ProductId,
    long Version,
    decimal Price);

public sealed class PriceUpdateHandler(ProductPriceStore store)
    : IRequestBatchHandler<PriceUpdate>
{
    public async ValueTask HandleAsync(
        IReadOnlyList<PriceUpdate> requests,
        CancellationToken cancellationToken = default)
    {
        var latestUpdates = requests
            .GroupBy(request => request.ProductId)
            .Select(group => group.MaxBy(request => request.Version)!)
            .ToArray();

        await store.UpsertLatestAsync(latestUpdates, cancellationToken);
    }
}

但要注意,分区键只负责路由:

  • 它不会把同一个键的所有请求永久收集到同一个批次;
  • 它不会自动删除重复请求;
  • 不同键可能映射到同一个分区;
  • 同一键在当前批次中合并了,也可能在后续批次再次出现。

因此,跨批次正确性仍然要由存储层保证。例如,价格表的 UPSERT 应只允许更高版本覆盖旧版本。分区键避免同一商品被并发处理,数据库版本条件则避免后到的旧数据覆盖新状态,两者解决的不是同一个问题。

数值型分区键必须是有限整数,字符串键不能为 null。键选择器应该稳定、无副作用、可以被并发调用,并且只表达业务上的顺序需求。

如果显式组提交中的 Partition Key 选择器抛出异常,整组会失败且不会只处理其中一部分,已经预留的容量也会被释放。

批量写入场景

查询是本文的主例子,但相同的协调方式也可以用于批量写入。对于某个数据写入场景,不同调用方逐条提交数据,Handler 再将当前批次一次写入下游:

public sealed record DataWriteRequest(
    long DataId,
    string Payload);

public interface IDataStore
{
    Task WriteBatchAsync(
        IReadOnlyList<DataWriteRequest> requests,
        CancellationToken cancellationToken);
}

public sealed class DataWriteBatchHandler(IDataStore store)
    : IRequestBatchHandler<DataWriteRequest>
{
    public async ValueTask HandleAsync(
        IReadOnlyList<DataWriteRequest> requests,
        CancellationToken cancellationToken = default)
    {
        await store.WriteBatchAsync(requests, cancellationToken);
    }
}

注册和调用方式与查询相同:

builder.Services.AddRequestBatcher<DataWriteRequest, DataWriteBatchHandler>(
    ServiceLifetime.Scoped,
    options =>
    {
        options.BatchSize = 256;
        options.MaxConcurrency = 4;
        options.MaxPendingRequests = 10_000;
        options.FullMode = RequestBatchFullMode.Wait;
    });

public sealed class DataWriteService(
    IRequestBatcher<DataWriteRequest> requestBatcher)
{
    public Task WriteAsync(
        DataWriteRequest request,
        CancellationToken cancellationToken = default) =>
        requestBatcher.ProcessAsync(request, cancellationToken);
}

WriteAsync 返回的 Task 表示这条数据的实际处理结果,而不是仅表示它已经进入内存队列。真正的批量写入应在 Handler 中完成,例如使用数据库数组参数、批量写入 Pipeline 或下游 API 的批量接口。

项目仓库中的 PostgreSQL 示例同时展示了批量 UPSERT、查询去重和结果回填。

背压与容量控制

BatchSize 只能限制一次 Handler 的输入数量,不能限制总共有多少请求正在等待和执行。

如果生产速度长期高于消费速度,无界排队最终只会把下游过载变成应用内存过载。因此 RequestBatcher 还提供 MaxPendingRequests,限制已经接收、仍在排队或正在由 Handler 处理的请求数量;Wait 模式下,一次显式提交可以超过这个数量。

默认配置如下:

选项 默认值 含义
BatchSize 128 单次 Handler 调用的请求上限
MaxConcurrency 1 Handler 最大并发数,同时也是分区数
MaxPendingRequests 8192 已接收且尚未完成的请求上限;Wait 模式下显式提交可超过该值
FullMode Wait 容量不足时异步等待;也可以选择 Fail
UsePartitionKey(...) 未配置 默认逐项轮询路由

Wait 模式

RequestBatchFullMode.Wait 会异步等待容量。调用方不会阻塞线程,并且在等待期间可以通过自己的 CancellationToken 取消。

这种模式把背压自然传回调用链:下游变慢后,上游的 ProcessAsync 也会变慢,而不是继续无限制接收请求。

对 ProcessAsync(IEnumerable<TRequest>) 而言,超大请求组也会被处理。假设 MaxPendingRequests = 10_000,一次提交 20_000 项:前面最多 10_000 项会先进入队列,剩余请求随着前面请求完成、容量释放而继续进入。调用方仍只等待一个 Task,它会在整组请求都成功、失败或取消后结束。这是逐步入队的策略,不是事务或 Handler 批次。

Fail 模式

RequestBatchFullMode.Fail 在容量不足时立即让返回的 Task 以 RequestBatchQueueFullException 失败:

try
{
    await requestBatcher.ProcessAsync(request, cancellationToken);
}
catch (RequestBatchQueueFullException exception)
{
    logger.LogWarning(
        "Request batch queue is full. Capacity: {Capacity}, Requested: {Requested}",
        exception.Capacity,
        exception.RequestedCount);

    // 按业务约定返回 429、降级或交给可靠队列
}

在 Fail 模式下,显式提交必须整组获得当前可用容量;不会出现前半组已经入队、后半组因为容量不足被拒绝的状态。整组数量超过 MaxPendingRequests,或当前容量不足时,都会以 RequestBatchQueueFullException 失败。

请求取消的时机

一个 Handler 批次可能同时包含多个调用方的请求。如果把其中任意一个调用方的 CancellationToken 直接传给 Handler,那么一个 HTTP 客户端断开连接,就可能把其他调用方共享的数据库操作一起取消。

更麻烦的是,Handler 开始执行后可能已经产生了副作用。此时把调用方的 Task 标记为取消,会让上游误以为操作没有发生,重试后反而造成重复写入。

RequestBatcher 因此只在 Handler 分发之前接受调用方取消:

Queued ---------> Processing ---------> Succeeded / Faulted
  |                    |
  | caller cancel      | caller cancel
  v                    v
Canceled          继续等待 Handler 的真实结果

具体语义是:

  1. 请求还在等待容量或排队时,调用方取消可以移除这项请求并取消它的 Task。
  2. Consumer 准备处理时,会用原子状态切换把请求从 Queued 改为 Processing。
  3. 一旦切换成功,之后的调用方取消不再改变结果;Task 最终反映 Handler 的成功或异常。
  4. 传给 Handler 的 Token 属于 RequestBatcher 的 Consumer 生命周期,不是任意一个调用方的 Token。

这不是忽略取消,而是避免取消状态掩盖一个可能已经发生的共享副作用。即使 BatchSize = 1,RequestBatcher 也不会把调用方 Token 传给 Handler。业务如果要求“调用方一断开,正在执行的下游操作必须立刻停止”,RequestBatcher 就不适合这条调用路径,应直接调用下游,或使用另一套与调用方生命周期绑定的取消机制。

应用关闭时的请求处理

RequestBatchCoordinator<TRequest> 实现了 IAsyncDisposable,并提供 StopAsync。停止过程按下面的顺序执行:

停止接收新请求
    -> 处理完停止前已经开始的所有提交,包括仍在等待容量的部分
    -> 停止 Consumer
    -> 释放生命周期资源

开始关闭后提交的新请求会以 ObjectDisposedException 失败;在关闭开始前已经提交、但仍在等待容量的请求会继续等待并处理完成。调用方自己的 Token 更早取消时,仍可能观察到取消结果。在 Consumer 正常运行的前提下,已经接收的请求会继续处理到结束。

传给 StopAsync 的 CancellationToken 只取消调用方对停止过程的等待,不会撤销已经开始的处理。即使等待 StopAsync(token) 时传入的 Token 已取消,后台停止过程仍会继续。

这类生命周期语义很重要。请求合并通常处在 HTTP、数据库和应用宿主之间,如果应用停止时直接取消所有 Consumer,调用方可能永远等不到结果,已经接收的业务请求也会无声丢失。

内部实现:从入队到完成

RequestBatcher 的公共 API 很小,真正的协调由几个内部组件完成:

RequestBatcher 底层使用我开源的 BufferQueue。

IRequestBatcher<TRequest>.ProcessAsync
    |
    v
RequestBatchCoordinator<TRequest>
    |
    +-- PendingRequestProducer<TRequest>
    |       |
    |       v
    |   BufferQueue Memory Topic
    |       |
    |       +-- Partition 0 -> Consumer 0
    |       +-- Partition 1 -> Consumer 1
    |       +-- ...
    |
    +-- PendingBatchRequest<TRequest>
            |
            +-- request
            +-- queued / processing / canceled / completed
            +-- completion

注册 AddRequestBatcher 时,内部会创建一个 BufferQueue Memory Topic:

  • PartitionNumber 使用 MaxConcurrency;
  • BoundedCapacity 使用 MaxPendingRequests;
  • 队列满策略映射自 RequestBatcher 的 Wait 或 Fail;
  • 配置了 Partition Key 时,路由规则一起交给内部 Topic。

Coordinator 随后创建与 MaxConcurrency 相同数量的 Pull Consumer,也就是主动从分区取请求的处理循环。每个 Consumer 顺序读取自己的分区,每次最多拉取 BatchSize 项,并关闭自动提交(Auto Commit):Handler 处理结束后,再由 Consumer 显式提交这一批的消费进度。

Consumer 拿到一批内部请求后,会先跳过已经取消的项,再把剩余的 TRequest 复制到数组中交给 Handler。Handler 正常返回或抛出异常时,Consumer 都会先完成对应请求的状态,再提交这一批的消费进度。读取批次、Consumer 循环或提交进度本身失败时,属于消费基础设施错误;这时不会保证本批进度已提交。

内部监控组件(Monitor)会记录第一个 Consumer 故障,停止继续接收请求,并让仍在排队的请求以该异常失败,避免调用方持有一个永远不会完成的 Task。

显式提交一组请求时,内部会创建共享的 BatchSubmissionCompletion。它只保留一个 TaskCompletionSource:组内每项结束时更新共享计数和异常状态,最后一项结束时再完成这个 Task。这样不需要为组内每项额外创建完成源,也不需要构造 Task[] 后调用 Task.WhenAll;它只优化完成聚合的分配与协调开销,不改变取消、异常和 Handler 批次语义。

BufferQueue 在这里是实现细节。应用不需要注册 Topic、Producer 或 Consumer,也不应该依赖内部 Topic 名称;对外契约始终只有 IRequestBatcher<TRequest>、IRequestBatchHandler<TRequest> 和配置项。

使用限制

确定场景适合之后,还需要确认几项实现上的限制:

  1. ProcessAsync(IEnumerable<TRequest>) 表示一次提交,不保证只产生一次 Handler 调用。
  2. BatchSize 是单批上限,不是最小数量;低流量下可能持续产生单项批次。
  3. MaxConcurrency > 1 时只保证分区内顺序,不保证跨分区全局 FIFO。
  4. Partition Key 只负责路由,不会自动去重,也不保证一个 Key 独占一个分区或进入同一批次。
  5. 显式组提交跨批次失败时,不会回滚已经成功的项;它不是事务边界。
  6. Singleton Handler 在并发大于 1 时必须线程安全;Scoped 或 Transient Handler 则按处理批次创建 Scope。
  7. MaxPendingRequests 只限制内存中的未完成请求,并形成背压;它不提供可靠投递保证。

总结

优雅地合并并发请求,不是让每个调用方都学会收集批次,而是把“单项提交”和“批量执行”分成两个独立契约。

RequestBatcher 让调用方继续使用简单的 ProcessAsync(request),由内部完成机会式合并、分区路由、并发限制、背压、结果回传和有序停止;Handler 只关心如何把 IReadOnlyList<TRequest> 变成一次真正的数据库、缓存或下游批量操作。

选择它之前,可以先确认三件事:

  • 请求允许在当前进程内短暂排队,进程失败后不要求自动恢复;
  • 下游确实有批量处理能力,并且批量能够减少固定成本;
  • 业务接受“BatchSize 是上限、调用方取消只在分发前有效、分区内有序”的处理语义。

满足这些条件时,它可以把原本分散在各个调用方中的批次协调逻辑收回到一个清晰边界里,让上游保持简单,也让下游真正获得批量处理的机会。

项目地址:eventhorizon-cli/RequestBatcher

本文对应版本:RequestBatcher v0.0.2

完整示例:RequestBatcher.Deduplication

中文文档:RequestBatcher README

底层队列:eventhorizon-cli/BufferQueue

核心实现:RequestBatchCoordinator | RequestBatchConsumer

行为测试:RequestBatchCoordinatorTests | RequestBatchSubmissionTests

NuGet:RequestBatcher

posted @ 2026-08-16 19:29  黑洞视界  阅读(559)  评论(6)    收藏  举报