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
|
#include <library/cpp/testing/gtest/gtest.h>
#include <library/cpp/yt/threading/spin_lock.h>
#include <library/cpp/yt/threading/recursive_spin_lock.h>
#include <library/cpp/yt/threading/rw_spin_lock.h>
#include <library/cpp/yt/threading/fork_aware_spin_lock.h>
namespace NYT::NThreading {
namespace {
////////////////////////////////////////////////////////////////////////////////
#ifndef NDEBUG
TEST(TSpinLockCountTest, SpinLock)
{
EXPECT_EQ(GetActiveSpinLockCount(), 0);
YT_DECLARE_SPIN_LOCK(TSpinLock, lock);
auto guard = Guard(lock);
EXPECT_EQ(GetActiveSpinLockCount(), 1);
guard.Release();
EXPECT_EQ(GetActiveSpinLockCount(), 0);
}
TEST(TSpinLockCountTest, RecursiveSpinLock)
{
EXPECT_EQ(GetActiveSpinLockCount(), 0);
YT_DECLARE_SPIN_LOCK(TRecursiveSpinLock, lock);
auto guard1 = Guard(lock);
EXPECT_EQ(GetActiveSpinLockCount(), 1);
auto guard2 = Guard(lock);
EXPECT_EQ(GetActiveSpinLockCount(), 2);
guard2.Release();
EXPECT_EQ(GetActiveSpinLockCount(), 1);
guard1.Release();
EXPECT_EQ(GetActiveSpinLockCount(), 0);
}
TEST(TSpinLockCountTest, ForkAwareSpinLock)
{
EXPECT_EQ(GetActiveSpinLockCount(), 0);
YT_DECLARE_SPIN_LOCK(TForkAwareSpinLock, lock);
auto guard = Guard(lock);
EXPECT_EQ(GetActiveSpinLockCount(), 1);
guard.Release();
EXPECT_EQ(GetActiveSpinLockCount(), 0);
}
TEST(TSpinLockCountTest, RWSpinLock)
{
EXPECT_EQ(GetActiveSpinLockCount(), 0);
YT_DECLARE_SPIN_LOCK(TReaderWriterSpinLock, lock);
auto readerGuard = ReaderGuard(lock);
EXPECT_EQ(GetActiveSpinLockCount(), 1);
readerGuard.Release();
EXPECT_EQ(GetActiveSpinLockCount(), 0);
auto writerGuard = WriterGuard(lock);
EXPECT_EQ(GetActiveSpinLockCount(), 1);
writerGuard.Release();
EXPECT_EQ(GetActiveSpinLockCount(), 0);
}
#endif
////////////////////////////////////////////////////////////////////////////////
} // namespace
} // namespace NYT::NThreading
|
