ASP.NET Core中间件事务提交延迟问题排查与解决方案咨询

问题背景 #

实现了DbTransactionMiddleware，当端点标记DbTransactionAttribute时会在数据库事务中处理请求。测试流程为：

1. 调用修改接口变更数据库数据；
2. 调用独立GET接口验证数据持久化结果。
   但偶尔出现GET请求验证失败，日志显示Transaction completed消息在GET请求被接收后才出现，怀疑修改接口的响应在事务CommitAsync执行前就返回给了客户端，不想通过测试加延迟解决。

核心原因 #

问题的本质是端点处理逻辑（_next(context)）执行过程中，ASP.NET Core框架已将响应发送给客户端，而事务提交操作是在端点逻辑完成后才执行。客户端收到修改接口的响应后立即发起GET请求时，数据库事务尚未提交，因此读取到旧数据。

可行解决方案 #

方案1：利用Response.OnStarting回调确保事务提交后再返回响应 #

将事务提交操作注册到HttpContext.Response.OnStarting回调中，该回调会在响应即将发送到客户端之前触发，保证事务提交完成后才返回响应，从根源上避免客户端提前发起请求的问题。

修改后的中间件代码：

public class DbTransactionMiddleware
{
    private readonly RequestDelegate _next;

    public DbTransactionMiddleware(RequestDelegate next)
    {
        _next = next;
    }

    public async Task Invoke(HttpContext context, IDbContext dbContext, ILogger<DbTransactionMiddleware> logger)
    {
        var attribute = context.Features
            .Get<IEndpointFeature>()
            ?.Endpoint
            ?.Metadata
            .GetMetadata<DbTransactionAttribute>();

        if (attribute == null)
        {
            await _next(context);
            return;
        }

        var transaction = await dbContext.Database.BeginTransactionAsync();
        var hasRolledBack = false;

        try
        {
            // 注册响应发送前的回调，在此提交事务
            context.Response.OnStarting(async () =>
            {
                if (!hasRolledBack)
                {
                    await transaction.CommitAsync();
                    logger.LogInformation("Transaction completed");
                }
            });

            // 执行端点业务逻辑
            await _next(context);
        }
        catch
        {
            hasRolledBack = true;
            await transaction.RollbackAsync();
            throw;
        }
        finally
        {
            // 自动释放事务资源，避免泄漏
            if (!hasRolledBack)
            {
                context.Response.RegisterForDispose(transaction);
            }
            else
            {
                await transaction.DisposeAsync();
            }
        }
    }
}

方案2：检查端点是否存在提前结束响应的操作 #

排查标记DbTransactionAttribute的端点代码，确保没有手动提前完成响应的逻辑（比如直接调用HttpContext.Response.WriteAsync后结束请求），这类操作会导致_next(context)提前完成，事务还未提交就返回响应。

测试补充说明 #

采用方案1后，无需在测试中添加延迟。若使用WebApplicationFactory等集成测试框架，只需保证测试代码正常等待修改请求的响应即可，事务提交逻辑会自动在响应发送前完成。

内容的提问来源于stack exchange，提问作者Alexander