HeaderForwarder组件不仅能够从当前接收请求提取指定的HTTP报头,并自动将其添加到任何一个通过HttpClient发出的请求中,它同时也提供了一种基于Context/ContextScope的编程模式是我们可以很方便地将任何报头添加到指定范围内的所有由HttpClient发出的请求中。上篇介绍了HeaderForwarder组件的使用方式,现在我们来简单聊聊该组件的设计和实现原理。[源代码从这里下载]
目录
一、HeaderForwardObserver
二、HttpClientObserver
三、HeaderForwaderStartupFilter
四、HttpInvocationContext/HttpInvocationContextScope
五、OutgoingHeaderCollectionProvider
六、服务注册
一、HeaderForwardObserver
HeaderForwarder组件利用HeaderForwardObserver对HttpClient进行拦截,并将需要的报头添加到由它发出的请求消息中,我们曾经在《 四种为HttpClient添加默认请求报头的解决方案 》一文中介绍过这种方案,这也是大部分APM自动添加跟踪报头的解决方案。具体的原理其实很简单:当HttpClient发送请求过程中会利用DiagnosticListener触发一些列事件,并在事件中提供相应的对象,比如发送的HttpRequestMessage和接收的HttpResponseMessage。如果我们需要这个过程进行干预,只需要订阅相应的事件并将干预操作实现在提供的回调中。《ASP.NET Core 3框架揭秘》第8“诊断日志”具有对DiagnosticListener的详细介绍。
HeaderForwarder用来添加请求报头的是一个类型为HeaderForwardObserver的对象。在介绍该类型之前,我们得先来介绍如下这个IOutgoingHeaderCollectionProvider接口,顾名思义,它用来提供需要被添加的所有HTTP请求报头。
public interface IOutgoingHeaderCollectionProvider{ IDictionary GetHeaders();}
如下所示的是HeaderForwardObserver的定义。如代码片段所示,HeaderForwardObserver实现了IObserver> 接。在实现的OnNext中,通过对事件名称( System.Net.Http.HttpRequestOut.Start )的比较订阅了HttpClient在发送请求前触发的事件,并从提供的参数提取出表示待发送请求的HttpRequestMessage对象(对应Request属性)。有了这个待发送的请求,我们只需要从构造函数中注入的IOutgoingHeaderCollectionProvider 对象提取出所有报头列表,并将其添加这个HttpRequestMessage对象中即可。
public sealed class HeaderForwardObserver : IObserver>{ private static Func _requestAccessor; private readonly IOutgoingHeaderCollectionProvider _provider; public HeaderForwardObserver(IOutgoingHeaderCollectionProvider provider) { _provider = provider ?? throw new ArgumentNullException(nameof