bencher: add solana cli version to markdown (#97)

Author: buffalojoec

bencher/src/lib.rs

@@ -64,7 +64,7 @@ use {
     solana_account::Account,
     solana_instruction::Instruction,
     solana_pubkey::Pubkey,
-    std::path::PathBuf,
+    std::{path::PathBuf, process::Command},
 };
 
 /// A bench is a tuple of a name, an instruction, and a list of accounts.
@@ -113,6 +113,7 @@ impl<'a> MolluskComputeUnitBencher<'a> {
 
     /// Execute the benches.
     pub fn execute(&mut self) {
+        let solana_version = get_solana_version();
         let bench_results = std::mem::take(&mut self.benches)
             .into_iter()
             .map(|(name, instruction, accounts)| {
@@ -131,6 +132,15 @@ impl<'a> MolluskComputeUnitBencher<'a> {
                 MolluskComputeUnitBenchResult::new(name, result)
             })
             .collect::<Vec<_>>();
-        write_results(&self.out_dir, bench_results);
+        write_results(&self.out_dir, &solana_version, bench_results);
+    }
+}
+
+fn get_solana_version() -> String {
+    match Command::new("solana").arg("--version").output() {
+        Ok(output) if output.status.success() => {
+            String::from_utf8_lossy(&output.stdout).trim().to_string()
+        }
+        _ => "Unknown".to_string(),
     }
 }

bencher/src/result.rs

@@ -19,7 +19,11 @@ impl<'a> MolluskComputeUnitBenchResult<'a> {
     }
 }
 
-pub(crate) fn write_results(out_dir: &Path, results: Vec<MolluskComputeUnitBenchResult>) {
+pub(crate) fn write_results(
+    out_dir: &Path,
+    solana_version: &str,
+    results: Vec<MolluskComputeUnitBenchResult>,
+) {
     let path = out_dir.join("compute_units.md");
 
     // Load the existing bench content and parse the most recent table.
@@ -34,7 +38,7 @@ pub(crate) fn write_results(out_dir: &Path, results: Vec<MolluskComputeUnitBench
         .map(|content| parse_last_md_table(content));
 
     // Prepare to write a new table.
-    let mut md_table = md_header();
+    let mut md_table = md_header(solana_version);
 
     // Evaluate the results against the previous table, if any.
     // If there are changes, write a new table.
@@ -76,15 +80,17 @@ pub(crate) fn write_results(out_dir: &Path, results: Vec<MolluskComputeUnitBench
     }
 }
 
-fn md_header() -> String {
+fn md_header(solana_version: &str) -> String {
     let now: DateTime<Utc> = Utc::now();
     format!(
         r#"#### Compute Units: {}
 
+Solana CLI Version: {}
+
 | Name | CUs | Delta |
 |------|------|-------|
 "#,
-        now
+        now, solana_version,
     )
 }