Add keys method to nodes/rels

neo4j-labs · Aug 4, 2023 · 34b8b20 · 34b8b20
1 parent 48e5a9d
commit 34b8b20
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 0 deletions.
diff --git a/lib/src/row.rs b/lib/src/row.rs
@@ -149,6 +149,11 @@ impl Node {
         self.inner.labels.iter().map(|l| l.to_string()).collect()
     }
 
+    /// Get the names of the attributes of this node
+    pub fn keys(&self) -> Vec<&str> {
+        self.to::<crate::Keys<_>>().unwrap().0
+    }
+
     /// Get the attributes of the node
     pub fn get<T: std::convert::TryFrom<BoltType>>(&self, key: &str) -> Option<T> {
         self.inner.get(key)
@@ -184,6 +189,11 @@ impl Relation {
         self.inner.typ.value.clone()
     }
 
+    /// Get the names of the attributes of this relationship
+    pub fn keys(&self) -> Vec<&str> {
+        self.to::<crate::Keys<_>>().unwrap().0
+    }
+
     pub fn get<T: std::convert::TryFrom<BoltType>>(&self, key: &str) -> Option<T> {
         self.inner.get(key)
     }
@@ -210,6 +220,11 @@ impl UnboundedRelation {
         self.inner.typ.value.clone()
     }
 
+    /// Get the names of the attributes of this relationship
+    pub fn keys(&self) -> Vec<&str> {
+        self.to::<crate::Keys<_>>().unwrap().0
+    }
+
     pub fn get<T: std::convert::TryFrom<BoltType>>(&self, key: &str) -> Option<T> {
         self.inner.get(key)
     }

diff --git a/lib/tests/nodes.rs b/lib/tests/nodes.rs
@@ -20,9 +20,11 @@ async fn nodes() {
         let node: Node = row.get("friend").unwrap();
         let id = node.id();
         let labels = node.labels();
+        let keys = node.keys();
         let name: String = node.get("name").unwrap();
         assert_eq!(name, "Mr Mark");
         assert_eq!(labels, vec!["Person"]);
+        assert_eq!(keys, vec!["name"]);
         assert!(id >= 0);
     }
 }
diff --git a/lib/tests/relationships.rs b/lib/tests/relationships.rs
@@ -17,5 +17,6 @@ async fn relationships() {
     assert!(relation.start_node_id() > -1);
     assert!(relation.end_node_id() > -1);
     assert_eq!(relation.typ(), "WORKS_AT");
+    assert_eq!(relation.keys(), vec!["as"]);
     assert_eq!(relation.get::<String>("as").unwrap(), "Engineer");
 }
diff --git a/lib/tests/unbounded_relationships.rs b/lib/tests/unbounded_relationships.rs
@@ -16,5 +16,6 @@ async fn unbounded_relationships() {
     assert!(relation.start_node_id() > -1);
     assert!(relation.end_node_id() > -1);
     assert_eq!(relation.typ(), "RELATED");
+    assert_eq!(relation.keys(), vec!["as"]);
     assert_eq!(relation.get::<String>("as").unwrap(), "friend");
 }