modify rust package structure

Signed-off-by: Kosuke Morimoto <[email protected]>
vdaas · Aug 22, 2024 · 42f548b · 42f548b
1 parent 4560a2b
commit 42f548b
Show file tree

Hide file tree

Showing 16 changed files with 355 additions and 126 deletions.
diff --git a/rust/Cargo.lock b/rust/Cargo.lock
diff --git a/rust/Cargo.toml b/rust/Cargo.toml
@@ -15,9 +15,10 @@
 #
 [workspace]
 members = [
-	"libs/ngt",
-	"libs/ngt-rs",
 	"libs/observability",
 	"libs/proto",
 	"bin/agent",
+	"libs/algorithm",
+	"libs/algorithms/ngt",
+	"libs/algorithms/faiss",
 ]
diff --git a/rust/bin/agent/Cargo.toml b/rust/bin/agent/Cargo.toml
@@ -21,9 +21,9 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-ngt = { version = "0.1.0", path = "../../libs/ngt" }
-prost = "0.12.4"
+algorithm = { version = "0.1.0", path = "../../libs/algorithm" }
+prost = "0.13.1"
 proto = { version = "0.1.0", path = "../../libs/proto" }
-tokio = { version = "1.37.0", features = ["full"] }
+tokio = { version = "1.39.3", features = ["full"] }
 tokio-stream = { version = "0.1.15", features = ["full"] }
-tonic = "0.11.0"
+tonic = "0.12.1"
diff --git a/rust/libs/algorithm/Cargo.toml b/rust/libs/algorithm/Cargo.toml
@@ -0,0 +1,8 @@
+[package]
+name = "algorithm"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+faiss = { version = "0.1.0", path = "../algorithms/faiss" }
+ngt = { version = "0.1.0", path = "../algorithms/ngt" }
diff --git a/rust/libs/algorithm/src/lib.rs b/rust/libs/algorithm/src/lib.rs
@@ -0,0 +1,14 @@
+pub fn add(left: u64, right: u64) -> u64 {
+    left + right
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn it_works() {
+        let result = add(2, 2);
+        assert_eq!(result, 4);
+    }
+}
diff --git a/rust/libs/algorithms/faiss/Cargo.toml b/rust/libs/algorithms/faiss/Cargo.toml
@@ -0,0 +1,6 @@
+[package]
+name = "faiss"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
diff --git a/rust/libs/algorithms/faiss/src/lib.rs b/rust/libs/algorithms/faiss/src/lib.rs
@@ -0,0 +1,14 @@
+pub fn add(left: u64, right: u64) -> u64 {
+    left + right
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn it_works() {
+        let result = add(2, 2);
+        assert_eq!(result, 4);
+    }
+}
diff --git a/rust/libs/ngt-rs/Cargo.toml → rust/libs/algorithms/ngt/Cargo.toml b/rust/libs/ngt-rs/Cargo.toml → rust/libs/algorithms/ngt/Cargo.toml
@@ -14,17 +14,17 @@
 # limitations under the License.
 #
 [package]
-name = "ngt-rs"
+name = "ngt"
 version = "0.1.0"
 edition = "2021"
 
 [dependencies]
-anyhow = "1.0.81"
-cxx = { version = "1.0.117", features = ["c++20"] }
+anyhow = "1.0.86"
+cxx = { version = "1.0.126", features = ["c++20"] }
 
 [build-dependencies]
-cxx-build = "1.0.117"
-miette = { version = "7.1.0", features = ["fancy"] }
+cxx-build = "1.0.126"
+miette = { version = "7.2.0", features = ["fancy"] }
 
 [dev-dependencies]
 rand = "0.8.5"
diff --git a/rust/libs/ngt-rs/build.rs → rust/libs/algorithms/ngt/build.rs b/rust/libs/ngt-rs/build.rs → rust/libs/algorithms/ngt/build.rs
diff --git a/rust/libs/ngt-rs/src/input.cpp → rust/libs/algorithms/ngt/src/input.cpp b/rust/libs/ngt-rs/src/input.cpp → rust/libs/algorithms/ngt/src/input.cpp
@@ -14,8 +14,8 @@
 // limitations under the License.
 //
 #include <vector>
-#include "ngt-rs/src/input.h"
-#include "ngt-rs/src/lib.rs.h"
+#include "ngt/src/input.h"
+#include "ngt/src/lib.rs.h"
 
 Property::Property(): p() {}
 

diff --git a/rust/libs/ngt-rs/src/input.h → rust/libs/algorithms/ngt/src/input.h b/rust/libs/ngt-rs/src/input.h → rust/libs/algorithms/ngt/src/input.h
diff --git a/rust/libs/ngt-rs/src/lib.rs → rust/libs/algorithms/ngt/src/lib.rs b/rust/libs/ngt-rs/src/lib.rs → rust/libs/algorithms/ngt/src/lib.rs
@@ -42,7 +42,7 @@ pub mod ffi {
     }
 
     unsafe extern "C++" {
-        include!("ngt-rs/src/input.h");
+        include!("ngt/src/input.h");
 
         type Property;
         fn new_property() -> UniquePtr<Property>;

diff --git a/rust/libs/ngt/Cargo.toml b/rust/libs/ngt/Cargo.toml
diff --git a/rust/libs/ngt/src/lib.rs b/rust/libs/ngt/src/lib.rs
diff --git a/rust/libs/observability/Cargo.toml b/rust/libs/observability/Cargo.toml
@@ -24,8 +24,8 @@ edition = "2021"
 opentelemetry = { version = "0.23" }
 opentelemetry_sdk = { version = "0.23", features = ["rt-tokio"] }
 opentelemetry-otlp = { version = "0.16.0", features = ["http-proto", "reqwest-client", "logs"] }
-tokio = { version = "1.38.0", features = ["full"] }
-serde_json = { version="1.0.120" }
+tokio = { version = "1.39.3", features = ["full"] }
+serde_json = { version="1.0.125" }
 opentelemetry-semantic-conventions = { version = "0.16.0"}
 scopeguard = { version = "1.2.0"}
 paste = {version = "1.0.15"}

diff --git a/rust/libs/proto/Cargo.toml b/rust/libs/proto/Cargo.toml
@@ -22,6 +22,6 @@ edition = "2021"
 
 [dependencies]
 futures-core = "0.3.30"
-prost = "0.12.3"
-tonic = "0.11.0"
-tonic-types = "0.11.0"
+prost = "0.13.1"
+tonic = "0.12.1"
+tonic-types = "0.12.1"