当前位置: 欣欣网 > 码农

.NET 聊聊 IChangeToken 接口

2024-03-01码农

前言

不管你的工作是什么,忙或者不忙,报酬高或低,但是,人,总得活,总得过日子。咱们最好多给自己点福利,多整点可以自娱自乐的东西,这就是生活。下棋、打游戏、绘画、书法、钓鱼、飙车、唢呐……不管玩点啥,只要积极正向的就好,可以大大降低得抑郁症、高血压的机率;可以减少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 只能触发一次?为何要重新注册回调?

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

staticvoidMain(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。它有四个字段:

CREATETABLE [dbo].[tb_configdata](
[ID] [intNOTNULL,
[config_key] [nvarchar](15NOTNULL,
[config_value] [nvarchar](30NOTNULL,
[remark] [nvarchar](50NULL,
CONSTRAINT [PK_tb_configdata] PRIMARY KEY CLUSTERED 
(
[IDASC,
[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 = OFFON [PRIMARY]
ON [PRIMARY]
GO

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

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

// 实体类
public classMyConfigData
{
publicint ID { getset; }
publicstring ConfigKey { getset; } = string.Empty;
publicstring ConfigValue { getset; } = string.Empty;
publicstring? Remark { getset; }
}
// 数据库上下文对象
public classDemoConfigDBContext : DbContext
{
public DbSet<MyConfigData> ConfigData => Set<MyConfigData>();
protectedoverridevoidOnConfiguring(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");
}
protectedoverridevoidOnModelCreating(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 classMyConfigurationProvider : ConfigurationProviderIDisposable
{
private System.Threading.Timer theTimer;
publicMyConfigurationProvider()
{
theTimer = new Timer(OnTimer, null10010000);
}
privatevoidOnTimer(object? state)
{
// 先调用Load方法,然后用OnReload触发更新通知
Load();
OnReload();
}
publicvoidDispose()
{
theTimer?.Change(00);
theTimer?.Dispose();
}
publicoverridevoidLoad()
{
// 先读取一下
using DemoConfigDBContext dbctx = new();
// 如果无数据,先初始化
if(dbctx.ConfigData.Count() == 0)
{
InitData(dbctx.ConfigData);
}
// 加载数据
Data = dbctx.ConfigData.ToDictionary(k => k.ConfigKey, k => (string?)k.ConfigValue);
// 本地函数
voidInitData(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 classMyConfigurationSource : IConfigurationSource
{
public IConfigurationProvider Build(IConfigurationBuilder builder)
{
returnnew 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 classMyConfigurationSource : IConfigurationSource
{
public IConfigurationProvider Build(IConfigurationBuilder builder)
{
// 此处可以临时build一个配置树,就能获取到JSON配置文件里面的连接字符串了
var config = builder.Build();
string connStr = config["ConnectionStrings:test"]!;
returnnew MyConfigurationProvider(connStr);
}
}

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

先是 MyConfigurationProvider 的构造函数。

public classMyConfigurationProvider : ConfigurationProviderIDisposable
{
private System.Threading.Timer theTimer;
privatestring connectString;
publicMyConfigurationProvider(string cnnstr)
{
connectString = cnnstr;
……
}
……
}

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

public classDemoConfigDBContext : DbContext
{
privatestring connStr;
publicDemoConfigDBContext(string connectionString)
{
connStr = connectionString;
}
……
protectedoverridevoidOnConfiguring(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 classTestForm : Form
{
private IDisposable _changeTokenReg;
private TextBox _txtMsg;
publicTestForm(Func<IChangeToken?> getToken)
{
// 初始化子级控件
_txtMsg = new()
{
Dock = Dock style.Fill,
Margin = new Padding(5),
Multiline = true,
ScrollBars = ScrollBars.Vertical
};
Controls.Add(_txtMsg);
_changeTokenReg = ChangeToken.OnChange(getToken, OnCallback);
}
// 回调方法
voidOnCallback()
{
DateTime curtime = DateTime.Now;
string str = $"{curtime.ToLongTimeString()} 新年快乐\r\n";
_txtMsg.BeginInvoke(() =>
{
_txtMsg.AppendText(str);
});
}
protectedoverridevoidDispose(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 的源代码就明白了。

publicstatic IDisposable OnChange<TState>(Func<IChangeToken?> changeTokenProducer, Action<TState> changeTokenConsumer, TState state)
{
if (changeTokenProducer isnull)
{
ThrowHelper.ThrowArgumentNullException(ExceptionArgument.changeTokenProducer);
}
if (changeTokenConsumer isnull)
{
ThrowHelper.ThrowArgumentNullException(ExceptionArgument.changeTokenConsumer);
}
returnnew ChangeTokenRegistration<TState>(changeTokenProducer, changeTokenConsumer, state);
}

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

privatevoidOnChangeTokenFired()
{
// 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);
}
}
privatevoidRegisterChangeTokenCallback(IChangeToken? token)
{
if (token isnull)
{
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。

publicpartial classForm1 : Form
{
private CancellationTokenSource _cancelTkSource;
private CancellationChangeToken _changeToken;
publicForm1()
{
InitializeComponent();
_cancelTkSource = new CancellationTokenSource();
_changeToken = new(_cancelTkSource.Token);
button1.Click += OnButton1Click;
button2.Click += OnButton2Click;
}
privatevoidOnButton2Click(object? sender, EventArgs e)
{
for(int t= 0; t < 5; t++)
{
TestForm frm = new(GetChangeToken);
frm.Text = "窗口" + (t + 1);
frm.Size = new Size(300240);
frm.StartPosition = FormStartPosition.CenterParent;
frm.Show(this);
}
}
// 这个地方就是触发token了,所以要先换上新的实例
privatevoidOnButton1Click(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线程上传递的,但还是遵守这个习惯好一些。

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

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

转自:东邪独孤

链接:cnblogs.com/tcjiaan/p/18012397

- EOF -

推荐阅读 点击标题可跳转

看完本文有收获?请转发分享给更多人

推荐关注「DotNet」,提升.Net技能

点赞和在看就是最大的支持❤️