refactor(bench): continuous benchmarking improvements (#14821)

1 files changed, 99 insertions, 0 deletions

diff --git a/cli/bench/metrics.rs b/cli/bench/metrics.rs
new file mode 100644
index 000000000..c49ca9020
--- /dev/null
+++ b/cli/bench/metrics.rs
@@ -0,0 +1,99 @@
+// Copyright 2018-2022 the Deno authors. All rights reserved. MIT license.
+
+use google_storage1::api::Object;
+use google_storage1::hyper;
+use google_storage1::hyper_rustls;
+use google_storage1::oauth2;
+use google_storage1::Storage;
+use once_cell::sync::Lazy;
+use std::collections::HashMap;
+use std::io::Cursor;
+
+static GIT_HASH: Lazy<String> = Lazy::new(|| {
+  test_util::run_collect(&["git", "rev-parse", "HEAD"], None, None, None, true)
+    .0
+    .trim()
+    .to_string()
+});
+
+#[derive(serde::Serialize)]
+struct Metric {
+  name: String,
+  value: i64,
+  sha1: String,
+  #[serde(rename = "type")]
+  type_: String,
+  time: i64,
+}
+
+pub struct Reporter {
+  wtr: csv::Writer<Vec<u8>>,
+  gcloud_client: Option<Storage>,
+}
+
+impl Reporter {
+  pub async fn new() -> Self {
+    dotenv::dotenv().ok();
+    let gcloud_client =
+      match std::env::var("CI").map(|_| std::env::var("GOOGLE_SVC_KEY")) {
+        Ok(Ok(key_str)) => {
+          let secret = oauth2::parse_service_account_key(key_str)
+            .expect("Failed to load service account key");
+          let auth =
+            oauth2::authenticator::ServiceAccountAuthenticator::builder(secret)
+              .build()
+              .await
+              .unwrap();
+          let client = hyper::Client::builder().build(
+            hyper_rustls::HttpsConnectorBuilder::new()
+              .with_native_roots()
+              .https_or_http()
+              .enable_http1()
+              .enable_http2()
+              .build(),
+          );
+          Some(Storage::new(client, auth))
+        }
+        _ => None,
+      };
+    Self {
+      wtr: csv::Writer::from_writer(vec![]),
+      gcloud_client,
+    }
+  }
+
+  pub fn write_one(&mut self, type_: &str, name: &str, value: i64) {
+    self
+      .wtr
+      .serialize(Metric {
+        name: name.to_string(),
+        type_: type_.to_string(),
+        value,
+        sha1: GIT_HASH.clone(),
+        time: chrono::Utc::now().timestamp_millis(),
+      })
+      .unwrap();
+  }
+
+  pub fn write(&mut self, type_: &str, hashmap: &HashMap<String, i64>) {
+    for (name, value) in hashmap {
+      self.write_one(type_, name, *value);
+    }
+  }
+
+  pub async fn submit(mut self) {
+    self.wtr.flush().unwrap();
+    if let Some(client) = self.gcloud_client.take() {
+      let mut reader = Cursor::new(self.wtr.into_inner().unwrap());
+      let object: Object = Object::default();
+      client
+        .objects()
+        .insert(object, "deno_benchmark_data")
+        .name(&format!("{}.csv", *GIT_HASH))
+        .param("uploadType", "multipart")
+        .upload(&mut reader, "text/csv".parse().unwrap())
+        .await
+        .unwrap();
+    }
+  }
+}