move new functions back to top

anza-xyz · Feb 19, 2025 · deb6390 · deb6390
1 parent 608fbbb
commit deb6390
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 27 deletions.
diff --git a/core/src/banking_stage/transaction_scheduler/greedy_scheduler.rs b/core/src/banking_stage/transaction_scheduler/greedy_scheduler.rs
@@ -48,6 +48,21 @@ pub struct GreedyScheduler<Tx: TransactionWithMeta> {
     config: GreedySchedulerConfig,
 }
 
+impl<Tx: TransactionWithMeta> GreedyScheduler<Tx> {
+    pub(crate) fn new(
+        consume_work_senders: Vec<Sender<ConsumeWork<Tx>>>,
+        finished_consume_work_receiver: Receiver<FinishedConsumeWork<Tx>>,
+        config: GreedySchedulerConfig,
+    ) -> Self {
+        Self {
+            common: SchedulingCommon::new(consume_work_senders, finished_consume_work_receiver),
+            working_account_set: ReadWriteAccountSet::default(),
+            unschedulables: Vec::with_capacity(config.max_scanned_transactions_per_scheduling_pass),
+            config,
+        }
+    }
+}
+
 impl<Tx: TransactionWithMeta> Scheduler<Tx> for GreedyScheduler<Tx> {
     fn schedule<S: StateContainer<Tx>>(
         &mut self,
@@ -195,21 +210,6 @@ impl<Tx: TransactionWithMeta> Scheduler<Tx> for GreedyScheduler<Tx> {
     }
 }
 
-impl<Tx: TransactionWithMeta> GreedyScheduler<Tx> {
-    pub(crate) fn new(
-        consume_work_senders: Vec<Sender<ConsumeWork<Tx>>>,
-        finished_consume_work_receiver: Receiver<FinishedConsumeWork<Tx>>,
-        config: GreedySchedulerConfig,
-    ) -> Self {
-        Self {
-            common: SchedulingCommon::new(consume_work_senders, finished_consume_work_receiver),
-            working_account_set: ReadWriteAccountSet::default(),
-            unschedulables: Vec::with_capacity(config.max_scanned_transactions_per_scheduling_pass),
-            config,
-        }
-    }
-}
-
 fn try_schedule_transaction<Tx: TransactionWithMeta>(
     transaction_state: &mut TransactionState<Tx>,
     pre_lock_filter: impl Fn(&Tx) -> bool,

diff --git a/core/src/banking_stage/transaction_scheduler/prio_graph_scheduler.rs b/core/src/banking_stage/transaction_scheduler/prio_graph_scheduler.rs
@@ -67,6 +67,20 @@ pub(crate) struct PrioGraphScheduler<Tx> {
     config: PrioGraphSchedulerConfig,
 }
 
+impl<Tx: TransactionWithMeta> PrioGraphScheduler<Tx> {
+    pub(crate) fn new(
+        consume_work_senders: Vec<Sender<ConsumeWork<Tx>>>,
+        finished_consume_work_receiver: Receiver<FinishedConsumeWork<Tx>>,
+        config: PrioGraphSchedulerConfig,
+    ) -> Self {
+        Self {
+            common: SchedulingCommon::new(consume_work_senders, finished_consume_work_receiver),
+            prio_graph: PrioGraph::new(passthrough_priority),
+            config,
+        }
+    }
+}
+
 impl<Tx: TransactionWithMeta> Scheduler<Tx> for PrioGraphScheduler<Tx> {
     /// Schedule transactions from the given `StateContainer` to be
     /// consumed by the worker threads. Returns summary of scheduling, or an
@@ -323,18 +337,6 @@ impl<Tx: TransactionWithMeta> Scheduler<Tx> for PrioGraphScheduler<Tx> {
 }
 
 impl<Tx: TransactionWithMeta> PrioGraphScheduler<Tx> {
-    pub(crate) fn new(
-        consume_work_senders: Vec<Sender<ConsumeWork<Tx>>>,
-        finished_consume_work_receiver: Receiver<FinishedConsumeWork<Tx>>,
-        config: PrioGraphSchedulerConfig,
-    ) -> Self {
-        Self {
-            common: SchedulingCommon::new(consume_work_senders, finished_consume_work_receiver),
-            prio_graph: PrioGraph::new(passthrough_priority),
-            config,
-        }
-    }
-
     /// Gets accessed accounts (resources) for use in `PrioGraph`.
     fn get_transaction_account_access(
         transaction: &SanitizedTransactionTTL<impl SVMMessage>,