CoreLib/SrwLock.cpp at main · freezerdev/CoreLib · GitHub

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
#include "Base.h"
#include "SrwLock.h"

NS_BEGIN

/////////////////////////////////////////////////
// CSrwLock
/////////////////////////////////////////////////

//#################################################################################################
CSrwLock::CSrwLock(void)
	: m_srw(SRWLOCK_INIT)
{
}

//#################################################################################################
CSrwLocker::CSrwLocker(CSrwLock &srw, const ELockType eType)
	: m_srw(srw),
	m_eType(ELT_NoLock)
{
	if(eType == ELT_ReaderLock)
		LockReader();
	else if(eType == ELT_WriterLock)
		LockWriter();
}

//#################################################################################################
CSrwLocker::~CSrwLocker(void)
{
	if(m_eType == ELT_ReaderLock)
		UnlockReader();
	else if(m_eType == ELT_WriterLock)
		UnlockWriter();
}

//#################################################################################################
CSrwLocker::ELockType CSrwLocker::GetLockStatus(void) const noexcept
{
	return m_eType;
}

//#################################################################################################
void CSrwLocker::LockReader(void)
{
	Assert(m_eType == ELT_NoLock);

	m_srw.LockReader();
	m_eType = ELT_ReaderLock;
}

//#################################################################################################
void CSrwLocker::LockWriter(void)
{
	Assert(m_eType == ELT_NoLock);

	m_srw.LockWriter();
	m_eType = ELT_WriterLock;
}

//#################################################################################################
bool CSrwLocker::TryLockReader(void)
{
	Assert(m_eType == ELT_NoLock);

	if(m_srw.TryLockReader())
	{
		m_eType = ELT_ReaderLock;
		return true;
	}
	else
		return false;
}

//#################################################################################################
bool CSrwLocker::TryLockWriter(void)
{
	Assert(m_eType == ELT_NoLock);

	if(m_srw.TryLockWriter())
	{
		m_eType = ELT_WriterLock;
		return true;
	}
	else
		return false;
}

//#################################################################################################
void CSrwLocker::UnlockReader(void)
{
	Assert(m_eType == ELT_ReaderLock);

	m_srw.UnlockReader();
	m_eType = ELT_NoLock;
}

//#################################################################################################
void CSrwLocker::UnlockWriter(void)
{
	Assert(m_eType == ELT_WriterLock);

	m_srw.UnlockWriter();
	m_eType = ELT_NoLock;
}

NS_END