| /** |
| * Copyright 2011-2015 Quickstep Technologies LLC. |
| * Copyright 2015 Pivotal Software, Inc. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| **/ |
| |
| #include <atomic> |
| #include <chrono> |
| #include <thread> // NOLINT(build/c++11) |
| |
| #include "gtest/gtest.h" |
| |
| #include "threading/Mutex.hpp" |
| #include "threading/SharedMutex.hpp" |
| #include "threading/SpinSharedMutex.hpp" |
| #include "threading/Thread.hpp" |
| #include "utility/Macros.hpp" |
| |
| namespace quickstep { |
| |
| template <typename SharedMutexT> |
| struct SharedDataItem { |
| typedef MutexLockImpl<SharedMutexT, true> ExclusiveLockT; |
| typedef SharedMutexSharedLockImpl<SharedMutexT, true> SharedLockT; |
| |
| int payload; |
| SharedMutexT mutex; |
| std::atomic<int> active_readers; |
| }; |
| |
| template <typename ItemT> |
| class ModifierThread : public Thread { |
| public: |
| ModifierThread(ItemT *item, const int set_value) |
| : item_(item), |
| set_value_(set_value) { |
| } |
| |
| protected: |
| void run() override { |
| typename ItemT::ExclusiveLockT lock(item_->mutex); |
| std::this_thread::sleep_for(std::chrono::seconds(1)); |
| item_->payload = set_value_; |
| } |
| |
| private: |
| ItemT *item_; |
| const int set_value_; |
| |
| DISALLOW_COPY_AND_ASSIGN(ModifierThread); |
| }; |
| |
| template <typename ItemT> |
| class AccessorThread : public Thread { |
| public: |
| explicit AccessorThread(ItemT *item) |
| : item_(item) { |
| } |
| |
| int getReadValue() const { |
| return read_value_; |
| } |
| |
| bool sawOtherThreadAfterLocking() const { |
| return other_threads_seen_after_lock_ > 0; |
| } |
| |
| protected: |
| void run() override { |
| std::this_thread::sleep_for(std::chrono::milliseconds(100)); |
| |
| typename ItemT::SharedLockT lock(item_->mutex); |
| other_threads_seen_after_lock_ = item_->active_readers.fetch_add(1); |
| |
| read_value_ = item_->payload; |
| |
| std::this_thread::sleep_for(std::chrono::milliseconds(100)); |
| item_->active_readers.fetch_sub(1); |
| } |
| |
| private: |
| ItemT *item_; |
| int read_value_; |
| int other_threads_seen_after_lock_; |
| |
| DISALLOW_COPY_AND_ASSIGN(AccessorThread); |
| }; |
| |
| TEST(MutexTest, SharedMutexBasicTest) { |
| SharedDataItem<SharedMutex> item; |
| item.payload = 0; |
| item.active_readers.store(0); |
| |
| ModifierThread<SharedDataItem<SharedMutex>> modifier(&item, 42); |
| AccessorThread<SharedDataItem<SharedMutex>> accessor_a(&item); |
| AccessorThread<SharedDataItem<SharedMutex>> accessor_b(&item); |
| |
| modifier.start(); |
| accessor_a.start(); |
| accessor_b.start(); |
| |
| modifier.join(); |
| accessor_a.join(); |
| accessor_b.join(); |
| |
| EXPECT_EQ(42, accessor_a.getReadValue()); |
| EXPECT_EQ(42, accessor_b.getReadValue()); |
| EXPECT_TRUE(accessor_a.sawOtherThreadAfterLocking() |
| || accessor_b.sawOtherThreadAfterLocking()); |
| } |
| |
| TEST(MutexTest, SpinSharedMutexBasicTest) { |
| SharedDataItem<SpinSharedMutex<false>> item; |
| item.payload = 0; |
| item.active_readers.store(0); |
| |
| ModifierThread<SharedDataItem<SpinSharedMutex<false>>> modifier(&item, 42); |
| AccessorThread<SharedDataItem<SpinSharedMutex<false>>> accessor_a(&item); |
| AccessorThread<SharedDataItem<SpinSharedMutex<false>>> accessor_b(&item); |
| |
| modifier.start(); |
| accessor_a.start(); |
| accessor_b.start(); |
| |
| modifier.join(); |
| accessor_a.join(); |
| accessor_b.join(); |
| |
| EXPECT_EQ(42, accessor_a.getReadValue()); |
| EXPECT_EQ(42, accessor_b.getReadValue()); |
| EXPECT_TRUE(accessor_a.sawOtherThreadAfterLocking() |
| || accessor_b.sawOtherThreadAfterLocking()); |
| } |
| |
| TEST(MutexTest, SpinSharedMutexWithYieldBasicTest) { |
| SharedDataItem<SpinSharedMutex<true>> item; |
| item.payload = 0; |
| item.active_readers.store(0); |
| |
| ModifierThread<SharedDataItem<SpinSharedMutex<true>>> modifier(&item, 42); |
| AccessorThread<SharedDataItem<SpinSharedMutex<true>>> accessor_a(&item); |
| AccessorThread<SharedDataItem<SpinSharedMutex<true>>> accessor_b(&item); |
| |
| modifier.start(); |
| accessor_a.start(); |
| accessor_b.start(); |
| |
| modifier.join(); |
| accessor_a.join(); |
| accessor_b.join(); |
| |
| EXPECT_EQ(42, accessor_a.getReadValue()); |
| EXPECT_EQ(42, accessor_b.getReadValue()); |
| EXPECT_TRUE(accessor_a.sawOtherThreadAfterLocking() |
| || accessor_b.sawOtherThreadAfterLocking()); |
| } |
| |
| } // namespace quickstep |