YARP：后端返回绝对URI时如何动态替换Location请求头？

核心思路 #

利用YARP的响应转换扩展能力，在网关接收到后端API的重定向响应后，动态修改Location请求头：根据当前请求匹配的路由前缀（如/api1），替换后端返回的绝对路径中的基础地址，同时自动修正为网关的端口和前缀。

具体实现步骤 #

1. 编写自定义响应转换逻辑 #

创建一个自定义转换提供者，专门处理重定向的Location头修正：

public class GraphQLRedirectTransformProvider : ITransformProvider
{
    public void Apply(TransformBuilderContext context)
    {
        // 只处理3xx系列的重定向响应
        context.AddResponseTransform(async context =>
        {
            if (context.Response.StatusCode is >= 300 and < 400 && 
                context.Response.Headers.TryGetValue(HeaderNames.Location, out var locationValues))
            {
                var originalLocation = locationValues.ToString();
                // 从当前请求的端点元数据中获取匹配的路由前缀
                if (context.HttpContext.GetEndpoint()?.Metadata.GetMetadata<RouteMetadata>() is { } routeMetadata)
                {
                    if (Uri.TryCreate(originalLocation, UriKind.Absolute, out var backendUri))
                    {
                        // 拼接路由前缀与后端的路径部分
                        var pathWithPrefix = $"{routeMetadata.RoutePattern.RawText.TrimEnd('/')}{backendUri.PathAndQuery}";
                        // 基于当前网关请求的地址生成最终的重定向地址
                        var gatewayBaseUri = new Uri(context.HttpContext.Request.GetEncodedUrl());
                        var finalLocation = new Uri(gatewayBaseUri, pathWithPrefix).ToString();
                        
                        // 更新Location头
                        context.Response.Headers[HeaderNames.Location] = finalLocation;
                    }
                }
            }
            await Task.CompletedTask;
        });
    }
}

2. 注册自定义转换提供者 #

在Program.cs中，将自定义转换注入到YARP的反向代理服务中：

builder.Services.AddReverseProxy()
    .LoadFromConfig(builder.Configuration.GetSection("ReverseProxy"))
    .AddTransforms<GraphQLRedirectTransformProvider>(); // 注册自定义转换逻辑

3. 确保路由配置正确 #

你的YARP路由配置（appsettings.json）需明确每个API的路由前缀，示例如下：

{
  "ReverseProxy": {
    "Routes": {
      "api1-route": {
        "ClusterId": "api1-cluster",
        "Match": {
          "Path": "/api1/{**catch-all}"
        }
      },
      "api2-route": {
        "ClusterId": "api2-cluster",
        "Match": {
          "Path": "/api2/{**catch-all}"
        }
      }
    },
    "Clusters": {
      "api1-cluster": {
        "Destinations": {
          "api1-dest": {
            "Address": "http://localhost:1234/"
          }
        }
      },
      "api2-cluster": {
        "Destinations": {
          "api2-dest": {
            "Address": "http://localhost:5678/"
          }
        }
      }
    }
  }
}

4. 可选：直接在HotChocolate中配置相对路径重定向 #

如果不想在网关层统一处理，也可以在每个GraphQL API中配置返回相对路径重定向，让YARP自动补全前缀：

builder.Services.AddGraphQLServer()
    // 其他GraphQL配置...
    .ModifyHttpRequestInterceptor(async (context, next) =>
    {
        // 拦截重定向请求，替换为相对路径
        if (context.Response.StatusCode is >= 300 and < 400)
        {
            context.Response.Headers.Remove(HeaderNames.Location);
            context.Response.Headers.Add(HeaderNames.Location, "/graphql/");
        }
        await next();
    });

此方法需要每个API单独配置，灵活性不如网关层统一处理。

关键说明 #

• 自定义转换会自动匹配当前请求的路由前缀，比如请求/api1/graphql时，会把后端返回的http://localhost:1234/graphql/转换成http://网关地址/api1/graphql/
• 端口问题通过context.HttpContext.Request.GetEncodedUrl()自动获取网关的当前地址，无需手动配置
• 逻辑仅针对3xx重定向响应，不会干扰其他正常请求

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