ASP.NET Core TransactionMiddleware在SaveChangesAsync失败时如何返回错误响应

问题根因 #

你遇到的问题本质是执行顺序导致的：

• 控制器动作执行return CreatedAtAction后，ASP.NET Core会直接将201响应写入并发送给客户端，此时你后续在中间件中执行SaveChangesAsync抛出异常时，响应流已经完成发送，无法再修改返回给用户的内容
• UseEndpoints确实是终端中间件，所有注册在它之后的中间件都不会被触发，这是ASP.NET Core中间件管道的设计特性

解决方案 #

你可以通过响应缓冲+异常捕获的方式调整中间件实现，核心思路是先把响应内容缓存到内存，等SaveChangesAsync执行成功后再将缓存的响应发送给客户端，失败则直接返回错误响应：

1. 修改TransactionMiddleware实现 #

namespace DotNetCMS.Persistence.EntityFrameworkCore.AspNetCore
{
    public sealed class TransactionMiddleware
    {
        private readonly RequestDelegate _next;

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

        public async Task InvokeAsync(HttpContext httpContext, CmsContext cmsContext, IWebHostEnvironment env)
        {
            // 替换原始响应流为内存流，缓存响应内容不直接发送
            var originalResponseBody = httpContext.Response.Body;
            using var responseBuffer = new MemoryStream();
            httpContext.Response.Body = responseBuffer;

            try
            {
                // 执行后续中间件和控制器逻辑
                await _next(httpContext);

                // 保存数据，成功才把缓存的响应发回客户端
                await cmsContext.SaveChangesAsync();
                
                // 把缓冲的内容写回原始响应流
                responseBuffer.Seek(0, SeekOrigin.Begin);
                await responseBuffer.CopyToAsync(originalResponseBody);
            }
            catch (Exception ex)
            {
                // 保存失败，重置响应，返回错误信息
                httpContext.Response.Clear();
                httpContext.Response.StatusCode = StatusCodes.Status500InternalServerError;
                httpContext.Response.ContentType = "application/json";
                // 开发环境返回异常详情，生产环境返回通用错误避免泄露敏感信息
                var errorResp = env.IsDevelopment() 
                    ? new { error = ex.Message, stackTrace = ex.StackTrace }
                    : new { error = "服务器处理请求时发生错误" };
                await httpContext.Response.WriteAsJsonAsync(errorResp);
            }
            finally
            {
                // 恢复原始响应流
                httpContext.Response.Body = originalResponseBody;
            }
        }
    }
}

2. 注册顺序不需要修改 #

你原本的注册位置UseRouting之后、UseEndpoints之前是正确的，不需要调整。

可选优化：显式事务支持 #

如果需要更严格的事务保证，可以把SaveChangesAsync包裹在显式事务中，失败时自动回滚所有未提交的变更：

public async Task InvokeAsync(HttpContext httpContext, CmsContext cmsContext, IWebHostEnvironment env)
{
    var originalResponseBody = httpContext.Response.Body;
    using var responseBuffer = new MemoryStream();
    httpContext.Response.Body = responseBuffer;
    // 开启显式事务
    using var transaction = await cmsContext.Database.BeginTransactionAsync();
    
    try
    {
        await _next(httpContext);
        await cmsContext.SaveChangesAsync();
        await transaction.CommitAsync();
        
        responseBuffer.Seek(0, SeekOrigin.Begin);
        await responseBuffer.CopyToAsync(originalResponseBody);
    }
    catch (Exception ex)
    {
        await transaction.RollbackAsync();
        // 错误响应逻辑和上文一致
    }
    finally
    {
        httpContext.Response.Body = originalResponseBody;
    }
}

备选方案：使用ActionFilter #

如果你不需要覆盖所有请求，只需要针对控制器动作生效，也可以用IAsyncActionFilter实现，不需要处理响应缓冲，直接在动作结果执行前完成数据保存：

public class TransactionFilter : IAsyncActionFilter
{
    private readonly CmsContext _cmsContext;
    private readonly IWebHostEnvironment _env;

    public TransactionFilter(CmsContext cmsContext, IWebHostEnvironment env)
    {
        _cmsContext = cmsContext;
        _env = env;
    }

    public async Task OnActionExecutionAsync(ActionExecutingContext context, ActionExecutionDelegate next)
    {
        var resultContext = await next();
        // 只有动作执行无异常、且响应为成功状态时才保存
        if (resultContext.Exception == null && context.HttpContext.Response.StatusCode < 400)
        {
            try
            {
                await _cmsContext.SaveChangesAsync();
            }
            catch (Exception ex)
            {
                // 直接修改结果返回错误
                var errorResp = _env.IsDevelopment() 
                    ? new { error = ex.Message, stackTrace = ex.StackTrace }
                    : new { error = "数据保存失败" };
                resultContext.Result = new ObjectResult(errorResp)
                {
                    StatusCode = StatusCodes.Status500InternalServerError
                };
                resultContext.ExceptionHandled = true;
            }
        }
    }
}

注册过滤器只需要在Startup.cs的ConfigureServices方法中添加：

services.AddControllers(options =>
{
    options.Filters.Add<TransactionFilter>();
});

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