Fork me on GitHub

Asp.NetCore源码学习[1-1]:配置[Configuration]

Asp.NetCore源码学习[1-1]:配置[Configuration]

在Asp. NetCore中,配置系统支持不同的配置源(文件、环境变量等),虽然有多种的配置源,但是最终提供给系统使用的只有一个对象,那就是ConfigurationRoot。其内部维护了一个集合,用于保存各种配置源的IConfigurationProviderIConfigurationProvider提供了对配置源的实际访问。当通过key去ConfigurationRoot查找对应的Value时,实际上会通过遍历IConfigurationProvider去查找对应的键值。 本篇文章主要描述ConfigurationRoot对象的构建过程。

本系列源码地址

1. Asp.NetCore 入口点代码

CreateWebHostBuilder(args).Build().Run();

2. Asp.NetCore 部分源码

WebHostBuilder内部维护了_configureAppConfigurationBuilder字段,其类型是 Action<WebHostBuilderContext, IConfigurationBuilder>,该委托用于对ConfigurationBuilder进行配置。首先在构造函数中先将环境变量的配置加载到 _config 字段中,用于设置默认监听目录为程序执行目录。CreateDefaultBuilder方法中通过调用ConfigureAppConfiguration方法保存委托,然后在Build方法中构建配置系统目标类ConfigurationRoot,最后通过单例模式注入到依赖系统中。

public class WebHostBuilder
{
    private Action<WebHostBuilderContext, IConfigurationBuilder> _configureAppConfigurationBuilder;
    
    private IConfiguration _config;
    
    public WebHostBuilder()
    {
        _hostingEnvironment = new HostingEnvironment();
        /// 
        _config = new ConfigurationBuilder()
            .AddEnvironmentVariables(prefix: "ASPNETCORE_")
            .Build();
    }
    
    public IWebHostBuilder ConfigureAppConfiguration(Action<WebHostBuilderContext, IConfigurationBuilder> configureDelegate)
    {
        _configureAppConfigurationBuilder += configureDelegate;
        return this;
    }
    
    public IWebHost Build()
    {
        var builder = new ConfigurationBuilder();
        //通过委托配置IConfigurationBuilder
        _configureAppConfigurationBuilder?.Invoke(_context, builder);
        //构建ConfigurationRoot
        var configuration = builder.Build();
        // register configuration as factory to make it dispose with the service provider
        services.AddSingleton<IConfiguration>(_ => configuration);
    }
}
public static IWebHostBuilder CreateDefaultBuilder(string[] args)
{
    var builder = new WebHostBuilder();
    builder.ConfigureAppConfiguration((hostingContext, config) =>
    {
        //为 IConfigurationBuilder 注册配置源(JsonConfigurationSource)
        config.AddJsonFile("appsettings.json", optional: true, reloadOnChange: true);
    });
    return builder;
}

3. 参照以上 Asp.NetCore 代码,写静态测试方法

    public class ConfigurationTest
    {
        public static void Run()
        {
            //1.实例化ConfigurationBuilder
            var builder = new ConfigurationBuilder();
            //2.增加配置源
            builder.AddJsonFile(null, "appsettings.json", true,true);
            //3.构建ConfigurationRoot对象
            var configuration = builder.Build();
            //观察ConfigurationRoot是否发生更改
            Task.Run(() => {
                ChangeToken.OnChange(() => configuration.GetReloadToken(), () => {
                    Console.WriteLine("Configuration has changed");
                });
            });
            Thread.Sleep(60000);
        }
    }

4. 通过ConfigurationBuilder类构建目标类ConfigurationRoot

ConfigurationBuilder是配置系统的构建类,通过Build方法构建配置系统的目标类ConfigurationRoot。其维护了一个用于保存IConfigurationSource的集合,IConfigurationSource用于提供IConfigurationProvider。在Build方法中,遍历IList构建IConfigurationProvider对象,然后将IConfigurationProvider集合传到ConfigurationRoot的构造函数中。代码如下:

    /// <summary>
    /// 配置系统构建类
    /// </summary>
    public class ConfigurationBuilder : IConfigurationBuilder
    {
        /// 配置源集合
        public IList<IConfigurationSource> Sources { get; } = new List<IConfigurationSource>();

        /// 增加一个新的配置源
        public IConfigurationBuilder Add(IConfigurationSource source)
        {
            if (source == null)
            {
                throw new ArgumentNullException(nameof(source));
            }
            Sources.Add(source);
            return this;
        }

        /// 通过配置源中提供的IConfigurationProvider构建配置根对象ConfigurationRoot
        public IConfigurationRoot Build()
        {
            var providers = new List<IConfigurationProvider>();
            foreach (var source in Sources)
            {
                var provider = source.Build(this);
                providers.Add(provider);
            }
            return new ConfigurationRoot(providers);
        }
    }

IConfigurationSource对象不仅仅用于创建IConfigurationProvider,还保存了构建IConfigurationProvider需要的依赖和配置选项。


4.1 ConfigurationRoot 类实现

该类通过IList进行初始化。其内部维护了类型为ConfigurationReloadToken的字段,该字段提供给外部,来进行所有配置源的监听。每个IConfigurationProvider对象同样维护了类型为ConfigurationReloadToken的字段。当IConfigurationProvider监测到配置源发生更改时,更改IConfigurationProvider.IChangeToken的状态
在构造函数中执行以下操作:

  • 1 调用IConfigurationProvider.Load()从配置源(文件、环境变量等)加载配置项
  • 2 通过ChangeToken.OnChange()方法 监听每个IConfigurationProvider.IChangeToken的状态改变,当其状态发生改变时更改ConfigurationRoot.IChangeToken的状态。(在ConfigurationRoot外部可以通过监听IChangeToken状态的改变,得知配置源发生了改变)
    /// <summary>
    /// 配置系统的根节点
    /// </summary>
    public class ConfigurationRoot : IConfigurationRoot, IDisposable
    {
        private readonly IList<IConfigurationProvider> _providers;
        private readonly IList<IDisposable> _changeTokenRegistrations;
        private ConfigurationReloadToken _changeToken = new ConfigurationReloadToken();

        /// <summary>
        /// 使用IConfigurationProvider集合初始化ConfigurationRoot
        /// </summary>
        /// <param name="providers">The <see cref="IConfigurationProvider"/>s for this configuration.</param>
        public ConfigurationRoot(IList<IConfigurationProvider> providers)
        {
            _providers = providers ?? throw new ArgumentNullException(nameof(providers));

            _changeTokenRegistrations = new List<IDisposable>(providers.Count);
            foreach (var p in providers)
            {
                p.Load();
                //将每个IConfigurationProvider的change token与ConfigurationRoot 的change token绑定
                //当IConfigurationProvider._cts.Cancel()触发时,触发当ConfigurationRoot._cts.Cancel()
                _changeTokenRegistrations.Add(ChangeToken.OnChange(() => p.GetReloadToken(), () => RaiseChanged()));
            }
        }

        public IEnumerable<IConfigurationProvider> Providers => _providers;

        /// 遍历_providers来设置、获取配置项的键值对
        public string this[string key]
        {
            get
            {
                for (var i = _providers.Count - 1; i >= 0; i--)
                {
                    var provider = _providers[i];

                    if (provider.TryGet(key, out var value))
                    {
                        return value;
                    }
                }

                return null;
            }
            set
            {
                if (!_providers.Any())
                {
                    throw new InvalidOperationException("Can't find any IConfigurationProvider");
                }

                foreach (var provider in _providers)
                {
                    provider.Set(key, value);
                }
            }
        }

        /// 获取IChangeToken,用于供外部使用者收到配置改变的消息通知 
        public IChangeToken GetReloadToken() => _changeToken;

        public void Reload()
        {
            foreach (var provider in _providers)
            {
                provider.Load();
            }
            RaiseChanged();
        }

        /// 生成一个新的change token,并触发ConfigurationRoot的change token(旧)状态改变
        private void RaiseChanged()
        {
            var previousToken = Interlocked.Exchange(ref _changeToken, new ConfigurationReloadToken());
            previousToken.OnReload();
        }

        /// <inheritdoc />
        public void Dispose()
        {
            // dispose change token registrations
            foreach (var registration in _changeTokenRegistrations)
            {
                registration.Dispose();
            }

            // dispose providers
            foreach (var provider in _providers)
            {
                (provider as IDisposable)?.Dispose();
            }
        }
    }

4.2 ConfigurationReloadToken 的实现

其使用适配器模式,通过CancellationTokenSource实现IChangeToken接口。代码如下:

    /// <summary>
    /// 用于发送更改通知
    /// </summary>
    public interface IChangeToken
    {
        /// 指示是否发生更改
        bool HasChanged { get; }

        /// 指示token是否会主动调用callbacks,false的情况下:token的消费者需要轮询 HasChanged 属性检测是否发生更改
        bool ActiveChangeCallbacks { get; }

        /// 注册回调函数, 更改发生时(HasChanged为true),会被调用(只会被调用一次)
        IDisposable RegisterChangeCallback(Action<object> callback, object state);
    }
    /// 基于CancellationTokenSource实现IChangeToken接口(适配器模式)
    public class ConfigurationReloadToken:IChangeToken
    {
        private CancellationTokenSource _cts = new CancellationTokenSource();

        /// CancellationTokenSource会主动调用callbacks,所以为true
        public bool ActiveChangeCallbacks => true;

        public bool HasChanged => _cts.IsCancellationRequested;

        public IDisposable RegisterChangeCallback(Action<object> callback, object state) => _cts.Token.Register(callback, state);

        public void OnReload() => _cts.Cancel();
    }

4.3 简述CancellationTokenSource 对象

基于协作取消模式设计的对象,用于取消异步操作或者长时间同步操作。( .NET指南/取消托管线程)

image

CancellationTokenSource对象的特点:

  • 1 CancellationTokenSource.Token是值类型,传递副本
  • 2 调用 CancellationTokenSource.Cancel 方法提供取消通知后,CancellationTokenSource.Token的状态发生改变,调用callbacks,并改变所有Token副本的状态
  • 3 需要调用dispose释放CancellationTokenSource
  • 4 多次调用CancellationTokenSource.Cancel,callbacks也只会执行一次
  • 5 再CancellationTokenSource.Cancel之后,新注册的callback同样也会被执行

4.4 通过ChangeToken.OnChange 静态方法实现更改通知的持续消费

由于CancellationTokenSource.Cancel只会触发一次callbacks,需要ChangeToken.OnChange来实现持续监听取消通知。
实现原理:每次需要发生更改通知时,首先生成一个新的cts,然后改变旧的cts状态,触发回调函数,最后将新的cts与回调函数绑定。

    /// <summary>
    /// 将changeToken消费者注册到IChangeToken的回调函数中,并实现IChangeToken状态改变的持续消费
    /// </summary>
    public static class ChangeToken
    {
        /// 为changetoken生产者绑定消费者. 
        /// 1.在IChangeToken的状态未改变的情况下,生产者每次返回相同的IChangeToken
        /// 2.状态改变时,生产者生成新的IChangeToken,消费者执行响应动作,为新的IChangeToken绑定消费者,释放旧的IChangeToken
        public static IDisposable OnChange(Func<IChangeToken> changeTokenProducer, Action changeTokenConsumer)
        {
            if (changeTokenProducer == null)
            {
                throw new ArgumentNullException(nameof(changeTokenProducer));
            }
            if (changeTokenConsumer == null)
            {
                throw new ArgumentNullException(nameof(changeTokenConsumer));
            }
            return new ChangeTokenRegistration<Action>(changeTokenProducer, callback => callback(), changeTokenConsumer);
        }

        private class ChangeTokenRegistration<TState> : IDisposable
        {
            private readonly Func<IChangeToken> _changeTokenProducer;
            private readonly Action<TState> _changeTokenConsumer;
            private readonly TState _state;
            private IDisposable _disposable;//用于保存当前正在使用的 IChangeToken

            private static readonly NoopDisposable _disposedSentinel = new NoopDisposable();

            public ChangeTokenRegistration(Func<IChangeToken> changeTokenProducer, Action<TState> changeTokenConsumer, TState state)
            {
                _changeTokenProducer = changeTokenProducer;
                _changeTokenConsumer = changeTokenConsumer;
                _state = state;

                var token = changeTokenProducer();

                RegisterChangeTokenCallback(token);
            }

            /// 1.先执行消费者动作,再绑定新的token,防止消费者执行并发动作
            /// 2.否则的话可能出现以下情况:如果在绑定新token的回调方法后,并且在执行callback之前,新token的状态发生了改变,此时第二次callback也会执行,这样会造成callback的并发执行。
            private void OnChangeTokenFired()
            {
                var token = _changeTokenProducer();

                try
                {
                    _changeTokenConsumer(_state);
                }
                finally
                {
                    RegisterChangeTokenCallback(token);
                }
            }

            private void RegisterChangeTokenCallback(IChangeToken token)
            {
                var registraton = token.RegisterChangeCallback(s => ((ChangeTokenRegistration<TState>)s).OnChangeTokenFired(), this);
                SetDisposable(registraton);
            }

            /// 1.将当前使用的IChangeToken保存到_disposable字段中
            /// 2.如果本对象已经释放,立刻释放新产生的 IChangeToken
            /// 3.已经失效的 IChangeToken 由于已经不被引用,等待GC自动释放(为什么不手动释放?)
            private void SetDisposable(IDisposable disposable)
            {
                // 读取当前保存的 IChangeToken
                var current = Volatile.Read(ref _disposable);

                // 如果本对象已经释放,立刻释放新产生的IChangeToken
                if (current == _disposedSentinel)
                {
                    disposable.Dispose();
                    return;
                }

                // 否则更新_disposable字段,返回原值
                var previous = Interlocked.CompareExchange(ref _disposable, disposable, current);

                // current = 之前的 IChangeToken
                
                if (previous == _disposedSentinel)
                {
                    // 更新失败 说明对象已释放 previous = _disposedSentinel
                    // 本对象已经释放,立刻释放新产生的IChangeToken
                    disposable.Dispose();
                }
                else if (previous == current)
                {
                    // 更新成功 previous 是之前的 IChangeToken
                }
                else
                {
                    // 如果其他人为 _disposable赋值,且值不为 _disposedSentinel
                    // 会造成对象未释放、更新失败的情况
                    throw new InvalidOperationException("Somebody else set the _disposable field");
                }
            }

            // 释放当前保存的 IChangeToken,将字段赋值为_disposedSentinel
            public void Dispose()
            {
                Interlocked.Exchange(ref _disposable, _disposedSentinel).Dispose();
            }

            private class NoopDisposable : IDisposable
            {
                public void Dispose()
                {
                }
            }
        }
    }

4.6 ChangeToken.OnChange 测试方法

该测试方法通过一个ChangeTokenProducer类来模拟内部的状态改变。通过内部维护一个ConfigurationReloadToken,可以向外部发出更改通知(一次性)。为了实现向外部持续发出更改通知,可以在更改ConfigurationReloadToken状态之前,重新实例化有一个新的IChangeToken,供外部重新绑定回调方法。

    class ChangeTokenTest
    {
        public static void Run() {
            var ctsProducer = new ChangeTokenProducer();
            var subscriber = ChangeToken.OnChange(() => ctsProducer.GetReloadToken(), () =>
            {
                Console.WriteLine("消费者观察到改变事件");
            });
            Console.ReadLine();
        }

        /// <summary>
        /// 假设该类需要在内部状态发生改变时向外界发送更改通知
        /// </summary>
        private class ChangeTokenProducer
        {
            // cts只能执行一次相应动作
            private ConfigurationReloadToken _changetoken = new ConfigurationReloadToken();

            /// <summary>
            /// 模拟状态改变
            /// </summary>
            public ChangeTokenProducer()
            {
                Task.Run(()=> {
                    while (true)
                    {
                        Thread.Sleep(3000);//模拟耗时
                        //内部状态发生改变,通知外部
                        RaiseChanged();
                    }
                });
            }

            public IChangeToken GetReloadToken () => _changetoken;

            private void RaiseChanged() {
                //产生新的cts
                var previousToken = Interlocked.Exchange(ref _changetoken, new ConfigurationReloadToken());
                //触发老的cts动作
                //外界执行响应动作时,通过GetReloadToken()获取新的cts,执行相应动作,并重新绑定回调函数
                previousToken.OnReload();
            }
        }
    }

5. IConfigurationSource 的实现

IConfigurationSource拥有一个实现IFileProvider接口的类属性。默认实现为PhysicalFileProvider类,文件监控目录默认为程序集根目录。该类提供文件的访问和监控功能。在Build方法中实例化JsonFileConfigurationProvider,并将自身传递进去。
在.NetCore源码中JsonConfigurationSource 是继承 抽象类FileConfigurationSource 的。此处合并了两个类的代码。

    public class JsonFileConfigurationSource : IConfigurationSource
    {
        public IFileProvider FileProvider { get; set; }

        public IConfigurationProvider Build(IConfigurationBuilder builder) {
            EnsureDefaults(builder);
            return new JsonFileConfigurationProvider(this);
        }

        public void EnsureDefaults(IConfigurationBuilder builder)
        {
            FileProvider = FileProvider ?? builder.GetFileProvider();
        }
    }
    
    public static class FileConfigurationExtensions 
    {
        /// 获取默认的IFileProvider,root目录默认为程序集根目录(AppContext.BaseDirectory)
        public static IFileProvider GetFileProvider(this IConfigurationBuilder builder)
        {
            if (builder == null)
            {
                throw new ArgumentNullException(nameof(builder));
            }
            return new PhysicalFileProvider(AppContext.BaseDirectory ?? string.Empty);
        }
    }

6. IConfigurationProvider 的实现

在Core的源码中继承关系为JsonConfigurationProvider: FileConfigurationProvider:ConfigurationProvider:IConfigurationProvider
。本项目代码合并了JsonConfigurationProvider FileConfigurationProvider这两个类

6.1 ConfigurationProvider 的实现

该类使用一个字典用于保存配置项的字符串键值对。并拥有一个类型为 ConfigurationReloadToken 的字段。在配置文件发生更改时,_reloadToken的状态发生改变,外部可以通过观察该字段的状态来得知配置文件发生更改。

    /// <summary>
    /// 配置提供者抽象类
    /// </summary>
    public abstract class ConfigurationProvider : IConfigurationProvider
    {
        private ConfigurationReloadToken _reloadToken = new ConfigurationReloadToken();

        /// 初始化存储配置的字典,键值忽略大小写
        protected ConfigurationProvider()
        {
            Data = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
        }
        
        /// 存储配置的键值对,protected只能在子类中访问
        protected IDictionary<string, string> Data { get; set; }

        /// 读取键值
        public virtual bool TryGet(string key, out string value) => Data.TryGetValue(key, out value);

        /// 设置键值
        public virtual void Set(string key, string value) => Data[key] = value;

        /// 加载配置数据源,使用virtual修饰符,在子类中实现重写
        public virtual void Load()
        { }

        public IChangeToken GetReloadToken()
        {
            return _reloadToken;
        }

        /// <summary>
        /// 触发change token,并生成一个新的change token
        /// </summary>
        protected void OnReload()
        {
            //原子操作:赋值并返回原始值
            var previousToken = Interlocked.Exchange(ref _reloadToken, new ConfigurationReloadToken());
            previousToken.OnReload();
        }
    }

6.2 JsonFileConfigurationProvider 的实现

该类继承于抽象类ConfigurationProvider
在构造函数中监听 FileProvider.Watch() 方法返回的IChangeToken,收到更改通知时执行以下两个动作,一个是重新读取文件流,加载到字典中;另一个是改变_reloadToken 的状态,用于通知外部:已经重新加载配置文件。由于在本项目中直接引用了MS的PhysicalFileProvider,而该类监听文件返回的是微软的IChangeToken。为了兼容项目代码,通过一个适配类来转换接口。

    public class JsonFileConfigurationProvider : ConfigurationProvider, IDisposable
    {
        private readonly IDisposable _changeTokenRegistration;
        
        public JsonFileConfigurationSource Source { get; }

        public JsonFileConfigurationProvider(JsonFileConfigurationSource source)
        {
            if (source == null)
            {
                throw new ArgumentNullException(nameof(source));
            }
            Source = source;

            if (Source.ReloadOnChange && Source.FileProvider != null)
            {
                //1.IFileProvider.Watch(string filter) 返回IChangeToken
                //2.绑定IChangeToken的回调函数(1。生成新的IChangeToken 2.读取配置文件、向ConfigurationRoot传递消息)
                //3.检测到文件更改时,触发回调
                //4.为新的IChangeToken绑定回调函数
                _changeTokenRegistration = ChangeToken.OnChange(
                    () => new IChangeTokenAdapter(Source.FileProvider.Watch(Source.Path)),
                    () => {
                        Thread.Sleep(Source.ReloadDelay);
                        Load(reload: true);
                    });
            }
        }

        //重新加载文件并向IConfigurationRoot传递更改通知
        private void Load(bool reload)
        {
            var file = Source.FileProvider?.GetFileInfo(Source.Path);
            if (file == null || !file.Exists)
            {
                if (Source.Optional || reload) // Always optional on reload
                {
                    Data = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
                }
                else
                {
                    //处理异常
                }
            }
            else
            {
                // Always create new Data on reload to drop old keys
                if (reload)
                {
                    Data = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
                }
                using (var stream = file.CreateReadStream())
                {
                    try
                    {
                        Load(stream);
                    }
                    catch (Exception e)
                    {
                        HandleException(new FileLoadExceptionContext() { Exception = e, Provider = this, Ignore = true });
                    }
                }
            }
            //触发IConfigurationProvider._cts.Cancel(),向IConfigurationRoot传递更改通知
            OnReload();
        }

        public override void Load()
        {
            Load(reload: false);
        }

        /// 从文件流中加载数据到IConfigurationProvider的Data中
        public void Load(Stream stream) {
            try
            {
                //.NetCore3.0使用JsonDocument读取json文件,生成结构化文档:
                /// [key] 节点1-1:节点2-1:节点3-1 [value] Value1
                /// [key] 节点1-1:节点2-2:节点3-2 [value] Value2
                /// Data = JsonConfigurationFileParser.Parse(stream);
                //此处使用Newtonsoft.Json,简单的序列化为普通键值对
                using (StreamReader sr = new StreamReader(stream))
                {
                    String jsonStr = sr.ReadToEnd();
                    Data = Newtonsoft.Json.JsonConvert.DeserializeObject<Dictionary<string, string>>(jsonStr);
                }
            }
            catch (Exception e)
            {
                throw new FormatException("读取文件流失败", e);
            }
        }

        public void Dispose() => Dispose(true);

        /// 释放_changeTokenRegistration
        protected virtual void Dispose(bool disposing)
        {
            _changeTokenRegistration?.Dispose();
        }
    }

    /// 适配器类 
    /// 将Microsoft.Extensions.Primitives.IChangeToken转换为CoreWebApp.Primitives.IChangeToken
    public class IChangeTokenAdapter : IChangeToken
    {
        public IChangeTokenAdapter(IChangeTokenMS msToken)
        {
            MsToken = msToken ?? throw new ArgumentNullException(nameof(msToken));
        }

        private IChangeTokenMS MsToken { get; set; }

        public bool HasChanged => MsToken.HasChanged;

        public bool ActiveChangeCallbacks => MsToken.ActiveChangeCallbacks;

        public IDisposable RegisterChangeCallback(Action<object> callback, object state)
        {
            return MsToken.RegisterChangeCallback(callback, state);
        }
    }

7. ConfigurationSection 类的实现

{
  "OptionV1": {
    "OptionV21": "ValueV21",
    "OptionV22": {
      "OptionV31": "ValueV31",
      "OptionV32": "ValueV32"
    }
  }
}

对于如上的配置文件会保存为key "OptionV1:OptionV22:OptionV31" value "ValueV31"的格式,这样同时将节点间的层级关系也保存了下来。通过ConfigurationRoot访问键值需要提供键的全路径。ConfigurationSection 类相当于定位了某个节点,通过ConfigurationSection访问键值只需要通过相对路径。


结语

到此为止,ConfigurationRoot已经构建完成,然后通过DI模块以单例模式注入到系统中。在控制器中可以通过IConfiguration访问到所有配置源的键值对,并且当配置文件发生改变时重新加载IConfigurationProvider。下篇文章将会讲述从如何通过强类型IOptions 访问配置项。

posted @ 2019-08-07 22:45  Kane_Blake  阅读(678)  评论(0编辑  收藏  举报