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
|
// <copyright>
// Copyright (c) Microsoft Corporation. All rights reserved.
// </copyright>
namespace System.ServiceModel.Channels
{
using System.Runtime;
using System.Threading;
// IMPORTANT: Only meant to be used within a using statement.
struct LockHelper : IDisposable
{
ReaderWriterLockSlim readerWriterLock;
bool isReaderLock;
bool isLockHeld;
LockHelper(ReaderWriterLockSlim readerWriterLock, bool isReaderLock)
{
this.readerWriterLock = readerWriterLock;
this.isReaderLock = isReaderLock;
if (isReaderLock)
{
this.readerWriterLock.EnterReadLock();
}
else
{
this.readerWriterLock.EnterWriteLock();
}
this.isLockHeld = true;
}
public void Dispose()
{
if (this.isLockHeld)
{
this.isLockHeld = false;
if (this.isReaderLock)
{
this.readerWriterLock.ExitReadLock();
}
else
{
this.readerWriterLock.ExitWriteLock();
}
}
}
internal static IDisposable TakeWriterLock(ReaderWriterLockSlim readerWriterLock)
{
Fx.Assert(readerWriterLock != null, "The readerWriterLock cannot be null.");
return new LockHelper(readerWriterLock, false);
}
internal static IDisposable TakeReaderLock(ReaderWriterLockSlim readerWriterLock)
{
Fx.Assert(readerWriterLock != null, "The readerWriterLock cannot be null.");
return new LockHelper(readerWriterLock, true);
}
}
}
|
