Add BTD graph cache Scuba logging subcommand

Summary:
Adds a new `supertd log-graph-cache` CLI command that reads the metadata JSON
file produced by BTDCachedGraphDownloaderScriptController.php and logs it as a
BTD_GRAPH_CACHE_LOOKUP event to the supertd_events Scuba dataset.

This enables tracking of BTD graph cache hit/miss metrics for analysis and
dashboarding.

Differential Revision: D90788251

fbshipit-source-id: 3d70d4ffb712c353964a0dd9e8bc61c5fd727264

Author: Aaron Ho

supertd/bin/main.rs

@@ -35,6 +35,8 @@ enum Args {
     #[cfg(fbcode_build)]
     GraphCompressor(graph_compressor::Args),
     #[cfg(fbcode_build)]
+    LogGraphCache(td_util::btd_graph_cache::Args),
+    #[cfg(fbcode_build)]
     MetadataFetcher(metadata_fetcher::Args),
     #[cfg(fbcode_build)]
     VerifiableMatcher(verifiable_matcher::Args),
@@ -94,6 +96,8 @@ pub async fn main(fb: FacebookInit) -> ExitCode {
         #[cfg(fbcode_build)]
         Args::GraphCompressor(args) => graph_compressor::main(args),
         #[cfg(fbcode_build)]
+        Args::LogGraphCache(args) => td_util::btd_graph_cache::main(args),
+        #[cfg(fbcode_build)]
         Args::MetadataFetcher(args) => metadata_fetcher::main(args).await,
         #[cfg(fbcode_build)]
         Args::VerifiableMatcher(args) => verifiable_matcher::main(args).await,

td_util/src/btd_graph_cache.rs

@@ -0,0 +1,52 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is dual-licensed under either the MIT license found in the
+ * LICENSE-MIT file in the root directory of this source tree or the Apache
+ * License, Version 2.0 found in the LICENSE-APACHE file in the root directory
+ * of this source tree. You may select, at your option, one of the
+ * above-listed licenses.
+ */
+
+//! CLI command to log BTD graph cache lookup metadata to Scuba.
+//!
+//! Reads the JSON metadata file produced by BTDCachedGraphDownloaderScriptController.php
+//! and logs it as a BTD_GRAPH_CACHE_LOOKUP event.
+
+use std::path::PathBuf;
+
+use clap::Parser;
+
+use crate::workflow_error::WorkflowError;
+
+/// CLI arguments for the log-graph-cache subcommand.
+#[derive(Parser, Debug)]
+pub struct Args {
+    /// Path to the metadata JSON file produced by BTDCachedGraphDownloaderScriptController
+    #[arg(long)]
+    pub metadata_file: PathBuf,
+}
+
+/// Read graph cache lookup metadata from a file and log it to Scuba.
+pub fn main(args: Args) -> Result<(), WorkflowError> {
+    let start = std::time::Instant::now();
+
+    let contents = std::fs::read_to_string(&args.metadata_file).map_err(|e| {
+        anyhow::anyhow!(
+            "Failed to read metadata file {:?}: {}",
+            args.metadata_file,
+            e
+        )
+    })?;
+
+    let metadata: serde_json::Value = serde_json::from_str(&contents)
+        .map_err(|e| anyhow::anyhow!("Failed to parse metadata JSON: {}", e))?;
+
+    crate::scuba!(
+        event: BTD_GRAPH_CACHE_LOOKUP,
+        duration: start.elapsed(),
+        data: metadata
+    );
+
+    Ok(())
+}

td_util/src/lib.rs

@@ -9,6 +9,7 @@
  */
 
 #![forbid(unsafe_code)]
+pub mod btd_graph_cache;
 pub mod cli;
 pub mod command;
 pub mod executor;

td_util/src/supertd_events.rs

@@ -41,6 +41,7 @@ pub enum Event {
     TARGETS_WITHOUT_BUDGET_ENTITY,
     SCUBA_TARGET_LOGGING_FAILURE,
     XGBOOST_SIZING_PACKING_FAILURE,
+    BTD_GRAPH_CACHE_LOOKUP,
 }
 
 #[derive(Debug)]