From 3d2c863ae14287b0b24e0fd2dfcd599cc1b396c0 Mon Sep 17 00:00:00 2001
From: louib <code@louib.net>
Date: Fri, 22 Mar 2024 21:19:00 -0400
Subject: [PATCH] feat: finish implementing reachable nodes

---
 src/nix.rs | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/nix.rs b/src/nix.rs
index 5222b16..f536eb0 100644
--- a/src/nix.rs
+++ b/src/nix.rs
@@ -903,12 +903,12 @@ impl PackageGraph {
         package_graph_stats.nodes_count = self.nodes.len();
         package_graph_stats.root_nodes_count = self.root_nodes.len();
         for root_node in &self.root_nodes {
-            package_graph_stats
-                .reachable_nodes_count
-                .insert(root_node.clone(), 0);
             let mut visited_children: HashSet<String> = HashSet::default();
             let package_node = self.nodes.get(root_node).unwrap();
-            let reachable_nodes_count = package_node.get_nodes_count(&self.nodes, &mut visited_children);
+            package_graph_stats.reachable_nodes_count.insert(
+                root_node.clone(),
+                package_node.get_nodes_count(&self.nodes, &mut visited_children),
+            );
         }
         package_graph_stats
     }