add multithreads test for RedisCluster

Author: sewenew

test/src/sw/redis++/test_main.cpp

@@ -115,7 +115,7 @@ int main(int argc, char **argv) {
 
         std::cout << "Pass pipeline and transaction tests" << std::endl;
 
-        sw::redis::test::ThreadsTest threads_test(opts);
+        sw::redis::test::ThreadsTest threads_test(opts, cluster_node_opts);
         threads_test.run();
 
         std::cout << "Pass threads tests" << std::endl;

test/src/sw/redis++/threads_test.cpp

@@ -18,6 +18,7 @@
 #include <thread>
 #include <chrono>
 #include <atomic>
+#include <memory>
 #include "utils.h"
 
 namespace sw {
@@ -26,7 +27,9 @@ namespace redis {
 
 namespace test {
 
-ThreadsTest::ThreadsTest(const ConnectionOptions &opts) : _opts(opts) {}
+ThreadsTest::ThreadsTest(const ConnectionOptions &opts,
+                            const ConnectionOptions &cluster_opts) : _opts(opts),
+                                                                    _cluster_opts(cluster_opts) {}
 
 void ThreadsTest::run() {
     // 100 * 10000 = 1 million writes
@@ -36,23 +39,32 @@ void ThreadsTest::run() {
     // Default pool options: single connection and wait forever.
     _test_multithreads(Redis(_opts), thread_num, times);
 
+    _test_multithreads(RedisCluster(_cluster_opts), thread_num, times);
+
     // Pool with 10 connections.
     ConnectionPoolOptions pool_opts;
     pool_opts.size = 10;
     _test_multithreads(Redis(_opts, pool_opts), thread_num, times);
 
+    _test_multithreads(RedisCluster(_cluster_opts, pool_opts), thread_num, times);
+
     _test_timeout();
 }
 
-void ThreadsTest::_test_multithreads(Redis redis, int thread_num, int times) {
+template <typename RedisType>
+void ThreadsTest::_test_multithreads(RedisType redis, int thread_num, int times) {
     std::vector<std::string> keys;
     keys.reserve(thread_num);
     for (auto idx = 0; idx != thread_num; ++idx) {
         auto key = test_key("multi-threads::" + std::to_string(idx));
         keys.push_back(key);
     }
 
-    KeyDeleter deleter(redis, keys.begin(), keys.end());
+    using DeleterUPtr = std::unique_ptr<KeyDeleterTpl<RedisType>>;
+    std::vector<DeleterUPtr> deleters;
+    for (const auto &key : keys) {
+        deleters.emplace_back(new KeyDeleterTpl<RedisType>(redis, key));
+    }
 
     std::vector<std::thread> workers;
     workers.reserve(thread_num);

test/src/sw/redis++/threads_test.h

@@ -27,16 +27,19 @@ namespace test {
 
 class ThreadsTest {
 public:
-    explicit ThreadsTest(const ConnectionOptions &opts);
+    ThreadsTest(const ConnectionOptions &opts, const ConnectionOptions &cluster_opts);
 
     void run();
 
 private:
-    void _test_multithreads(Redis redis, int threads_num, int times);
+    template <typename RedisType>
+    void _test_multithreads(RedisType redis, int threads_num, int times);
 
     void _test_timeout();
 
     ConnectionOptions _opts;
+
+    ConnectionOptions _cluster_opts;
 };
 
 }