Blame view

netcore/src/Infrastructure/NCC/IPCChannel/CallContext.cs 4 KB
96009bc9   hexiaodong   hxd
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实例的拷贝副本,但这是浅复制行为而非深复制
      ///   在设置新的值之前完成复制操作
      /// 获取当前线程 IdThread.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实例的拷贝副本,但这是浅复制行为而非深复制
      ///   在设置新的值之前完成复制操作
      /// 获取当前线程 IdThread.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;
          }
      }
  }