calculate total block weights inside rust gbt

backend/rust-gbt/src/gbt.rs

@@ -62,6 +62,7 @@ pub fn gbt(mempool: &mut ThreadTransactionsMap) -> GbtResult {
     let mut audit_pool: AuditPool = u32hashmap_with_capacity(STARTING_CAPACITY);
     let mut mempool_stack: Vec<u32> = Vec::with_capacity(STARTING_CAPACITY);
     let mut clusters: Vec<Vec<u32>> = Vec::new();
+    let mut block_weights: Vec<u32> = Vec::new();
 
     info!("Initializing working structs");
     for (uid, tx) in mempool {
@@ -192,6 +193,7 @@ pub fn gbt(mempool: &mut ThreadTransactionsMap) -> GbtResult {
             // finalize this block
             if !transactions.is_empty() {
                 blocks.push(transactions);
+                block_weights.push(block_weight);
             }
             // reset for the next block
             transactions = Vec::with_capacity(STARTING_CAPACITY);
@@ -221,6 +223,7 @@ pub fn gbt(mempool: &mut ThreadTransactionsMap) -> GbtResult {
     info!("add the final unbounded block if it contains any transactions");
     if !transactions.is_empty() {
         blocks.push(transactions);
+        block_weights.push(block_weight);
     }
 
     info!("make a list of dirty transactions and their new rates");
@@ -238,6 +241,7 @@ pub fn gbt(mempool: &mut ThreadTransactionsMap) -> GbtResult {
 
     GbtResult {
         blocks,
+        block_weights,
         clusters,
         rates,
     }

backend/rust-gbt/src/lib.rs

@@ -109,12 +109,14 @@ impl GbtGenerator {
 /// The result from calling the gbt function.
 ///
 /// This tuple contains the following:
-///   blocks: A 2D Vector of transaction IDs (u32), the inner Vecs each represent a block.
-/// clusters: A 2D Vector of transaction IDs representing clusters of dependent mempool transactions
-///    rates: A Vector of tuples containing transaction IDs (u32) and effective fee per vsize (f64)
+///        blocks: A 2D Vector of transaction IDs (u32), the inner Vecs each represent a block.
+/// block_weights: A Vector of total weights per block.
+///      clusters: A 2D Vector of transaction IDs representing clusters of dependent mempool transactions
+///         rates: A Vector of tuples containing transaction IDs (u32) and effective fee per vsize (f64)
 #[napi(constructor)]
 pub struct GbtResult {
     pub blocks: Vec<Vec<u32>>,
+    pub block_weights: Vec<u32>,
     pub clusters: Vec<Vec<u32>>,
     pub rates: Vec<Vec<f64>>, // Tuples not supported. u32 fits inside f64
 }