--- /dev/null
+#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