feat: extract new prio-graph-scheduler to reuse PrioGraphScheduler

soonlabs · Oct 11, 2024 · 064d9a1 · 064d9a1
1 parent c7a7c76
commit 064d9a1
Show file tree

Hide file tree

Showing 10 changed files with 2,860 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/core/src/banking_stage.rs b/core/src/banking_stage.rs
@@ -62,7 +62,7 @@ pub mod qos_service;
 pub mod unprocessed_packet_batches;
 pub mod unprocessed_transaction_storage;
 
-mod consume_worker;
+pub mod consume_worker;
 mod decision_maker;
 mod forward_packet_batches_by_accounts;
 mod forward_worker;
@@ -73,7 +73,7 @@ mod multi_iterator_scanner;
 mod packet_deserializer;
 mod packet_filter;
 mod packet_receiver;
-mod read_write_account_set;
+pub mod read_write_account_set;
 #[allow(dead_code)]
 mod scheduler_messages;
 mod transaction_scheduler;

diff --git a/prio-graph-scheduler/Cargo.toml b/prio-graph-scheduler/Cargo.toml
@@ -12,9 +12,25 @@ edition.workspace = true
 [dependencies]
 solana-core = { workspace = true }
 solana-sdk = { workspace = true }
+solana-poh = { workspace = true }
+solana-metrics = { workspace = true }
+solana-ledger = { workspace = true }
+solana-runtime = { workspace = true }
+solana-gossip = { workspace = true }
+solana-cost-model = { workspace = true }
+solana-measure = { workspace = true }
 
 ahash = { workspace = true }
+prio-graph = { workspace = true }
+thiserror = { workspace = true }
+itertools = { workspace = true }
+log = { workspace = true }
+crossbeam-channel = { workspace = true }
+arrayvec = { workspace = true }
+min-max-heap = { workspace = true }
 
+[dev-dependencies]
+assert_matches = { workspace = true }
 
 [package.metadata.docs.rs]
 targets = ["x86_64-unknown-linux-gnu"]

diff --git a/prio-graph-scheduler/src/lib.rs b/prio-graph-scheduler/src/lib.rs
@@ -3,4 +3,17 @@ pub mod transaction_state;
 pub mod scheduler_messages;
 pub mod id_generator;
 pub mod in_flight_tracker;
-pub mod thread_aware_account_locks;
+pub mod thread_aware_account_locks;
+pub mod transaction_priority_id;
+pub mod scheduler_error;
+pub mod scheduler_metrics;
+// pub mod scheduler_controller;
+pub mod transaction_state_container;
+pub mod prio_graph_scheduler;
+
+#[macro_use]
+extern crate solana_metrics;
+
+#[cfg(test)]
+#[macro_use]
+extern crate assert_matches;