test/stress/misc/mcslock_driver.cpp

   1 #include "common.h"
   2 #include <atomic>
   3 #include <cds/gc/dhp.h>
   4 #include <cds/gc/hp.h>
   5 #include <cds/misc/mcs-lock.h>
   6 #include <cds_test/stress_test.h>
   7 #include <iostream>
   8 #include <memory>
   9 #include <thread>
  10
  11 using namespace std;
  12
  13 namespace {
  14
  15 static size_t s_nMCSLockThreadCount = 4;
  16 static size_t s_nMCSLockPassCount = 3000000;
  17
  18 class MCSLockTest : public cds_test::stress_fixture {
  19 protected:
  20   static ullong x;
  21   static cds_others::mcs_mutex *my_mutex;
  22
  23   static void SetUpTestCase() {
  24     cds_test::config const &cfg = get_config("Misc");
  25     GetConfig(MCSLockPassCount);
  26     GetConfig(MCSLockThreadCount);
  27   }
  28
  29   static void Thread() {
  30     cds_others::mcs_mutex::guard g(my_mutex);
  31     my_mutex->unlock(&g);
  32     for (ullong i = 0; i < s_nMCSLockPassCount; i++) {
  33       my_mutex->lock(&g);
  34       x++;
  35       my_mutex->unlock(&g);
  36     }
  37     my_mutex->lock(&g);
  38   }
  39 };
  40
  41 ullong MCSLockTest::x;
  42 cds_others::mcs_mutex *MCSLockTest::my_mutex;
  43
  44 TEST_F(MCSLockTest, BasicLockUnlock) {
  45   my_mutex = new cds_others::mcs_mutex();
  46   x = 0;
  47   std::unique_ptr<std::thread[]> threads(
  48       new std::thread[s_nMCSLockThreadCount]);
  49   for (size_t i = 0; i < s_nMCSLockThreadCount; i++) {
  50     threads[i] = std::thread(Thread);
  51   }
  52   for (size_t i = 0; i < s_nMCSLockThreadCount; i++) {
  53     threads[i].join();
  54   }
  55   if (x != s_nMCSLockPassCount * s_nMCSLockThreadCount) {
  56     cout << "MCS lock incorrect\n";
  57   }
  58 }
  59
  60 } // namespace