UseCaseAwareShuffler (#13738)

aptos-labs · Jul 16, 2024 · d98b854 · d98b854
1 parent 3754419
commit d98b854
Show file tree

Hide file tree

Showing 14 changed files with 1,157 additions and 16 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/consensus/Cargo.toml b/consensus/Cargo.toml
@@ -100,6 +100,7 @@ claims = { workspace = true }
 mockall = { workspace = true }
 move-core-types = { workspace = true }
 proptest = { workspace = true }
+proptest-derive = { workspace = true }
 tempfile = { workspace = true }
 
 [features]

diff --git a/consensus/src/transaction_shuffler/mod.rs b/consensus/src/transaction_shuffler/mod.rs
@@ -8,6 +8,7 @@ use std::sync::Arc;
 
 mod deprecated_fairness;
 mod sender_aware;
+mod use_case_aware;
 
 /// Interface to shuffle transactions
 pub trait TransactionShuffler: Send + Sync {
@@ -61,5 +62,21 @@ pub fn create_transaction_shuffler(
                 entry_fun_conflict_window_size: entry_fun_conflict_window_size as usize,
             })
         },
+        UseCaseAware {
+            sender_spread_factor,
+            platform_use_case_spread_factor,
+            user_use_case_spread_factor,
+        } => {
+            let config = use_case_aware::Config {
+                sender_spread_factor,
+                platform_use_case_spread_factor,
+                user_use_case_spread_factor,
+            };
+            info!(
+                config = ?config,
+                "Using use case aware transaction shuffling."
+            );
+            Arc::new(use_case_aware::UseCaseAwareShuffler { config })
+        },
     }
 }