为 HttpClient 注册自定义请求标头

为 HttpClient 注册自定义请求标头

前言

上次，我们介绍了《​​在 ASP.NET Core 中使用 HTTP 标头传播​​》。

但是有时候，当服务间需要互相调用时，也需要将创建一些自定义标头传播到目标服务。

比如， ServiceA 已经进行了身份验证，那么当它调用 ServiceB 时，不需要传播 HTTP 标头 ​​Authorization​​​，可以自定义 ​​x-user-id​​ 标头， 直接告诉 ServiceB 当前用户 ID。

1. 使用 HttpRequestMessage

最简单的方法，创建 HttpRequestMessage 实例直接添加标头。

ServiceA 的实现代码如下：

[HttpGet]public async Task Get(){ var userId = GetUserId();//从认证信息中获取 var request = new HttpRequestMessage(HttpMethod.Get, "/ServiceB"); if (!string.IsNullOrWhiteSpace(userId)) { request.Headers.Add("x-user-id", userId); } var client = _clientFactory.CreateClient("ServiceB-Client"); var response = await client.SendAsync(request); return await response.Content.ReadAsStringAsync();}

但是，这种解决方案存在一个问题：

如果有多个调用其他服务的方法，我们需要重复相同的逻辑；

2.使用 DelegatingHandler

在 ASP.NET Core 中，我们经常会使用 Middleware, 它使用管道模式，可以对服务收到的请求进行处理：

而 ​​DelegatingHandler​​ 提供了几乎相同的概念，但却是在 HttpClient 发出传出请求时。

实现代码如下：

public class UserIdDelegatingHandler : DelegatingHandler{ protected override async Task SendAsync(HttpRequestMessage request, CancellationToken cancellationToken) { var userId = GetUserId();//从认证信息中获取 if (!string.IsNullOrWhiteSpace(userId) && !request.Headers.Contains("x-user-id")) { request.Headers.TryAddWithoutValidation(Headers.CorrelationId, userId); } return await base.SendAsync(request, cancellationToken); }}

然后修改 Startup.cs，注册 UserIdDelegatingHandler：

public void ConfigureServices(IServiceCollection services){ services.AddHttpClient("ServiceB-Client", options => options.BaseAddress = new Uri(" .AddHttpMessageHandler() ......}

现在，我们可以从调用服务的代码中删除有关注册自定义标头的所有代码：

[HttpGet]public async Task Get(){ var client = _clientFactory.CreateClient("ServiceB-Client"); var response = await client.GetAsync("/ServiceB"); return await response.Content.ReadAsStringAsync();}

结论

如果我们想向 HttpClient 添加任何标头，无需修改业务代码，只需调用 ​​.AddHttpMessageHandler​​ 注册新的 DelegatingHandler 即可。

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。