From ddf85e9dca048abb614f1197d317855af47663f0 Mon Sep 17 00:00:00 2001
From: wuyun <BlackCloud0526@163.com>
Date: Wed, 19 Mar 2025 08:04:06 +0000
Subject: [PATCH 1/2] feat:rtc can intercept more command

---
 src/pika_client_conn.cc | 43 ++++++++++++++++++++++++++++++++++-------
 1 file changed, 36 insertions(+), 7 deletions(-)

diff --git a/src/pika_client_conn.cc b/src/pika_client_conn.cc
index 768cb6d5ad..1bf77fef5f 100644
--- a/src/pika_client_conn.cc
+++ b/src/pika_client_conn.cc
@@ -258,12 +258,41 @@ void PikaClientConn::ProcessMonitor(const PikaCmdArgsType& argv) {
 }
 
 bool PikaClientConn::IsInterceptedByRTC(std::string& opt) {
-  // currently we only Intercept: Get, HGet
-  if (opt == kCmdNameGet && g_pika_conf->GetCacheString()) {
-    return true;
+  // only single key read command will be intercepted
+  std::shared_ptr<Cmd> c_ptr = g_pika_cmd_table_manager->GetCmd(opt);
+  if (!c_ptr) {
+    return false;
   }
-  if (opt == kCmdNameHGet && g_pika_conf->GetCacheHash()) {
-    return true;
+
+  if (c_ptr->IsNeedReadCache()) {
+    if (c_ptr->flag_ & kCmdFlagsOperateKey) {
+      return true;
+    }
+    if (c_ptr->flag_ & kCmdFlagsKv) {
+      if (g_pika_conf->GetCacheString()) {
+        return true;
+      }
+    } else if (c_ptr->flag_ & kCmdFlagsZset) {
+      if (g_pika_conf->GetCacheZset()) {
+        return true;
+      }
+    } else if (c_ptr->flag_ & kCmdFlagsHash) {
+      if (g_pika_conf->GetCacheHash()) {
+        return true;
+      }
+    } else if (c_ptr->flag_ & kCmdFlagsList) {
+      if (g_pika_conf->GetCacheList()) {
+        return true;
+      }
+    } else if (c_ptr->flag_ & kCmdFlagsBit) {
+      if (g_pika_conf->GetCacheBit()) {
+        return true;
+      }
+    }else if (c_ptr->flag_ & kCmdFlagsSet) {
+      if (g_pika_conf->GetCacheSet()) {
+        return true;
+      }
+    }
   }
   return false;
 }
@@ -349,7 +378,7 @@ bool PikaClientConn::ReadCmdInCache(const net::RedisCmdArgsType& argv, const std
   }
   // Initial
   c_ptr->Initial(argv, current_db_);
-  // dont store cmd with too large key(only Get/HGet cmd can reach here)
+  // dont store cmd with too large key
   // the cmd with large key should be non-exist in cache, except for pre-stored
   if (c_ptr->IsTooLargeKey(g_pika_conf->max_key_size_in_cache())) {
     resp_num--;
@@ -365,7 +394,7 @@ bool PikaClientConn::ReadCmdInCache(const net::RedisCmdArgsType& argv, const std
     // acl check failed
     return false;
   }
-  // only read command(Get, HGet) will reach here, no need of record lock
+  // only read command will reach here, no need of record lock
   bool read_status = c_ptr->DoReadCommandInCache();
   auto cmdstat_map = g_pika_cmd_table_manager->GetCommandStatMap();
   resp_num--;

From 9242005b3533a4005941363a2707f6d7e637fd6b Mon Sep 17 00:00:00 2001
From: wuyun <BlackCloud0526@163.com>
Date: Thu, 3 Apr 2025 05:24:27 +0000
Subject: [PATCH 2/2] delete error comment

---
 src/pika_client_conn.cc | 1 -
 1 file changed, 1 deletion(-)

diff --git a/src/pika_client_conn.cc b/src/pika_client_conn.cc
index 1bf77fef5f..8eb3070533 100644
--- a/src/pika_client_conn.cc
+++ b/src/pika_client_conn.cc
@@ -258,7 +258,6 @@ void PikaClientConn::ProcessMonitor(const PikaCmdArgsType& argv) {
 }
 
 bool PikaClientConn::IsInterceptedByRTC(std::string& opt) {
-  // only single key read command will be intercepted
   std::shared_ptr<Cmd> c_ptr = g_pika_cmd_table_manager->GetCmd(opt);
   if (!c_ptr) {
     return false;