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
|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
#ifndef __SPINLOCK_H__
#define __SPINLOCK_H__
// #SwitchToThreadSpinning
//
// If you call __SwitchToThread in a loop waiting for a condition to be met,
// it is critical that you insert periodic sleeps. This is because the thread
// you are waiting for to set that condition may need your CPU, and simply
// calling __SwitchToThread(0) will NOT guarantee that it gets a chance to run.
// If there are other runnable threads of higher priority, or even if there
// aren't and it is in another processor's queue, you will be spinning a very
// long time.
//
// To force all callers to consider this issue and to avoid each having to
// duplicate the same backoff code, __SwitchToThread takes a required second
// parameter. If you want it to handle backoff for you, this parameter should
// be the number of successive calls you have made to __SwitchToThread (a loop
// count). If you want to take care of backing off yourself, you can pass
// CALLER_LIMITS_SPINNING. There are three valid cases for doing this:
//
// - You count iterations and induce a sleep periodically
// - The number of consecutive __SwitchToThreads is limited
// - Your call to __SwitchToThread includes a non-zero sleep duration
//
// Lastly, to simplify this requirement for the following common coding pattern:
//
// while (!condition)
// SwitchToThread
//
// you can use the YIELD_WHILE macro.
#define YIELD_WHILE(condition) \
{ \
UInt32 __dwSwitchCount = 0; \
while (condition) \
{ \
__SwitchToThread(0, ++__dwSwitchCount); \
} \
}
class SpinLock
{
private:
enum LOCK_STATE
{
UNLOCKED = 0,
LOCKED = 1
};
volatile Int32 m_lock;
static void Lock(SpinLock& lock)
{ YIELD_WHILE (PalInterlockedExchange(&lock.m_lock, LOCKED) == LOCKED); }
static void Unlock(SpinLock& lock)
{ PalInterlockedExchange(&lock.m_lock, UNLOCKED); }
public:
SpinLock()
: m_lock(UNLOCKED) { }
typedef HolderNoDefaultValue<SpinLock&,
SpinLock::Lock,
SpinLock::Unlock>
Holder;
};
#endif
|