【.NET】聊聊 IChangeToken 接口

由于两个月的奋战,导致很久没更新了。就是上回老周说的那个产线和机械手搬货的项目,好不容易等到工厂放假了,我就偷偷乐了。当然也过年了,老周先给大伙伴们拜年了,P话不多讲,就祝大家身体健康、生活愉快。其实生活和健康是密不可分的,想活得好,就得健康。包括身体健康、思想健康、心理健康、精神健康。不能以为我无病无痛就很健康,你起码要全方位健康。

不管你的工作是什么,忙或者不忙,报酬高或低,但是,人,总得活,总得过日子。咱们最好多给自己点福利,多整点可以自娱自乐的东西,这就是生活。下棋、打游戏、绘画、书法、钓鱼、飙车、唢呐……不管玩点啥,只要积极正向的就好,可以大大降低得抑郁症、高血压的机率;可以减少70%无意义的烦恼;可以降低跳楼风险;在这个礼崩乐坏的社会环境中,可以抵御精神污染……总之,益处是大大的有。

然后老周再说一件事,一月份的时候常去工厂调试,也认识了机械臂厂商派的技术支持——吴大工程师。由于工厂所处地段非常繁华,因此每次出差,午饭只能在附近一家四川小吃店解决。毕竟这方圆百十里也仅此一家。不去那里吃饭除非自带面包蹲马路边啃,工厂不供食也不供午休场所。刚开始几次出差还真的像个傻子似的蹲马路边午休。后来去多了,直接钻进工厂的会议室睡午觉。

有一天吃午饭时,吴老师说:你说什么样的人编程水平最高?

我直接从潜意识深处回答他:我做一个排序,仅供参考。编程水平从高到低排行:

1、黑客。虽然大家都说黑客一代不如一代,但目前来说,这群人还是最强的;

2、纯粹技术爱好者;

3、著名开源项目贡献者。毕竟拿不出手的代码也不好意思与人分享;

4、做过许多项目的一线开发者。我强调的项目数量多,而不是长年只维护一个项目的。只有数量多你学到的才多;

5、社区贡献较多者,这个和3差不多。不过,老周认为的社区贡献就是不仅提供代码,还提供文档、思路、技巧等;

6、刚入坑但基础较好的开发者;

7、培训机构的吹牛专业户;

8、大学老师/教授;

9、短视频平台上的砖家、成宫人士;

10、刚学会写 main 函数的小朋友。

==========================================================================================================

下面进入主题,咱们今天聊聊 IChangeToken。它的主要功能是提供更改通知。比如你的配置源发生改变了,要通知配置的使用者重新加载。你可能会疑惑,这货跟使用事件有啥区别?这个老周也不好下结论,应该是为异步代码准备的吧。

下面是 IChangeToken 接口的成员:

bool HasChanged { get; }
bool ActiveChangeCallbacks { get; }
IDisposable RegisterChangeCallback(Action<object?> callback, object? state);

这个 Change Token 思路很清奇,实际功能类似事件,就是更改通知。咱们可以了解一下其原理,但如果你觉得太绕,不想了解也没关系的。在自定义配置源时,咱们是不需要自己写 Change Token 的,框架已有现成的。我们只要知道要触发更改通知时调用相关成员就行。

如果你想看源码的话,老周可以告你哪些文件(github 项目是 dotnet\runtime):

1、runtime-main\src\libraries\Common\src\Extensions\ChangeCallbackRegistrar.cs:这个主要是 UnsafeRegisterChangeCallback 方法,用于注册回调委托;

2、runtime-main\src\libraries\Microsoft.Extensions.Primitives\src\ChangeToken.cs:这个类主要是提供静态的辅助方法,用于注册回调委托。它的好处是可以循环——注册回调后,触发后委托被调用;调用完又自动重新注册,使得 Change Token 可以多次触发;

3、runtime-main\src\libraries\Microsoft.Extensions.Primitives\src\CancellationChangeToken.cs:这个类是真正实现 IChangeToken 接口的;

4、runtime-main\src\libraries\Microsoft.Extensions.Configuration\src\ConfigurationReloadToken.cs:这个也是实现 IChangeToken 接口,而且它才是咱们今天的主角,该类就是为重新加载配置数据而提供的。调用它的 OnReload 方法可以触发更改通知。

看了上面这些,你可能更疑惑了。啥原理?为啥 Token 只能触发一次?为何要重新注册回调?

咱们用一个简单例子演练一下。

static void Main(string[] args)
{
    CancellationTokenSource cs = new();
    // 这里获取token
    CancellationToken token = cs.Token;
    // token 可以注册回调
    token.Register(() =>
    {
        Console.WriteLine("你按下了【K】键");
    });
    // 启动一个新task
    Task myTask = Task.Run(() =>
    {
        // 等待输入,如果按下【K】键,就让CancellationTokenSource取消
        ConsoleKeyInfo keyInfo;
        while(true)
        {
            keyInfo = Console.ReadKey(true);
            if(keyInfo.Key == ConsoleKey.K)
            {
                // 取消
                cs.Cancel();
                break;
            }
        }
    });
    // 主线程等待任务完成
    Task.WaitAll(myTask);
}

CancellationTokenSource 类表示一个取消任务的标记,访问它的 Token 属性可以获得一个 CancellationToken 结构体实例,可以检索它的 IsCancellationRequested 属性以明确是否有取消请求(有则true,无则false)。

还有更重要的,CancellationToken 结构体的 Register 方法可以注册一个委托作为回调,当收到取消请求后会触发这个委托。对的,这个就是 Change Token 灵魂所在了。一旦回调被触发后,CancellationTokenSource 就处于取消状态了,你无法再次触发,除非重置或重新实例化。这就是回调只能触发一次的原因。

下面,咱们完成一个简单的演示——用数据库做配置源。在 SQL Server 里面随便建个数据库,然后添加一个表,名为 tb_configdata。它有四个字段:

CREATE TABLE [dbo].[tb_configdata](
    [ID] [int] NOT NULL,
    [config_key] [nvarchar](15) NOT NULL,
    [config_value] [nvarchar](30) NOT NULL,
    [remark] [nvarchar](50) NULL,
 CONSTRAINT [PK_tb_configdata] PRIMARY KEY CLUSTERED 
(
    [ID] ASC,
    [config_key] ASC
)WITH (PAD_INDEX = OFF, STATISTICS_NORECOMPUTE = OFF, IGNORE_DUP_KEY = OFF, ALLOW_ROW_LOCKS = ON, ALLOW_PAGE_LOCKS = ON, OPTIMIZE_FOR_SEQUENTIAL_KEY = OFF) ON [PRIMARY]
) ON [PRIMARY]
GO

ID和config_key设为主键,config_value 是配置的值,remark 是备注。备注字段其实可以不用,但实际应用的时候,可以用来给配置项写点注释。

然后,在程序里面咱们用到 EF Core,故要先生成与表对应的实体类。这里老周就不用工具了,直接手写更有效率。

// 实体类
public class MyConfigData
{
    public int ID { get; set; }
    public string ConfigKey { get; set; } = string.Empty;
    public string ConfigValue { get; set; } = string.Empty;
    public string? Remark { get; set; }
}

// 数据库上下文对象
public class DemoConfigDBContext : DbContext
{
    public DbSet<MyConfigData> ConfigData => Set<MyConfigData>();

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        optionsBuilder.UseSqlServer("Data Source=DEV-PC\\SQLTEST;Initial Catalog=Demo;Integrated Security=True;Connect Timeout=30;Encrypt=True;Trust Server Certificate=True;Application Intent=ReadWrite;Multi Subnet Failover=False");
    }

    protected override void OnModelCreating(ModelBuilder modelbd)
    {
        modelbd.Entity<MyConfigData>()
            .ToTable("tb_configdata")
            .HasKey(c => new { c.ID, c.ConfigKey });
        modelbd.Entity<MyConfigData>()
            .Property(c => c.ConfigKey)
            .HasColumnName("config_key");
        modelbd.Entity<MyConfigData>()
            .Property(c => c.ConfigValue)
            .HasColumnName("config_value");
        modelbd.Entity<MyConfigData>()
            .Property(c => c.Remark)
            .HasColumnName("remark");
    }
}

上述代码的情况特殊,实体类的名称和成员名称与数据表并不一致,所以在重写 OnModelCreating 方法时,需要进行映射。

1、ToTable("tb_configdata") 告诉 EF 实体类对应的数据表是 tb_configdata;

2、HasKey(c => new { c.ID, c.ConfigKey }):表明该实体有两个主键——ID和ConfigKey。这里指定的是实体类的属性,而不是数据表的字段名,因为后面咱们会进行列映射;

3、HasColumnName("config_key"):告诉 EF,实体的 ConfigKey 属性对应的是数据表中 config_key。后面的几个属性的道理一样,都是列映射。

做映射就类似于填坑,如果你不想挖坑,那就直接让实体类名与表名一样,属性名与表字段(列)一样,这样就省事多了。不过,在实际使用中真没有那么美好。很多时候数据库是小李负责的,人家早就建好了,存储过程都写了几万个了。后面前台程序是老张来开发,对老张来说,要么把实体的命名与数据库的一致,要么就做一下映射。多数情况下是要映射的,毕竟很多时候数据库对象的命名都比较奇葩。尤其有上千个表的时候,为了看得顺眼,很多人喜欢这样给数据表命名:ta_XXX、ta_YYY、tb_ZZZ、tc_FFF、tx_PPP、ty_EEE、tz_WWW。还有这样命名的:m1_Report、m2_ReportDetails…… m105_TMD、m106_WNM、m107_DOUBI。

这种命名用在实体类上面确实很不优雅,所以映射就很必要了。

此处咱们不用直接实现 IConfigurationProvider 接口,而是从 ConfigurationProvider 类派生就行了。自定义配置源的东东老周以前写过,只是当时没有实现更改通知。

public class MyConfigurationProvider : ConfigurationProvider, IDisposable
{
    private System.Threading.Timer theTimer;

    public MyConfigurationProvider()
    {
        theTimer = new Timer(OnTimer, null, 100, 10000);
    }

    private void OnTimer(object? state)
    {
        // 先调用Load方法,然后用OnReload触发更新通知
        Load();
        OnReload();
    }

    public void Dispose()
    {
        theTimer?.Change(0, 0);
        theTimer?.Dispose();
    }

    public override void Load()
    {
        // 先读取一下
        using DemoConfigDBContext dbctx = new();
        // 如果无数据,先初始化
        if(dbctx.ConfigData.Count() == 0)
        {
            InitData(dbctx.ConfigData);
        }
        // 加载数据
        Data = dbctx.ConfigData.ToDictionary(k => k.ConfigKey, k => (string?)k.ConfigValue);

        // 本地函数
        void InitData(DbSet<MyConfigData> set)
        {
            int _id = 1;
            set.Add(new()
            {
                ID = _id,
                ConfigKey = "page_size",
                ConfigValue = "25"
            });
            _id += 1;
            set.Add(new()
            {
                ID = _id,
                ConfigKey = "format",
                ConfigValue = "xml"
            });
            _id += 1;
            set.Add(new()
            {
                ID = _id,
                ConfigKey = "limited_height",
                ConfigValue = "1450"
            });
            _id += 1;
            set.Add(new()
            {
                ID = _id,
                ConfigKey = "msg_lead",
                ConfigValue = "TDXA_"
            });
            // 保存数据
            dbctx.SaveChanges();
        }
    }

}

由于老周不知道怎么监控数据库更新,最简单的办法就是用定时器循环检查。重点是重写 Load 方法,完成加载配置的逻辑。Load 方法覆写后不需要调用 base 的 Load 方法,因为基类的方法是空的,调用了也没毛用。

在 Timer 对象调用的方法(OnTimer)中,先调用 Load 方法,再调用 OnReload 方法。这样就可以在加载数据后触发更改通知。

然后实现 IConfigurationSource 接口,提供 MyConfigurationProvider 实例。

public class MyConfigurationSource : IConfigurationSource
{
    public IConfigurationProvider Build(IConfigurationBuilder builder)
    {
        return new MyConfigurationProvider();
    }
}

默认的配置源有JSON文件、命令行、环境变量等,为了排除干扰,便于查看效果,在 Main 方法中咱们先把配置源列表清空,再添加咱们自定义的配置源。

var builder = WebApplication.CreateBuilder(args);
// 清空配置源
builder.Configuration.Sources.Clear();
// 添加配置源到Sources
builder.Configuration.Sources.Add(new MyConfigurationSource());
var app = builder.Build();

最后,可以做个简单测试,直接注入 Mini-API 中读取配置。

app.MapGet("/", (IConfiguration config) =>
{
    StringBuilder bd = new();
    foreach(var kp in config.AsEnumerable())
    {
        bd.AppendLine($"{kp.Key} = {kp.Value}");
    }
    return bd.ToString();
});

运行效果如下:

这时候咱们到数据库里把配置值改一下。

update tb_configdata
    set config_value = N'55'
    where config_key = N'page_size'

update tb_configdata
    set config_value = N'1900'
    where config_key = N'limited_height'

接着回应用程序的页面,刷新一下,配置值已更新。

这里你可能会有个疑问:连接字符串硬编码了不太好,要不写在配置文件中,可是,写在JSON文件中咱们怎么获取呢?毕竟 ConfigurationProvider 不使用依赖注入。

IConfigurationSource 不是有个 Build 方法吗?Build 方法不是有个参数是 IConfigurationBuilder 吗?用它,用它,狠狠地用它。

public class MyConfigurationSource : IConfigurationSource
{
    public IConfigurationProvider Build(IConfigurationBuilder builder)
    {
        // 此处可以临时build一个配置树,就能获取到JSON配置文件里面的连接字符串了
        var config = builder.Build();
        string connStr = config["ConnectionStrings:test"]!;
        return new MyConfigurationProvider(connStr);
    }
}

前面定义的一些类也要改一下。

先是 MyConfigurationProvider 的构造函数。

public class MyConfigurationProvider : ConfigurationProvider, IDisposable
{
    private System.Threading.Timer theTimer;
    private string connectString;

    public MyConfigurationProvider(string cnnstr)
    {
        connectString = cnnstr;
        ……
    }

    ……
}

DemoConfigDBContext 类是连接字符串的最终使用者,所以也要改一下。

public class DemoConfigDBContext : DbContext
{
    private string connStr;

    public DemoConfigDBContext(string connectionString)
    {
        connStr = connectionString;
    }

    ……

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        optionsBuilder.UseSqlServer(connStr);
    }
}

在appsettings.json 文件中配置连接字符串。

{
  "Logging": {
    ……
  },
  "AllowedHosts": "*",
  "ConnectionStrings": {
    "test": "Data Source=DEV-PC\\SQLTEST;Initial Catalog=Demo;Integrated Security=True;Connect Timeout=30;Encrypt=True;Trust Server Certificate=True;Application Intent=ReadWrite;Multi Subnet Failover=False"
  }
}

回到 Main 方法,咱们还得加上 JSON 配置源。

var builder = WebApplication.CreateBuilder(args);
// 清空配置源
builder.Configuration.Sources.Clear();
// 添加配置源到Sources
builder.Configuration.AddJsonFile("appsettings.json");
builder.Configuration.Sources.Add(new MyConfigurationSource());
var app = builder.Build();

其他的不变。

-----------------------------------------------------------------------------------------------------

接下来,咱们弄个一对多的例子。逻辑是这样的:启动程序显示主窗口,接着创建五个子窗口。主窗口上有个大大的按钮,点击后,五个子窗口会收到通知。大概就这个样子:

子窗口名为 TextForm,代码如下:

internal class TestForm : Form
{
    private IDisposable _changeTokenReg;
    private TextBox _txtMsg;
    public TestForm(Func<IChangeToken?> getToken)
    {
        // 初始化子级控件
        _txtMsg = new()
        {
            Dock = DockStyle.Fill,
            Margin = new Padding(5),
            Multiline = true,
            ScrollBars = ScrollBars.Vertical
        };
        Controls.Add(_txtMsg);

        _changeTokenReg = ChangeToken.OnChange(getToken, OnCallback);
    }

    // 回调方法
    void OnCallback()
    {
        DateTime curtime = DateTime.Now;
        string str = $"{curtime.ToLongTimeString()} 新年快乐\r\n";
        _txtMsg.BeginInvoke(() =>
        {
            _txtMsg.AppendText(str);
        });
    }

    protected override void Dispose(bool disposing)
    {
        // 释放对象
        if (disposing)
        {
            _changeTokenReg?.Dispose();
        }
        base.Dispose(disposing);
    }
}

窗口上只放了一个文本框。上面代码中,使用了 ChangeToken.OnChange 静态方法,为 Change Token 注册回调委托,本例中回调委托绑定的是 OnCallback 方法,也就是说:当 Change Token 触发后会在文本框中追加文本。OnChange 静态方法有两个重载:

// 咱们示例中用的是这个版本
static IDisposable OnChange(Func<IChangeToken?> changeTokenProducer, Action changeTokenConsumer);
// 这是另一个重载
static IDisposable OnChange<TState>(Func<IChangeToken?> changeTokenProducer, Action<TState> changeTokenConsumer, TState state);

上述例子用的是第一个,其实里面调用的也是第二个重载,只是把咱们传递的 OnCallback 方法当作 TState 传进去了。

请大伙伴暂时记住 changeTokenProducer 和 changeTokenConsumer 这两参数。changeTokenProducer 也是一个委托,返回 IChangeToken。用的时候一定要注意,每次触发之前,Change Token 要先创建新实例。注意是先创建新实例再触发,否则会导致无限。尽管内部会判断 HasChanged 属性,可问题是这个判断是在注册回调之后的。这个是跟 Change Token 的清奇逻辑有关,咱们看看 OnChage 的源代码就明白了。

 public static IDisposable OnChange<TState>(Func<IChangeToken?> changeTokenProducer, Action<TState> changeTokenConsumer, TState state)
 {
     if (changeTokenProducer is null)
     {
         ThrowHelper.ThrowArgumentNullException(ExceptionArgument.changeTokenProducer);
     }
     if (changeTokenConsumer is null)
     {
         ThrowHelper.ThrowArgumentNullException(ExceptionArgument.changeTokenConsumer);
     }

     return new ChangeTokenRegistration<TState>(changeTokenProducer, changeTokenConsumer, state);
 }

简单来说,就是返回一个 ChangeTokenRegistration 实例,这是个私有类,咱们是访问不到的,以 IDisposable 接口公开。其中,它有两个方法是递归调用的:

private void OnChangeTokenFired()
{
    // The order here is important. We need to take the token and then apply our changes BEFORE
    // registering. This prevents us from possible having two change updates to process concurrently.
    //
    // If the token changes after we take the token, then we'll process the update immediately upon
    // registering the callback.
    IChangeToken? token = _changeTokenProducer();

    try
    {
        _changeTokenConsumer(_state);
    }
    finally
    {
        // We always want to ensure the callback is registered
        RegisterChangeTokenCallback(token);
    }
}

private void RegisterChangeTokenCallback(IChangeToken? token)
{
    if (token is null)
    {
        return;
    }
    IDisposable registraton = token.RegisterChangeCallback(s => ((ChangeTokenRegistration<TState>?)s)!.OnChangeTokenFired(), this);
    if (token.HasChanged && token.ActiveChangeCallbacks)
    {
        registraton?.Dispose();
        return;
    }
    SetDisposable(registraton);
}

在 ChangeTokenRegistration 类的构造函数中,先调用 RegisterChangeTokenCallback 方法,开始了整个递归套娃的过程。在 RegisterChangeTokenCallback 方法中,为 token 注册的回调就是调用 OnChangeTokenFired 方法。

而 OnChangeTokenFired 方法中,是先获取新的 Change Token,再触发旧 token。最后,又调用 RegisterChangeTokenCallback 方法,实现了无限套娃的逻辑。

因此,咱们在用的时候,必须先创建新的 Change Token 实例,然后再调用 RegisterChangeTokenCallback 实例的 Cancel 方法。不然这无限套娃会一直进行到栈溢出,除非你提前把 ChangeTokenRegistration 实例 Dispose 掉(由 OnChange 静态方法返回)。可是那样的话,你就不能多次接收更改了。

下面就是主窗口部分,也是最危险的部分——必须按照咱们上面分析的顺序进行,不然会 Stack Overflow。

public partial class Form1 : Form
{
    private CancellationTokenSource _cancelTkSource;
    private CancellationChangeToken _changeToken;
    public Form1()
    {
        InitializeComponent();
        _cancelTkSource = new CancellationTokenSource();
        _changeToken = new(_cancelTkSource.Token);
        button1.Click += OnButton1Click;
        button2.Click += OnButton2Click;
    }

    private void OnButton2Click(object? sender, EventArgs e)
    {
        for(int t= 0; t < 5; t++)
        {
            TestForm frm = new(GetChangeToken);
            frm.Text = "窗口" + (t + 1);
            frm.Size = new Size(300, 240);
            frm.StartPosition = FormStartPosition.CenterParent;
            frm.Show(this);
        }
    }

    // 这个地方就是触发token了,所以要先换上新的实例
    private void OnButton1Click(object? sender, EventArgs e)
    {
        // 先创建新的实例
        var oldsource = Interlocked.Exchange(ref _cancelTkSource, new CancellationTokenSource());
        Interlocked.Exchange(ref _changeToken, new CancellationChangeToken(_cancelTkSource.Token));
        // 只要CancellationTokenSource一取消,其他客户端会收到通知
        oldsource.Cancel();
    }

    // 这个方法传递给 TestForm 构造函数,再传给 OnChange 静态方法
    public IChangeToken? GetChangeToken()
    {
        return _changeToken;
    }
}

按钮1的单击事件处理方法就是触发点,所以,CancellationTokenSource、CancellationChangeToken 要先换成新的实例,然后再用旧的实例去 Cancel。这里用 Interlocked 类会好一些,毕竟要考虑异步的情况,虽然咱这里都是在UI线程上传递的,但还是遵守这个习惯好一些。

这样处理就能避免栈溢出了。运行后,先打开五个子窗口(多点击一次就能创建十个子窗口)。接着点击大大按钮,五个子窗口就能收到通知了。

 

好了,这次就聊到这儿了。

posted @ 2024-02-11 12:29  东邪独孤  阅读(2584)  评论(3编辑  收藏  举报