CallContext.cs
4 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
using NCC.Dependency;
using System.Collections.Concurrent;
using System.Threading;
namespace NCC.IPCChannel
{
/// <summary>
/// 提供线程异步流共享数据上下文（尽量在项目需要该操作的类中使用 AsyncLocal 方式使用，而不是调用 CallContext
/// </summary>
/// <typeparam name="T"></typeparam>
/// <remarks>
/// AsyncLocal 遇到 await 关键字时采用拷贝方式创建新的执行上下文并流转
/// 在Task方法内部修改其值，但在任务结束后仍为初始值，这是一种“写时复制”行为，AsyncLocal内部做了两步操作：
/// 进行AsyncLocal实例的拷贝副本，但这是浅复制行为而非深复制
/// 在设置新的值之前完成复制操作
/// 获取当前线程 Id：Thread.CurrentThread.ManagedThreadId
/// </remarks>
[SuppressSniffer]
public static class CallContext<T>
{
/// <summary>
/// 保存本地数据
/// </summary>
/// <remarks>这里存在内存溢出问题，因为该定义对象并没有任何释放内存的方式提供，所以尽可能的少使用</remarks>
private static readonly ConcurrentDictionary<string, AsyncLocal<T>> localValues = new();
/// <summary>
/// 设置值
/// </summary>
/// <param name="key"></param>
/// <param name="value"></param>
public static void SetLocalValue(string key, T value)
{
localValues.GetOrAdd(key, _ => new AsyncLocal<T>(args =>
{
// args.CurrentValue; // 当前值
// args.PreviousValue; // 更改之前值
// args.ThreadContextChanged; // 如果上下文发生改变返回 true，否则返回 false
})).Value = value;
}
/// <summary>
/// 读取值
/// </summary>
/// <param name="key"></param>
/// <returns></returns>
public static T GetLocalValue(string key)
{
return localValues.TryGetValue(key, out var value) ? value.Value : default;
}
}
/// <summary>
/// 提供线程异步流共享数据上下文（尽量在项目需要该操作的类中使用 AsyncLocal 方式使用，而不是调用 CallContext
/// </summary>
/// <remarks>
/// AsyncLocal 遇到 await 关键字时采用拷贝方式创建新的执行上下文并流转
/// 在Task方法内部修改其值，但在任务结束后仍为初始值，这是一种“写时复制”行为，AsyncLocal内部做了两步操作：
/// 进行AsyncLocal实例的拷贝副本，但这是浅复制行为而非深复制
/// 在设置新的值之前完成复制操作
/// 获取当前线程 Id：Thread.CurrentThread.ManagedThreadId
/// </remarks>
[SuppressSniffer]
public static class CallContext
{
/// <summary>
/// 保存本地数据
/// </summary>
/// <remarks>这里存在内存溢出问题，因为该定义对象并没有任何释放内存的方式提供，所以尽可能的少使用</remarks>
private static readonly ConcurrentDictionary<string, AsyncLocal<object>> localValues = new();
/// <summary>
/// 设置值
/// </summary>
/// <param name="key"></param>
/// <param name="value"></param>
public static void SetLocalValue(string key, object value)
{
localValues.GetOrAdd(key, _ => new AsyncLocal<object>(args =>
{
// args.CurrentValue; // 当前值
// args.PreviousValue; // 更改之前值
// args.ThreadContextChanged; // 如果上下文发生改变返回 true，否则返回 false
})).Value = value;
}
/// <summary>
/// 读取值
/// </summary>
/// <param name="key"></param>
/// <returns></returns>
public static object GetLocalValue(string key)
{
return localValues.TryGetValue(key, out var value) ? value.Value : default;
}
}
}