Aptos binding

rust/Cargo.toml

@@ -1,2 +1,2 @@
 [workspace]
-members = ["bft-bench-core", "bft-bench-shortcircuit"]
+members = ["bft-bench-core", "bft-bench-shortcircuit", "bft-bench-aptos"]

rust/bft-bench-aptos/Cargo.toml

@@ -0,0 +1,16 @@
+[package]
+name = "bft-bench-aptos"
+version = "0.0.1"
+edition = "2021"
+authors = ["Fabio Tudone <[email protected]>"]
+
+[dependencies]
+bft-bench-core = { path = "../bft-bench-core" }
+async-trait = "0.1.58"
+tokio = { version = "1.21.2", features = ["full"] }
+uuid = { version = "1.2.0", features = ["v4"] }
+rand = "0.7.3"
+anyhow = "1.0.66"
+once_cell = "1.16.0"
+url = "2.3.1"
+aptos-sdk = { git = "https://github.com/aptos-labs/aptos-core", branch = "devnet" }

rust/bft-bench-aptos/src/bin/aptos-bench.rs

@@ -0,0 +1,73 @@
+use anyhow::{Context, Result};
+use aptos_sdk::coin_client::CoinClient;
+use aptos_sdk::rest_client::{Client, FaucetClient};
+use aptos_sdk::types::LocalAccount;
+use once_cell::sync::Lazy;
+use std::str::FromStr;
+use url::Url;
+
+static NODE_URL: Lazy<Url> = Lazy::new(|| {
+    Url::from_str(
+        std::env::var("APTOS_NODE_URL")
+            .as_ref()
+            .map(|s| s.as_str())
+            .unwrap_or("https://fullnode.devnet.aptoslabs.com"),
+    )
+    .unwrap()
+});
+
+static FAUCET_URL: Lazy<Url> = Lazy::new(|| {
+    Url::from_str(
+        std::env::var("APTOS_FAUCET_URL")
+            .as_ref()
+            .map(|s| s.as_str())
+            .unwrap_or("https://faucet.devnet.aptoslabs.com"),
+    )
+    .unwrap()
+});
+
+#[tokio::main]
+async fn main() -> Result<()> {
+    // :!:>section_1a
+    let rest_client = Client::new(NODE_URL.clone());
+    let faucet_client = FaucetClient::new(FAUCET_URL.clone(), NODE_URL.clone()); // <:!:section_1a
+    let coin_client = CoinClient::new(&rest_client);
+
+    // Create two accounts locally, Alice and Bob.
+    let alice = LocalAccount::generate(&mut rand::rngs::OsRng);
+    let bob = LocalAccount::generate(&mut rand::rngs::OsRng); // <:!:section_2
+
+    // Print account addresses.
+    println!("\n=== Addresses ===");
+    println!("Alice: {}", alice.address().to_hex_literal());
+    println!("Bob: {}", bob.address().to_hex_literal());
+
+    // Create the accounts on chain, but only fund Alice.
+    faucet_client
+        .fund(alice.address(), 100_000_000)
+        .await
+        .context("Failed to fund Alice's account")?;
+    faucet_client
+        .create_account(bob.address())
+        .await
+        .context("Failed to fund Bob's account")?;
+
+    // Print initial balances.
+    println!("\n=== Initial Balances ===");
+    println!(
+        "Alice: {:?}",
+        coin_client
+            .get_account_balance(&alice.address())
+            .await
+            .context("Failed to get Alice's account balance")?
+    );
+    println!(
+        "Bob: {:?}",
+        coin_client
+            .get_account_balance(&bob.address())
+            .await
+            .context("Failed to get Bob's account balance")?
+    );
+
+    Ok(())
+}