feat(cubestore): Initial support for KV cache (#5861)

Author: ovr

rust/cubestore/Cargo.lock

@@ -52,7 +52,7 @@ tempfile = "3.2.0"
 tar = "0.4.38"
 
 [dev-dependencies]
-criterion = { version = "0.3.5", features = ["async_tokio"] }
+criterion = { version = "0.4.0", features = ["async_tokio", "html_reports"] }
 rocksdb = { version = "0.16.0", default-features = false, features = ["bzip2"] }
 
 [[bench]]

rust/cubestore/cubestore-sql-tests/Cargo.toml

@@ -35,6 +35,7 @@ pub fn cubestore_benches() -> Vec<Arc<dyn Bench>> {
     return vec![
         Arc::new(SimpleBench {}),
         Arc::new(ParquetMetadataCacheBench {}),
+        Arc::new(CacheSetGetBench {}),
     ];
 }
 
@@ -144,6 +145,42 @@ impl Bench for ParquetMetadataCacheBench {
     }
 }
 
+pub struct CacheSetGetBench;
+#[async_trait]
+impl Bench for CacheSetGetBench {
+    fn config(self: &Self, prefix: &str) -> (String, Config) {
+        let name = config_name(prefix, "cache_set_get");
+        let config = Config::test(name.as_str()).update_config(|c| c);
+        (name, config)
+    }
+
+    async fn setup(self: &Self, services: &CubeServices) -> Result<Arc<BenchState>, CubeError> {
+        services
+            .sql_service
+            .exec_query("CACHE SET TTL 600 'my_key' 'my_value'")
+            .await?;
+
+        let state = Arc::new(());
+        Ok(state)
+    }
+
+    async fn bench(
+        self: &Self,
+        services: &CubeServices,
+        _state: Arc<BenchState>,
+    ) -> Result<(), CubeError> {
+        let r = services
+            .sql_service
+            .exec_query("CACHE GET 'my_key'")
+            .await?;
+
+        let rows = to_rows(&r);
+        assert_eq!(rows, vec![vec![TableValue::String("my_value".to_string())]]);
+
+        Ok(())
+    }
+}
+
 async fn download_and_unzip(url: &str, dataset: &str) -> Result<Box<Path>, CubeError> {
     let root = std::env::current_dir()?.join("data");
     let dataset_path = root.join(dataset);