723
|
1 #include "gtest/gtest.h"
|
|
2
|
|
3 #include "../Core/OrthancException.h"
|
|
4 #include "../Core/Toolbox.h"
|
|
5 #include "../Core/MultiThreading/ArrayFilledByThreads.h"
|
760
|
6 #include "../Core/MultiThreading/Locker.h"
|
|
7 #include "../Core/MultiThreading/Mutex.h"
|
|
8 #include "../Core/MultiThreading/ReaderWriterLock.h"
|
723
|
9 #include "../Core/MultiThreading/ThreadedCommandProcessor.h"
|
|
10
|
|
11 using namespace Orthanc;
|
|
12
|
|
13 namespace
|
|
14 {
|
|
15 class DynamicInteger : public ICommand
|
|
16 {
|
|
17 private:
|
|
18 int value_;
|
|
19 std::set<int>& target_;
|
|
20
|
|
21 public:
|
|
22 DynamicInteger(int value, std::set<int>& target) :
|
|
23 value_(value), target_(target)
|
|
24 {
|
|
25 }
|
|
26
|
|
27 int GetValue() const
|
|
28 {
|
|
29 return value_;
|
|
30 }
|
|
31
|
|
32 virtual bool Execute()
|
|
33 {
|
|
34 static boost::mutex mutex;
|
|
35 boost::mutex::scoped_lock lock(mutex);
|
|
36 target_.insert(value_);
|
|
37 return true;
|
|
38 }
|
|
39 };
|
|
40
|
|
41 class MyFiller : public ArrayFilledByThreads::IFiller
|
|
42 {
|
|
43 private:
|
|
44 int size_;
|
|
45 unsigned int created_;
|
|
46 std::set<int> set_;
|
|
47
|
|
48 public:
|
|
49 MyFiller(int size) : size_(size), created_(0)
|
|
50 {
|
|
51 }
|
|
52
|
|
53 virtual size_t GetFillerSize()
|
|
54 {
|
|
55 return size_;
|
|
56 }
|
|
57
|
|
58 virtual IDynamicObject* GetFillerItem(size_t index)
|
|
59 {
|
|
60 static boost::mutex mutex;
|
|
61 boost::mutex::scoped_lock lock(mutex);
|
|
62 created_++;
|
|
63 return new DynamicInteger(index * 2, set_);
|
|
64 }
|
|
65
|
|
66 unsigned int GetCreatedCount() const
|
|
67 {
|
|
68 return created_;
|
|
69 }
|
|
70
|
|
71 std::set<int> GetSet()
|
|
72 {
|
|
73 return set_;
|
|
74 }
|
|
75 };
|
|
76 }
|
|
77
|
|
78
|
|
79
|
|
80
|
|
81 TEST(MultiThreading, SharedMessageQueueBasic)
|
|
82 {
|
|
83 std::set<int> s;
|
|
84
|
|
85 SharedMessageQueue q;
|
|
86 ASSERT_TRUE(q.WaitEmpty(0));
|
|
87 q.Enqueue(new DynamicInteger(10, s));
|
|
88 ASSERT_FALSE(q.WaitEmpty(1));
|
|
89 q.Enqueue(new DynamicInteger(20, s));
|
|
90 q.Enqueue(new DynamicInteger(30, s));
|
|
91 q.Enqueue(new DynamicInteger(40, s));
|
|
92
|
|
93 std::auto_ptr<DynamicInteger> i;
|
|
94 i.reset(dynamic_cast<DynamicInteger*>(q.Dequeue(1))); ASSERT_EQ(10, i->GetValue());
|
|
95 i.reset(dynamic_cast<DynamicInteger*>(q.Dequeue(1))); ASSERT_EQ(20, i->GetValue());
|
|
96 i.reset(dynamic_cast<DynamicInteger*>(q.Dequeue(1))); ASSERT_EQ(30, i->GetValue());
|
|
97 ASSERT_FALSE(q.WaitEmpty(1));
|
|
98 i.reset(dynamic_cast<DynamicInteger*>(q.Dequeue(1))); ASSERT_EQ(40, i->GetValue());
|
|
99 ASSERT_TRUE(q.WaitEmpty(0));
|
|
100 ASSERT_EQ(NULL, q.Dequeue(1));
|
|
101 }
|
|
102
|
|
103
|
|
104 TEST(MultiThreading, SharedMessageQueueClean)
|
|
105 {
|
|
106 std::set<int> s;
|
|
107
|
|
108 try
|
|
109 {
|
|
110 SharedMessageQueue q;
|
|
111 q.Enqueue(new DynamicInteger(10, s));
|
|
112 q.Enqueue(new DynamicInteger(20, s));
|
|
113 throw OrthancException("Nope");
|
|
114 }
|
|
115 catch (OrthancException&)
|
|
116 {
|
|
117 }
|
|
118 }
|
|
119
|
|
120
|
|
121 TEST(MultiThreading, ArrayFilledByThreadEmpty)
|
|
122 {
|
|
123 MyFiller f(0);
|
|
124 ArrayFilledByThreads a(f);
|
|
125 a.SetThreadCount(1);
|
|
126 ASSERT_EQ(0, a.GetSize());
|
|
127 }
|
|
128
|
|
129
|
|
130 TEST(MultiThreading, ArrayFilledByThread1)
|
|
131 {
|
|
132 MyFiller f(100);
|
|
133 ArrayFilledByThreads a(f);
|
|
134 a.SetThreadCount(1);
|
|
135 ASSERT_EQ(100, a.GetSize());
|
|
136 for (size_t i = 0; i < a.GetSize(); i++)
|
|
137 {
|
|
138 ASSERT_EQ(2 * i, dynamic_cast<DynamicInteger&>(a.GetItem(i)).GetValue());
|
|
139 }
|
|
140 }
|
|
141
|
|
142
|
|
143 TEST(MultiThreading, ArrayFilledByThread4)
|
|
144 {
|
|
145 MyFiller f(100);
|
|
146 ArrayFilledByThreads a(f);
|
|
147 a.SetThreadCount(4);
|
|
148 ASSERT_EQ(100, a.GetSize());
|
|
149 for (size_t i = 0; i < a.GetSize(); i++)
|
|
150 {
|
|
151 ASSERT_EQ(2 * i, dynamic_cast<DynamicInteger&>(a.GetItem(i)).GetValue());
|
|
152 }
|
|
153
|
|
154 ASSERT_EQ(100u, f.GetCreatedCount());
|
|
155
|
|
156 a.Invalidate();
|
|
157
|
|
158 ASSERT_EQ(100, a.GetSize());
|
|
159 ASSERT_EQ(200u, f.GetCreatedCount());
|
|
160 ASSERT_EQ(4u, a.GetThreadCount());
|
|
161 ASSERT_TRUE(f.GetSet().empty());
|
|
162
|
|
163 for (size_t i = 0; i < a.GetSize(); i++)
|
|
164 {
|
|
165 ASSERT_EQ(2 * i, dynamic_cast<DynamicInteger&>(a.GetItem(i)).GetValue());
|
|
166 }
|
|
167 }
|
|
168
|
|
169
|
|
170 TEST(MultiThreading, CommandProcessor)
|
|
171 {
|
|
172 ThreadedCommandProcessor p(4);
|
|
173
|
|
174 std::set<int> s;
|
|
175
|
|
176 for (size_t i = 0; i < 100; i++)
|
|
177 {
|
|
178 p.Post(new DynamicInteger(i * 2, s));
|
|
179 }
|
|
180
|
|
181 p.Join();
|
|
182
|
|
183 for (size_t i = 0; i < 200; i++)
|
|
184 {
|
|
185 if (i % 2)
|
|
186 ASSERT_TRUE(s.find(i) == s.end());
|
|
187 else
|
|
188 ASSERT_TRUE(s.find(i) != s.end());
|
|
189 }
|
|
190 }
|
760
|
191
|
|
192
|
|
193 TEST(MultiThreading, Mutex)
|
|
194 {
|
|
195 Mutex mutex;
|
|
196 Locker locker(mutex);
|
|
197 }
|
|
198
|
|
199
|
|
200 TEST(MultiThreading, ReaderWriterLock)
|
|
201 {
|
|
202 ReaderWriterLock lock;
|
|
203
|
|
204 {
|
|
205 Locker locker1(lock.ForReader());
|
|
206 Locker locker2(lock.ForReader());
|
|
207 }
|
|
208
|
|
209 {
|
|
210 Locker locker3(lock.ForWriter());
|
|
211 }
|
|
212 }
|