Adds misc benchmarks

[libcds.git] / test / stress / misc / rwlock_driver.cpp

diff --git a/test/stress/misc/rwlock_driver.cpp b/test/stress/misc/rwlock_driver.cpp
new file mode 100644 (file)
index 0000000..d2dfce9
--- /dev/null
+++ b/test/stress/misc/rwlock_driver.cpp
@@ -0,0 +1,121 @@
+#include <atomic>
+#include <cds/gc/dhp.h>
+#include <cds/gc/hp.h>
+#include <cds/sync/rwlock.h>
+#include <cds_test/stress_test.h>
+#include <iostream>
+#include <thread>
+
+using namespace std;
+
+namespace {
+
+typedef cds_others::RWLock RWLock;
+class RWLockTest : public cds_test::stress_fixture {
+protected:
+  static int sum;
+  static int x;
+  static RWLock *rwlock;
+  static const int kReaderThreads = 0;
+  static const int kWriterThreads = 0;
+  static const int kReaderWriterThreads = 6;
+  static const int kWriterPercentage = 20;
+  static const int kRWPassCount = 20000;
+
+  static void SetUpTestCase() {}
+
+  static void ReaderThread() {
+    for (int i = 0; i < 10000; i++) {
+      for (int j = 0; j < 10; i++) {
+        if (rwlock->read_can_lock()) {
+          if (!rwlock->read_trylock()) {
+            rwlock->read_lock();
+          }
+          sum += x;
+          rwlock->read_unlock();
+        } else {
+          rwlock->read_lock();
+          sum += x;
+          rwlock->read_unlock();
+        }
+      }
+    }
+  }
+
+  static void WriterThread() {
+    for (int i = 0; i < 10000; i++) {
+      if (rwlock->write_can_lock()) {
+        if (!rwlock->write_trylock()) {
+          rwlock->write_lock();
+        }
+        x += 1;
+        rwlock->write_unlock();
+      } else {
+        rwlock->write_lock();
+        x += 1;
+        rwlock->write_unlock();
+      }
+    }
+  }
+
+  static void ReaderWriterThread() {
+    for (int i = 0; i < kRWPassCount; i++) {
+      for (int j = 0; j < kRWPassCount; j++) {
+        if (rand(100) < kWriterPercentage) {
+          if (rwlock->read_can_lock()) {
+            if (!rwlock->read_trylock()) {
+              rwlock->read_lock();
+            }
+            sum += x;
+            rwlock->read_unlock();
+          } else {
+            rwlock->read_lock();
+            sum += x;
+            rwlock->read_unlock();
+          }
+        } else {
+          if (rwlock->write_can_lock()) {
+            if (!rwlock->write_trylock()) {
+              rwlock->write_lock();
+            }
+            x += 1;
+            rwlock->write_unlock();
+          } else {
+            rwlock->write_lock();
+            x += 1;
+            rwlock->write_unlock();
+          }
+        }
+      }
+    }
+  }
+};
+
+int RWLockTest::x;
+int RWLockTest::sum;
+RWLock *RWLockTest::rwlock;
+const int RWLockTest::kReaderThreads;
+const int RWLockTest::kWriterThreads;
+const int RWLockTest::kReaderWriterThreads;
+const int RWLockTest::kRWPassCount;
+
+TEST_F(RWLockTest, BasicLockUnlock) {
+  rwlock = new RWLock();
+  int num_threads = kReaderThreads + kWriterThreads + kReaderWriterThreads;
+  std::thread *threads = new std::thread[num_threads];
+  for (int i = 0; i < kReaderThreads; i++) {
+    threads[i] = std::thread(ReaderThread);
+  }
+  for (int i = kReaderThreads; i < (kReaderThreads + kWriterThreads); i++) {
+    threads[i] = std::thread(WriterThread);
+  }
+  for (int i = (kReaderThreads + kWriterThreads); i < num_threads; i++) {
+    threads[i] = std::thread(ReaderWriterThread);
+  }
+
+  for (int i = 0; i < num_threads; i++) {
+    threads[i].join();
+  }
+}
+
+} // namespace