use drop for logging

crates/iota-genesis-builder/src/stardust/process_outputs.rs

@@ -73,7 +73,6 @@ where
 
     fn next(&mut self) -> Option<Self::Item> {
         let Some(mut output) = self.outputs.next() else {
-            debug!("Number of scaled outputs: {}", self.num_scaled_outputs);
             return None;
         };
         if let Ok((_, inner)) = &mut output {
@@ -160,6 +159,12 @@ where
     }
 }
 
+impl<I> Drop for ScaleIotaAmountIterator<I> {
+    fn drop(&mut self) {
+        debug!("Number of scaled outputs: {}", self.num_scaled_outputs);
+    }
+}
+
 /// Filtering iterator that looks for vesting outputs that can be unlocked and
 /// stores them during the iteration. At the end of the iteration it merges all
 /// vesting outputs owned by a single address into a unique basic output.
@@ -218,14 +223,6 @@ where
         let Some((address, output_header_with_balance)) =
             self.unlocked_address_balances.pop_first()
         else {
-            debug!(
-                "Number of vesting outputs before merge: {}",
-                self.vesting_outputs.len()
-            );
-            debug!(
-                "Number of vesting outputs after merging: {}",
-                self.num_vesting_outputs
-            );
             return None;
         };
         self.num_vesting_outputs += 1;
@@ -240,6 +237,19 @@ where
     }
 }
 
+impl<I> Drop for UnlockedVestingIterator<I> {
+    fn drop(&mut self) {
+        debug!(
+            "Number of vesting outputs before merge: {}",
+            self.vesting_outputs.len()
+        );
+        debug!(
+            "Number of vesting outputs after merging: {}",
+            self.num_vesting_outputs
+        );
+    }
+}
+
 /// Iterator that looks for basic outputs having a tag being the Participation
 /// Tag and removes all features from the basic output.
 struct ParticipationOutputIterator<I> {
@@ -265,11 +275,6 @@ where
 
     fn next(&mut self) -> Option<Self::Item> {
         let Some(mut output) = self.outputs.next() else {
-            debug!(
-                "Number of participation outputs: {}",
-                self.participation_outputs.len()
-            );
-            debug!("Participation outputs: {:?}", self.participation_outputs);
             return None;
         };
         if let Ok((header, inner)) = &mut output {
@@ -287,6 +292,16 @@ where
     }
 }
 
+impl<I> Drop for ParticipationOutputIterator<I> {
+    fn drop(&mut self) {
+        debug!(
+            "Number of participation outputs: {}",
+            self.participation_outputs.len()
+        );
+        debug!("Participation outputs: {:?}", self.participation_outputs);
+    }
+}
+
 /// Extension trait that allows simpler methods for chaining filter iterators.
 trait IteratorExt: Iterator<Item = Result<(OutputHeader, Output)>> + Sized {
     fn scale_iota_amount(self) -> ScaleIotaAmountIterator<Self> {