TransactionScheduler: Consume Scheduler w/ PrioGraph (#33612)

solana-labs · Oct 24, 2023 · b0dcaf2 · b0dcaf2
1 parent 8954689
commit b0dcaf2
Show file tree

Hide file tree

Showing 9 changed files with 742 additions and 6 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -117,9 +117,7 @@ members = [
     "zk-token-sdk",
 ]
 
-exclude = [
-    "programs/sbf",
-]
+exclude = ["programs/sbf"]
 
 # This prevents a Travis CI error when building for Windows.
 resolver = "2"
@@ -221,7 +219,9 @@ index_list = "0.2.7"
 indexmap = "2.0.2"
 indicatif = "0.17.7"
 itertools = "0.10.5"
-jemallocator = { package = "tikv-jemallocator", version = "0.4.1", features = ["unprefixed_malloc_on_supported_platforms"] }
+jemallocator = { package = "tikv-jemallocator", version = "0.4.1", features = [
+    "unprefixed_malloc_on_supported_platforms",
+] }
 js-sys = "0.3.64"
 json5 = "0.4.1"
 jsonrpc-core = "18.0.0"
@@ -260,6 +260,7 @@ pickledb = { version = "0.5.1", default-features = false }
 pkcs8 = "0.8.0"
 predicates = "2.1"
 pretty-hex = "0.3.0"
+prio-graph = "0.1.0"
 proc-macro2 = "1.0.69"
 proptest = "1.3"
 prost = "0.11.9"

diff --git a/core/Cargo.toml b/core/Cargo.toml
@@ -31,6 +31,7 @@ log = { workspace = true }
 lru = { workspace = true }
 min-max-heap = { workspace = true }
 num_enum = { workspace = true }
+prio-graph = { workspace = true }
 quinn = { workspace = true }
 rand = { workspace = true }
 rand_chacha = { workspace = true }

diff --git a/core/src/banking_stage/transaction_scheduler/mod.rs b/core/src/banking_stage/transaction_scheduler/mod.rs
@@ -11,4 +11,7 @@ mod batch_id_generator;
 #[allow(dead_code)]
 mod in_flight_tracker;
 #[allow(dead_code)]
+mod prio_graph_scheduler;
+mod scheduler_error;
+#[allow(dead_code)]
 mod transaction_id_generator;