implement multiple stake pools in intercom

input-output-hk · Jan 26, 2021 · 0a74e03 · 0a74e03
1 parent 5b35373
commit 0a74e03
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/jormungandr/src/fragment/process.rs b/jormungandr/src/fragment/process.rs
@@ -92,13 +92,15 @@ impl Process {
                     reply_handle.reply_ok(statuses);
                 }
                 TransactionMsg::SelectTransactions {
+                    pool_idx,
                     ledger,
                     block_date,
                     ledger_params,
                     selection_alg,
                     reply_handle,
                 } => {
-                    let contents = pool.select(0, ledger, block_date, ledger_params, selection_alg);
+                    let contents =
+                        pool.select(pool_idx, ledger, block_date, ledger_params, selection_alg);
                     reply_handle.reply_ok(contents);
                 }
             }

diff --git a/jormungandr/src/intercom.rs b/jormungandr/src/intercom.rs
@@ -523,6 +523,7 @@ pub enum TransactionMsg {
         ReplyHandle<HashMap<FragmentId, FragmentStatus>>,
     ),
     SelectTransactions {
+        pool_idx: usize,
         ledger: Ledger,
         block_date: BlockDate,
         ledger_params: LedgerParameters,

diff --git a/jormungandr/src/leadership/process.rs b/jormungandr/src/leadership/process.rs
@@ -638,6 +638,7 @@ async fn prepare_block(
     let (reply_handle, reply_future) = unary_reply(logger.clone());
 
     let msg = TransactionMsg::SelectTransactions {
+        pool_idx: 0,
         ledger: ledger.as_ref().clone(),
         block_date,
         ledger_params: epoch_parameters.as_ref().clone(),