ASP.NET Core网关服务实现疑问：中间件方案是否可行？

一、基础思路的正确性 #

你的核心思路是对的：用中间件实现这种“开关式”请求转发，是ASP.NET Core里做简单API网关的合理方案。不过现有代码里有几个关键问题需要修正，还有不少可以优化的地方，下面一步步说明。

二、现有代码的核心问题 #

1. HttpClient创建错误
   绝对不能在每次请求里新建HttpClient，这会导致系统socket资源耗尽，应该用ASP.NET Core内置的IHttpClientFactory来获取实例，它会自动管理HttpClient的生命周期。

2. 请求头复制方式错误
   直接赋值request.Headers = context.Request.Headers会触发异常，因为有些头是HttpRequestMessage的保留字段，需要逐个筛选复制，而且Content相关的头要放到request.Content.Headers里。

3. 请求体处理缺失
   HTTP请求体只能被读取一次，你需要先把原始请求体读取出来，再设置到转发请求的Content中，还要注意重置原始请求的流位置（如果后续有其他中间件需要读取的话）。

4. 响应转发方式错误
   不能直接赋值context.Response = response，必须手动复制响应状态码、响应头，再把响应流拷贝到原始响应流里。

三、优化后的完整实现 #

1. 先在Program.cs里注册依赖 #

var builder = WebApplication.CreateBuilder(args);
// 注册IHttpClientFactory，用于管理HttpClient生命周期
builder.Services.AddHttpClient();
// 注册自定义中间件
builder.Services.AddTransient<GatewayMiddleware>();

var app = builder.Build();

app.UseMiddleware<GatewayMiddleware>();
app.Run();

2. 完善GatewayMiddleware代码 #

public class GatewayMiddleware
{
    private readonly RequestDelegate _next;
    private readonly IHttpClientFactory _httpClientFactory;

    // 通过构造注入获取IHttpClientFactory
    public GatewayMiddleware(RequestDelegate next, IHttpClientFactory httpClientFactory)
    {
        _next = next;
        _httpClientFactory = httpClientFactory;
    }

    public async Task InvokeAsync(HttpContext context)
    {
        // 1. 从查询字符串获取目标API标识（这里假设用api-target参数传URL，后续可以优化成配置别名）
        if (!context.Request.Query.TryGetValue("api-target", out var targetApiUrl))
        {
            context.Response.StatusCode = StatusCodes.Status400BadRequest;
            await context.Response.WriteAsync("缺少必填查询参数：api-target");
            return;
        }

        // 验证URL有效性
        if (!Uri.TryCreate(targetApiUrl, UriKind.Absolute, out var targetUri))
        {
            context.Response.StatusCode = StatusCodes.Status400BadRequest;
            await context.Response.WriteAsync("api-target参数不是有效的URL");
            return;
        }

        // 2. 获取HttpClient实例
        var httpClient = _httpClientFactory.CreateClient();

        // 3. 构建转发请求
        var requestMessage = new HttpRequestMessage(new HttpMethod(context.Request.Method), targetUri);

        // 4. 复制请求头（跳过保留头，区分普通头和Content头）
        foreach (var header in context.Request.Headers)
        {
            // 先尝试添加到请求头，失败的话（比如是Content相关的头）添加到Content头
            if (!requestMessage.Headers.TryAddWithoutValidation(header.Key, header.Value.ToArray()))
            {
                requestMessage.Content ??= new StringContent(string.Empty);
                requestMessage.Content.Headers.TryAddWithoutValidation(header.Key, header.Value.ToArray());
            }
        }

        // 5. 处理请求体
        if (context.Request.ContentLength > 0 || !string.IsNullOrEmpty(context.Request.ContentType))
        {
            // 读取原始请求体
            using var streamReader = new StreamReader(context.Request.Body);
            var requestBody = await streamReader.ReadToEndAsync();
            // 重置原始请求流位置，方便后续中间件读取（如果有的话）
            context.Request.Body.Position = 0;

            // 设置转发请求的Content
            requestMessage.Content = new StringContent(requestBody, Encoding.UTF8, context.Request.ContentType);
        }

        // 6. 发送请求到目标API
        var responseMessage = await httpClient.SendAsync(requestMessage, HttpCompletionOption.ResponseHeadersRead);

        // 7. 把目标API的响应转发给客户端
        context.Response.StatusCode = (int)responseMessage.StatusCode;

        // 复制响应头
        foreach (var header in responseMessage.Headers)
        {
            context.Response.Headers.TryAdd(header.Key, header.Value.ToArray());
        }
        foreach (var header in responseMessage.Content.Headers)
        {
            context.Response.Headers.TryAdd(header.Key, header.Value.ToArray());
        }

        // 复制响应体
        await responseMessage.Content.CopyToAsync(context.Response.Body);
        await context.Response.Body.FlushAsync();
    }
}

四、最佳实践补充 #

• 配置化管理API地址：别直接从查询字符串传完整URL，建议在appsettings.json里维护API列表，用别名来指定目标（比如api=user对应https://api-user.example.com），这样更安全也更容易维护：

  {
    "ApiEndpoints": {
      "user": "https://api-user.example.com",
      "order": "https://api-order.example.com"
    }
  }
  

  然后在中间件里通过IConfiguration读取配置，根据别名匹配目标URL。

• 添加超时与重试：注册HttpClient的时候可以配置超时和重试策略，提升网关的稳定性：

  builder.Services.AddHttpClient()
      // 设置10秒超时
      .AddPolicyHandler(Policy.TimeoutAsync<HttpResponseMessage>(TimeSpan.FromSeconds(10)))
      // 针对失败响应重试3次
      .AddPolicyHandler(Policy.HandleResult<HttpResponseMessage>(r => !r.IsSuccessStatusCode)
          .RetryAsync(3));
  

• 添加日志监控：在中间件里加日志，记录转发的目标URL、响应状态码、耗时等信息，方便排查问题。

• 安全限制：限制允许转发的目标域名，防止恶意请求通过网关转发到不安全的地址。

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