From 89378627e98d608d45aac82184a6764763c91c8e Mon Sep 17 00:00:00 2001
From: jasonxie <jasonxie@intranet.123u.com>
Date: Tue, 5 Mar 2024 12:19:41 +0800
Subject: [PATCH] Add mset Callback interface

---
 src/sw/redis++/async_redis.h         | 8 ++++++++
 src/sw/redis++/async_redis_cluster.h | 8 ++++++++
 test/src/sw/redis++/async_test.h     | 7 +++++++
 3 files changed, 23 insertions(+)

diff --git a/src/sw/redis++/async_redis.h b/src/sw/redis++/async_redis.h
index 211c0c6..425fdc3 100644
--- 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
index f640d1f..650d986 100644
--- 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
index d40d047..0a2f302 100644
--- 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>