feat: Add space id filter

0xPlaygrounds · Dec 12, 2024 · 669495e · 669495e
1 parent 35ea24c
commit 669495e
Show file tree

Hide file tree

Showing 6 changed files with 21 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/api/Cargo.toml b/api/Cargo.toml
@@ -17,6 +17,7 @@ tokio = { version = "1.42.0", features = ["macros", "rt-multi-thread"] }
 tracing = "0.1.41"
 tracing-subscriber = "0.3.19"
 
+sdk = { version = "0.1.0", path = "../sdk" }
 sink = { version = "0.1.0", path = "../sink" }
 
 [dev-dependencies]

diff --git a/api/src/main.rs b/api/src/main.rs
@@ -34,13 +34,22 @@ impl Query {
         &'a self,
         executor: &'a Executor<'_, '_, KnowledgeGraph, S>,
         id: String,
-        // context: &'a KnowledgeGraph,
+        space_id: Option<String>,
+        version_id: Option<String>,
     ) -> Option<Node> {
         // let query = QueryMapper::default().select_root_node(&id, &executor.look_ahead()).build();
         // tracing::info!("Query: {}", query);
 
+        let query = match (space_id, version_id) {
+            (Some(space_id), _) => sdk::neo4rs::query("MATCH (n {id: $id, space_id: $space_id}) RETURN n")
+                .param("id", id)
+                .param("space_id", space_id),
+            (None, _) => sdk::neo4rs::query("MATCH (n {id: $id}) RETURN n")
+                .param("id", id)
+        };
+
         executor.context().0
-            .find_node_by_id::<HashMap<String, serde_json::Value>>(&id)
+            .find_node::<HashMap<String, serde_json::Value>>(query)
             .await
             .expect("Failed to find node")
             .map(Node::from)
@@ -116,6 +125,10 @@ impl Node {
         &self.id
     }
 
+    fn space_id(&self) -> &str {
+        &self.space_id
+    }
+
     fn types(&self) -> &[String] {
         &self.types
     }

diff --git a/codegen/src/lib.rs b/codegen/src/lib.rs
@@ -39,7 +39,7 @@ pub fn ts_type_from_value_type(value_type: &Node<Named>) -> TsType {
     }
 }
 
-pub fn gen_type_constructor(kg: &sink::kg::Client, attributes: &[&(Node<Named>, Option<Node<Named>>)]) -> Constructor {
+pub fn gen_type_constructor(attributes: &[&(Node<Named>, Option<Node<Named>>)]) -> Constructor {
     let super_constructor = vec![quote_expr!("super(id, driver)")];
 
     let constuctor_setters = attributes.iter().map(|(attr, _)| {
@@ -225,7 +225,7 @@ pub async fn gen_type(kg: &sink::kg::Client, entity: &Node<Named>) -> anyhow::Re
     Ok(Decl::Class(class(
         entity.type_name(),
         attribute_class_props,
-        Some(gen_type_constructor(kg, &attributes)),
+        Some(gen_type_constructor(&attributes)),
         relation_methods,
         vec![],
         Some(ident("Entity")),

diff --git a/sdk/src/lib.rs b/sdk/src/lib.rs
@@ -8,3 +8,5 @@ pub mod relation;
 
 pub use ids::network_ids;
 pub use ids::system_ids;
+
+pub use neo4rs;
diff --git a/sink/src/kg/client.rs b/sink/src/kg/client.rs
@@ -608,7 +608,6 @@ impl Client {
             .next()
             .await?
             .map(|row| {
-                tracing::info!("Row: {:?}", row.to::<neo4rs::Relation>());
                 Ok::<_, DatabaseError>(Relation::<T>::try_from(row.to::<neo4rs::Relation>()?)?)
             })
             .transpose()?)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -8,3 +8,5 @@ pub mod relation;

		pub use ids::network_ids;
		pub use ids::system_ids;

		pub use neo4rs;