Create benchmark

aptos-labs · Nov 18, 2024 · 36ef86c · 36ef86c
1 parent e569dd2
commit 36ef86c
Show file tree

Hide file tree

Showing 4 changed files with 1,459 additions and 20 deletions.
diff --git a/aptos-move/e2e-benchmark/src/main.rs b/aptos-move/e2e-benchmark/src/main.rs
@@ -144,6 +144,8 @@ fn main() {
         (386, EntryPoints::TokenV2AmbassadorMint { numbered: true }),
         (467, EntryPoints::LiquidityPoolSwap { is_stable: true }),
         (429, EntryPoints::LiquidityPoolSwap { is_stable: false }),
+        (310, EntryPoints::APTPermissionedTransfer),
+        (150, EntryPoints::APTTransfer),
     ];
 
     let mut failures = Vec::new();

diff --git a/crates/transaction-generator-lib/src/publishing/module_simple.rs b/crates/transaction-generator-lib/src/publishing/module_simple.rs
@@ -267,6 +267,8 @@ pub enum EntryPoints {
         num_points_per_txn: usize,
     },
     DeserializeU256,
+    APTPermissionedTransfer,
+    APTTransfer,
 }
 
 impl EntryPoints {
@@ -309,7 +311,9 @@ impl EntryPoints {
             | EntryPoints::ResourceGroupsSenderWriteTag { .. }
             | EntryPoints::ResourceGroupsSenderMultiChange { .. }
             | EntryPoints::CoinInitAndMint
-            | EntryPoints::FungibleAssetMint => "framework_usecases",
+            | EntryPoints::FungibleAssetMint
+            | EntryPoints::APTPermissionedTransfer
+            | EntryPoints::APTTransfer => "framework_usecases",
             EntryPoints::TokenV2AmbassadorMint { .. } | EntryPoints::TokenV2AmbassadorBurn => {
                 "ambassador_token"
             },
@@ -381,7 +385,9 @@ impl EntryPoints {
             },
             EntryPoints::IncGlobalMilestoneAggV2 { .. }
             | EntryPoints::CreateGlobalMilestoneAggV2 { .. } => "counter_with_milestone",
-            EntryPoints::DeserializeU256 => "bcs_stream",
+            | EntryPoints::DeserializeU256 => "bcs_stream",
+            EntryPoints::APTPermissionedTransfer
+            | EntryPoints::APTTransfer => "permissioned_transfer",
         }
     }
 
@@ -722,6 +728,22 @@ impl EntryPoints {
                     ],
                 )
             },
+            EntryPoints::APTPermissionedTransfer => get_payload(
+                module_id,
+                ident_str!("transfer_permissioned").to_owned(),
+                vec![
+                    bcs::to_bytes(&other.expect("Must provide other")).unwrap(),
+                    bcs::to_bytes(&1u64).unwrap(),
+                ],
+            ),
+            EntryPoints::APTTransfer => get_payload(
+                module_id,
+                ident_str!("transfer").to_owned(),
+                vec![
+                    bcs::to_bytes(&other.expect("Must provide other")).unwrap(),
+                    bcs::to_bytes(&1u64).unwrap(),
+                ],
+            )
         }
     }
 
@@ -829,6 +851,8 @@ impl EntryPoints {
             EntryPoints::DeserializeU256 => AutomaticArgs::None,
             EntryPoints::IncGlobalMilestoneAggV2 { .. } => AutomaticArgs::None,
             EntryPoints::CreateGlobalMilestoneAggV2 { .. } => AutomaticArgs::Signer,
+            EntryPoints::APTPermissionedTransfer
+            | EntryPoints::APTTransfer => AutomaticArgs::Signer,
         }
     }
 }