Scheduler Messages (#30976)

core/src/banking_stage.rs

@@ -48,6 +48,8 @@ pub mod consumer;
 mod decision_maker;
 mod forwarder;
 mod packet_receiver;
+#[allow(dead_code)]
+mod scheduler_messages;
 
 #[allow(dead_code)]
 mod thread_aware_account_locks;

core/src/banking_stage/scheduler_messages.rs

@@ -0,0 +1,55 @@
+use {
+    crate::immutable_deserialized_packet::ImmutableDeserializedPacket,
+    solana_sdk::{clock::Slot, transaction::SanitizedTransaction},
+    std::sync::Arc,
+};
+
+/// A unique identifier for a transaction batch.
+#[derive(Clone, Copy, Debug, Hash, PartialEq, Eq)]
+pub struct TransactionBatchId(u64);
+
+impl TransactionBatchId {
+    pub fn new(index: u64) -> Self {
+        Self(index)
+    }
+}
+
+/// A unique identifier for a transaction.
+#[derive(Clone, Copy, Debug, Hash, PartialEq, Eq)]
+pub struct TransactionId(u64);
+
+impl TransactionId {
+    pub fn new(index: u64) -> Self {
+        Self(index)
+    }
+}
+
+/// Message: [Scheduler -> Worker]
+/// Transactions to be consumed (i.e. executed, recorded, and committed)
+pub struct ConsumeWork {
+    pub batch_id: TransactionBatchId,
+    pub ids: Vec<TransactionId>,
+    pub transactions: Vec<SanitizedTransaction>,
+    pub max_age_slots: Vec<Slot>,
+}
+
+/// Message: [Scheduler -> Worker]
+/// Transactions to be forwarded to the next leader(s)
+pub struct ForwardWork {
+    pub ids: Vec<TransactionId>,
+    pub packets: Vec<Arc<ImmutableDeserializedPacket>>,
+}
+
+/// Message: [Worker -> Scheduler]
+/// Processed transactions.
+pub struct FinishedConsumeWork {
+    pub work: ConsumeWork,
+    pub retryable_indexes: Vec<usize>,
+}
+
+/// Message: [Worker -> Scheduler]
+/// Forwarded transactions.
+pub struct FinishedForwardWork {
+    pub work: ForwardWork,
+    pub successful: bool,
+}