forked from SharpGenTools/SharpGenTools
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDisposeBase.Events.cs
More file actions
107 lines (95 loc) · 3.08 KB
/
Copy pathDisposeBase.Events.cs
File metadata and controls
107 lines (95 loc) · 3.08 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
98
99
100
101
102
103
104
105
106
107
using System;
using System.ComponentModel;
using System.Runtime.CompilerServices;
using System.Threading;
namespace SharpGen.Runtime;
using EventTable = ConditionalWeakTable<DisposeBase, DisposeEventHandler>;
/// <summary>
/// Base class for a <see cref="IDisposable"/> class.
/// </summary>
public abstract partial class DisposeBase
{
private static readonly EventTable DisposingEventTable = new();
private static readonly EventTable DisposedEventTable = new();
private static readonly ReaderWriterLockSlim DisposingEventLock = new(LockRecursionPolicy.NoRecursion);
private static readonly ReaderWriterLockSlim DisposedEventLock = new(LockRecursionPolicy.NoRecursion);
[MethodImpl(Utilities.MethodAggressiveOptimization)]
private void UpdateEventHandler(EventTable table, DisposeEventHandler value)
{
#if NETFRAMEWORK || NETSTANDARD1_3 || NETSTANDARD2_0
table.Remove(this);
table.Add(this, value);
#else
table.AddOrUpdate(this, value);
#endif
}
private void AddEventHandler(EventTable table, DisposeEventHandler value, ReaderWriterLockSlim rwLock)
{
rwLock.EnterWriteLock();
try
{
if (table.TryGetValue(this, out var handler))
UpdateEventHandler(table, handler + value);
else
table.Add(this, value);
}
finally
{
rwLock.ExitWriteLock();
}
}
private void RemoveEventHandler(EventTable table, DisposeEventHandler value, ReaderWriterLockSlim rwLock)
{
rwLock.EnterUpgradeableReadLock();
try
{
if (!table.TryGetValue(this, out var handler))
return;
rwLock.EnterWriteLock();
try
{
UpdateEventHandler(table, handler - value);
}
finally
{
rwLock.ExitWriteLock();
}
}
finally
{
rwLock.ExitUpgradeableReadLock();
}
}
/// <summary>
/// Occurs when this instance is starting to be disposed.
/// </summary>
[EditorBrowsable(EditorBrowsableState.Advanced)]
public event DisposeEventHandler Disposing
{
add => AddEventHandler(DisposingEventTable, value, DisposingEventLock);
remove => RemoveEventHandler(DisposingEventTable, value, DisposingEventLock);
}
/// <summary>
/// Occurs when this instance is fully disposed.
/// </summary>
[EditorBrowsable(EditorBrowsableState.Advanced)]
public event DisposeEventHandler Disposed
{
add => AddEventHandler(DisposedEventTable, value, DisposedEventLock);
remove => RemoveEventHandler(DisposedEventTable, value, DisposedEventLock);
}
private void InvokeDisposeEventHandler(bool disposing, ReaderWriterLockSlim rwLock, EventTable table)
{
DisposeEventHandler value;
rwLock.EnterReadLock();
try
{
table.TryGetValue(this, out value);
}
finally
{
rwLock.ExitReadLock();
}
value?.Invoke(this, disposing);
}
}