Add mset Callback interface (#551)

Co-authored-by: jasonxie <jasonxie@intranet.123u.com>
sewenew · Mar 9, 2024 · eb3b2a4 · eb3b2a4
1 parent f779c71
commit eb3b2a4
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 0 deletions.
diff --git a/src/sw/redis++/async_redis.h b/src/sw/redis++/async_redis.h
@@ -407,6 +407,14 @@ class AsyncRedis {
     Future<void> mset(std::initializer_list<T> il) {
         return mset(il.begin(), il.end());
     }
+
+    template <typename Input, typename Callback>
+    auto mset(Input first, Input last, Callback &&cb)
+        -> typename std::enable_if<IsInvocable<typename std::decay<Callback>::type, Future<void> &&>::value, void>::type {
+        range_check("MSET", first, last);
+
+        _callback_fmt_command<void>(std::forward<Callback>(cb), fmt::mset<Input>, first, last);
+    }
 
     template <typename Input>
     Future<bool> msetnx(Input first, Input last) {

diff --git a/src/sw/redis++/async_redis_cluster.h b/src/sw/redis++/async_redis_cluster.h
@@ -282,6 +282,14 @@ class AsyncRedisCluster {
         return mset(il.begin(), il.end());
     }
 
+    template <typename Input, typename Callback>
+    auto mset(Input first, Input last, Callback &&cb)
+        -> typename std::enable_if<IsInvocable<typename std::decay<Callback>::type, Future<void> &&>::value, void>::type {
+        range_check("MSET", first, last);
+
+        _callback_fmt_command<void>(std::forward<Callback>(cb), fmt::mset<Input>, first, last);
+    }
+
     template <typename Input>
     Future<bool> msetnx(Input first, Input last) {
         range_check("MSETNX", first, last);

diff --git a/test/src/sw/redis++/async_test.h b/test/src/sw/redis++/async_test.h
@@ -199,6 +199,13 @@ void AsyncTest<RedisInstance>::_test_hash() {
                 this->set_ready();
             });
     _wait();
+
+    set_ready(false);
+    std::map<std::string, std::string> mkeys = {{test_key("str")+"1", "val1"}, {test_key("str")+"2", "val2"}};
+    _redis.mset(mkeys.begin(), mkeys.end(), [this](Future<void> &&fut) {
+                this->set_ready();
+            });
+    _wait();
 }
 
 template <typename RedisInstance>