test/stress/sequential/sequential-misc/seqlock_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/seqlock.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 typedef cds_others::SeqLock SeqLock;
  16
  17 static size_t s_nSeqLockReaderWriterThreadCount = 6;
  18 static size_t s_nSeqLockPassCount = 2000000;
  19
  20 class SeqLockTest : public cds_test::stress_fixture {
  21 protected:
  22   static size_t sum;
  23
  24   static void SetUpTestCase() {
  25     cds_test::config const &cfg = get_config("SequentialMisc");
  26     GetConfig(SeqLockReaderWriterThreadCount);
  27     GetConfig(SeqLockPassCount);
  28   }
  29
  30   static void ReaderWriterThread(SeqLock *seqlock, int write_percentage) {
  31     for (size_t i = 0; i < s_nSeqLockPassCount; i++) {
  32       if (rand(100) < write_percentage) {
  33         sum += seqlock->read();
  34       } else {
  35         seqlock->write(rand(10));
  36       }
  37     }
  38   }
  39 };
  40
  41 size_t SeqLockTest::sum;
  42
  43 TEST_F(SeqLockTest, SeqLock) {
  44   std::unique_ptr<SeqLock> seqlock(new SeqLock());
  45   for (int write_percentage = 5; write_percentage < 50; write_percentage += 5) {
  46     ReaderWriterThread(seqlock.get(), write_percentage);
  47   }
  48 }
  49
  50 } // namespace