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