removed previous schema preservation for nodes, edges (#209)

* removed previous schema preservation for nodes, edges --------- Co-authored-by: Michael Chin <chnmch@amazon.com>
aws · Nov 28, 2023 · 90aec5d · 90aec5d
1 parent 040f4e5
commit 90aec5d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 16 deletions.
diff --git a/Changelog.md b/Changelog.md
@@ -10,6 +10,7 @@ The next release will include the following feature enhancements and bug fixes:
 
 **Bug fixes**
 
+- Fixed node and edge counts not updating on connection re-synchronization (<<https://github.com/aws/graph-explorer/pull/209>)
 - Fixed issue with Transition2 findDOMNode deprecation (<https://github.com/aws/graph-explorer/pull/211>
 - Fixed loading spinner rotation when synchronizing schema (<https://github.com/aws/graph-explorer/pull/207>)
 - Fixed highlight not persisting on selected graph element (<https://github.com/aws/graph-explorer/pull/187>)

diff --git a/packages/graph-explorer/src/hooks/useUpdateSchema.ts b/packages/graph-explorer/src/hooks/useUpdateSchema.ts
@@ -17,37 +17,28 @@ const useUpdateSchema = () => {
         const currentSchema =
           typeof schema === "function" ? schema(prevSchema) : schema;
 
-        // Preserve vertices counts
-        const vertices = (
-          currentSchema?.vertices ||
-          prevSchema?.vertices ||
-          []
-        ).map(vertex => {
-          const prevVertex = prevSchema?.vertices.find(
-            v => v.type === vertex.type
-          );
+        // Vertices counts
+        const vertices = (currentSchema?.vertices || []).map(vertex => {
           return {
             ...vertex,
-            total: vertex.total ?? prevVertex?.total,
+            total: vertex.total,
           };
         });
 
-        // Preserve edges counts
+        // Edges counts
         const edges = (currentSchema?.edges || prevSchema?.edges || []).map(
           edge => {
-            const prevEdge = prevSchema?.edges.find(e => e.type === edge.type);
             return {
               ...edge,
-              total: edge.total ?? prevEdge?.total,
+              total: edge.total,
             };
           }
         );
 
         updatedSchema.set(id, {
-          totalVertices:
-            currentSchema?.totalVertices || prevSchema?.totalVertices || 0,
+          totalVertices: currentSchema?.totalVertices || 0,
           vertices,
-          totalEdges: currentSchema?.totalEdges || prevSchema?.totalEdges || 0,
+          totalEdges: currentSchema?.totalEdges || 0,
           edges,
           prefixes: prevSchema?.prefixes || [],
           lastUpdate: !currentSchema ? prevSchema?.lastUpdate : new Date(),