feat: add Redis Vector Set support

src/cmd/src/admin.rs

@@ -26,6 +26,29 @@ use storage::storage::Storage;
 
 use crate::{AclCategory, Cmd, CmdFlags, CmdMeta, impl_cmd_clone_box, impl_cmd_meta};
 
+/// INFO VECTOR section: the Phase 1 index kind plus the FLAT query counters.
+/// vector_sets / vector_elements are omitted on purpose: counting them needs
+/// a full keyspace scan and INFO must stay O(1).
+fn vector_section(storage: &Storage) -> String {
+    let metrics = storage.vector_metrics();
+    format!(
+        "# Vector\r\n\
+         index-kind:flat\r\n\
+         vector_flat_queries_total:{}\r\n\
+         vector_flat_query_timeouts_total:{}\r\n\
+         vector_flat_query_errors_total:{}\r\n\
+         vector_search_capacity_rejected_total:{}\r\n\
+         vector_flat_query_duration_micros_total:{}\r\n\
+         vector_flat_query_duration_count:{}\r\n",
+        metrics.flat_queries_total,
+        metrics.flat_query_timeouts_total,
+        metrics.flat_query_errors_total,
+        metrics.capacity_rejected_total,
+        metrics.flat_query_duration_micros_total,
+        metrics.flat_query_duration_count,
+    )
+}
+
 /// INFO command - Show server information including cluster status
 #[derive(Clone, Default)]
 pub struct InfoCmd {
@@ -54,7 +77,7 @@ impl Cmd for InfoCmd {
         true
     }
 
-    fn do_cmd(&self, client: &Client, _storage: Arc<Storage>) {
+    fn do_cmd(&self, client: &Client, storage: Arc<Storage>) {
         let section = if client.argv().len() > 1 {
             String::from_utf8_lossy(&client.argv()[1]).to_lowercase()
         } else {
@@ -69,6 +92,9 @@ impl Cmd for InfoCmd {
                 info.push_str("cluster_enabled:0\r\n");
                 info.push_str("cluster_state:disabled\r\n");
             }
+            "vector" => {
+                info.push_str(&vector_section(&storage));
+            }
             "server" | "default" => {
                 info.push_str("# Server\r\n");
                 info.push_str("redis_version:7.0.0\r\n");
@@ -95,6 +121,8 @@ impl Cmd for InfoCmd {
                     info.push_str("\r\n# Cluster\r\n");
                     info.push_str("cluster_enabled:0\r\n");
                     info.push_str("cluster_state:disabled\r\n");
+                    info.push_str("\r\n");
+                    info.push_str(&vector_section(&storage));
                 }
             }
             _ => {

src/cmd/src/auth.rs

@@ -27,6 +27,14 @@ use crate::{impl_cmd_clone_box, impl_cmd_meta};
 
 pub type RequirepassProvider = Arc<dyn Fn() -> Option<String> + Send + Sync>;
 
+fn no_requirepass() -> Option<String> {
+    None
+}
+
+pub(crate) fn no_requirepass_provider() -> RequirepassProvider {
+    Arc::new(no_requirepass as fn() -> Option<String>)
+}
+
 #[derive(Clone)]
 pub struct AuthCmd {
     meta: CmdMeta,
@@ -43,7 +51,7 @@ impl Default for AuthCmd {
                 acl_category: AclCategory::CONNECTION,
                 ..Default::default()
             },
-            requirepass_provider: Arc::new(|| None),
+            requirepass_provider: no_requirepass_provider(),
         }
     }
 }
@@ -223,4 +231,11 @@ mod tests {
         let cmd = AuthCmd::default();
         assert!(cmd.has_flag(CmdFlags::NO_AUTH));
     }
+
+    #[test]
+    fn default_requirepass_provider_has_nonzero_trait_object_data_size() {
+        let cmd = AuthCmd::default();
+
+        assert_ne!(std::mem::size_of_val(cmd.requirepass_provider.as_ref()), 0);
+    }
 }

src/cmd/src/hello.rs

@@ -23,6 +23,7 @@ use resp::{CommandType, HelloAuthResult, RespCommand, RespData, RespError};
 use storage::storage::Storage;
 use subtle::ConstantTimeEq;
 
+use crate::auth::no_requirepass_provider;
 use crate::{
     AclCategory, Cmd, CmdFlags, CmdMeta, RequirepassProvider, impl_cmd_clone_box, impl_cmd_meta,
 };
@@ -43,7 +44,7 @@ impl Default for HelloCmd {
                 acl_category: AclCategory::CONNECTION | AclCategory::FAST,
                 ..Default::default()
             },
-            requirepass_provider: Arc::new(|| None),
+            requirepass_provider: no_requirepass_provider(),
         }
     }
 }
@@ -226,4 +227,11 @@ mod tests {
 
         assert_eq!(client.name().as_slice(), b"my-client");
     }
+
+    #[test]
+    fn default_requirepass_provider_has_nonzero_trait_object_data_size() {
+        let cmd = HelloCmd::default();
+
+        assert_ne!(std::mem::size_of_val(cmd.requirepass_provider.as_ref()), 0);
+    }
 }

src/cmd/src/lib.rs

@@ -93,6 +93,7 @@ pub mod sunionstore;
 pub mod table;
 pub mod ttl;
 pub mod type_cmd;
+pub mod vector;
 pub mod zadd;
 pub mod zcard;
 pub mod zcount;

src/cmd/src/substr.rs

@@ -71,6 +71,7 @@ mod tests {
     use storage::{StorageOptions, safe_cleanup_test_db, unique_test_db_path};
 
     use super::*;
+    use crate::auth::no_requirepass_provider;
     use crate::table::create_command_table;
 
     struct TestStream;
@@ -143,7 +144,7 @@ mod tests {
             b"-100".to_vec(),
         ]);
 
-        let command_table = create_command_table(Arc::new(|| None));
+        let command_table = create_command_table(no_requirepass_provider());
         command_table
             .get("substr")
             .expect("SUBSTR should be publicly registered")