Merge pull request #8778 from DarthMax/gds_arrow_server_triplet_field…

…_handler GDS Arrow Server - Triplet Field handlers
neo4j · Mar 7, 2024 · 076d677 · 076d677
2 parents af14d41 + 09fbc13
commit 076d677
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/core/src/main/java/org/neo4j/gds/core/loading/construction/PropertyValues.java b/core/src/main/java/org/neo4j/gds/core/loading/construction/PropertyValues.java
@@ -37,6 +37,8 @@ public abstract class PropertyValues {
 
     public abstract Iterable<String> propertyKeys();
 
+    public abstract Value get(String key);
+
     public static PropertyValues of(MapValue mapValue) {
         return new CypherPropertyValues(mapValue);
     }
@@ -71,6 +73,11 @@ public int size() {
         public Set<String> propertyKeys() {
             return this.properties.keySet();
         }
+
+        @Override
+        public Value get(String key) {
+            return properties.get(key);
+        }
     }
 
     private static final class CypherPropertyValues extends PropertyValues {
@@ -101,5 +108,10 @@ public int size() {
         public Iterable<String> propertyKeys() {
             return this.properties.keySet();
         }
+
+        @Override
+        public Value get(String key) {
+            return ValueConverter.toValue(properties.get(key));
+        }
     }
 }