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     x = 1;
  31     my_mutex->unlock(&g);
  32     for (ullong i = 0; i < s_nMCSLockPassCount; i++) {
  33       my_mutex->lock(&g);
  34       x = i;
  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   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
  51   for (int i = 0; i < s_nMCSLockThreadCount; i++) {
  52     threads[i].join();
  53   }
  54 }
  55
  56 } // namespace