add benchmark

rpcpool · Nov 20, 2024 · 27f2198 · 27f2198
1 parent 71c0390
commit 27f2198
Show file tree

Hide file tree

Showing 5 changed files with 335 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -27,6 +27,7 @@ bytes = "1.3.0"
 cargo-lock = "9.0.0"
 chrono = "0.4.26"
 clap = "4.3.0"
+criterion = "0.5.1"
 crossbeam-channel = "0.5.8"
 env_logger = "0.11.3"
 futures = "0.3.24"

diff --git a/yellowstone-grpc-proto/Cargo.toml b/yellowstone-grpc-proto/Cargo.toml
@@ -10,23 +10,31 @@ license = "Apache-2.0"
 keywords = { workspace = true }
 publish = true
 
+[[bench]]
+name = "encode"
+harness = false
+required-features = ["plugin-bench"]
+
 [dependencies]
 agave-geyser-plugin-interface = { workspace = true, optional = true }
 base64 = { workspace = true, optional = true }
 bincode = { workspace = true, optional = true }
 bs58 = { workspace = true, optional = true }
 bytes = { workspace = true, optional = true }
 prost = { workspace = true }
+prost_011 = { workspace = true, optional = true }
 serde = { workspace = true, optional = true }
 solana-account-decoder = { workspace = true, optional = true }
 solana-sdk = { workspace = true, optional = true }
+solana-storage-proto = { workspace = true, optional = true }
 solana-transaction-status = { workspace = true, optional = true }
 smallvec = { workspace = true, optional = true }
 spl-token-2022 = { workspace = true, optional = true }
 thiserror = { workspace = true, optional = true }
 tonic = { workspace = true }
 
 [dev-dependencies]
+criterion = { workspace = true }
 prost_011 = { workspace = true }
 solana-storage-proto = { workspace = true }
 
@@ -54,6 +62,7 @@ plugin = [
     "dep:spl-token-2022",
     "dep:thiserror"
 ]
+plugin-bench = ["plugin", "dep:prost_011", "dep:solana-storage-proto"]
 tonic-compression = ["tonic/gzip", "tonic/zstd"]
 
 [lints]