fx_hashmap for fixed graphql schemas

Pometry · Dec 18, 2024 · e3ce6bc · e3ce6bc
1 parent dae9491
commit e3ce6bc
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/raphtory-graphql/Cargo.toml b/raphtory-graphql/Cargo.toml
@@ -48,6 +48,7 @@ config = { workspace = true }
 url = { workspace = true }
 base64-compat = { workspace = true }
 reqwest = { workspace = true }
+rustc-hash = { workspace = true }
 moka = { workspace = true }
 # python binding optional dependencies
 pyo3 = { workspace = true, optional = true }

diff --git a/raphtory-graphql/src/model/schema/mod.rs b/raphtory-graphql/src/model/schema/mod.rs
@@ -2,7 +2,8 @@ use raphtory::{
     db::graph::node::NodeView,
     prelude::{GraphViewOps, NodeViewOps},
 };
-use std::collections::{HashMap, HashSet};
+use rustc_hash::FxHashMap;
+use std::collections::HashSet;
 
 pub(crate) mod edge_schema;
 pub(crate) mod graph_schema;
@@ -21,7 +22,7 @@ fn get_node_type<'graph, G: GraphViewOps<'graph>>(node: NodeView<G>) -> String {
     }
 }
 
-type SchemaAggregate = HashMap<(String, String), HashSet<String>>;
+type SchemaAggregate = FxHashMap<(String, String), HashSet<String>>;
 
 fn merge_schemas(mut s1: SchemaAggregate, s2: SchemaAggregate) -> SchemaAggregate {
     for ((key, prop_type), set2) in s2 {

diff --git a/raphtory-graphql/src/model/schema/node_schema.rs b/raphtory-graphql/src/model/schema/node_schema.rs
@@ -52,7 +52,7 @@ impl NodeSchema {
         let schema: SchemaAggregate = filtered_nodes
             .map(collect_node_schema)
             .reduce(merge_schemas)
-            .unwrap_or_else(|| FxHashMap::default());
+            .unwrap_or_default();
 
         schema.into_iter().map(|prop| prop.into()).collect_vec()
     }