add KVS trait

Signed-off-by: Kosuke Morimoto <[email protected]>

rust/Cargo.toml

@@ -14,4 +14,4 @@
 # limitations under the License.
 #
 [workspace]
-members = ["libs/proto", "libs/ngt", "bin/agent", "libs/algorithm"]
+members = ["libs/proto", "libs/ngt", "bin/agent", "libs/algorithm", "libs/kvs"]

rust/libs/algorithm/src/base.rs

@@ -1,11 +1,9 @@
 use anyhow::Result;
-use serde::Deserialize;
-use serde::Serialize;
 
-pub mod algorithm1 {
+pub mod algorithm {
     pub trait Base<T, U, Param: Deserialize, Response: Serialize> {
         fn search(&self, v: &Vec<T>, p: Option<&Param>) -> Result<&Response>;
-        fn insert(&self, v: &Vec<T>, p: Option<&Param>) -> Result<&Response>;
+        fn insert(&self, id: &U, v: &Vec<T>, p: Option<&Param>) -> Result<&Response>;
         fn update(&self, id: &U, v: &Vec<T>, p: Option<&Param>) -> Result<&Response>;
         fn remove(&self, id: &U, p: Option<&Param>) -> Result<&Response>;
         fn commit(&self, p: Option<&Param>) -> Result<&Response>;
@@ -14,45 +12,4 @@ pub mod algorithm1 {
         fn open(p: &str) -> Result<&Self>;
         fn save(p: &str) -> Result<&Self>;
     }
-
-    pub trait Param {
-        type A;
-        fn get_parameters() -> Result<&A>;
-        fn set_parameters(a: &A) -> Result<()>;
-    }
-
-    pub trait Response {
-        type Status;
-        type Result;
-        fn get_status() -> Status;
-        fn get_result() -> Result;
-    }
-}
-
-pub mod algorithm2 {
-    pub trait Base<Query: Deserialize, Response: Serialize, Param: Deserialize> {
-        fn search(&self, q: &Query) -> Result<&Response>;
-        fn insert(&self, q: &Query) -> Result<&Response>;
-        fn update(&self, q: &Query) -> Result<&Response>;
-        fn remove(&self, q: &Query) -> Result<&Response>;
-        fn commit(&self, q: &Query) -> Result<&Response>;
-
-        fn new(p: Option<&Param>) -> Result<&Self>;
-        fn open(p: &str) -> Result<&Self>;
-        fn save(p: &str) -> Result<&Self>;
-    }
-
-    // example like NGT
-    #[derive(Serialize, Deserialize, Debug)]
-    struct SearchQuery {
-        vector: Vec<f32>,
-        size: u64,
-        epsilon: f32,
-        radius: f32,
-    }
-
-    #[derive(Serialize, Deserialize, Debug)]
-    struct InsertQuery {
-        vector: Vec<f32>,
-    }
 }

rust/libs/kvs/Cargo.toml

@@ -0,0 +1,8 @@
+[package]
+name = "kvs"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]

rust/libs/kvs/src/kvs.rs

@@ -0,0 +1,12 @@
+use anyhow::Result;
+
+pub mod kvs {
+    pub trait KVS<K, V> {
+        fn set(k: &K, v: &V) -> Result<()>;
+        fn get_value(k: &K) -> Result<&V>;
+        fn get_key(v: &V) -> Result<&K>;
+
+        fn new(p: &str) -> Result<&Self>;
+        fn open(p: &str) -> Result<&Self>;
+    }
+}

rust/libs/kvs/src/lib.rs

@@ -0,0 +1,14 @@
+pub fn add(left: usize, right: usize) -> usize {
+    left + right
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn it_works() {
+        let result = add(2, 2);
+        assert_eq!(result, 4);
+    }
+}