Print a summary at the end of the benchmark

Summary:
## This stack

Update or add logging and stats that we need to troubleshoot the performance of modern sync.

## This diff

A later diff will use this to enable benchmarks to run for a set amount of time.

Reviewed By: RajivTS

Differential Revision: D71624215

fbshipit-source-id: c627df3cfe786f818263cf2d02e03324c5b0a6a9

Author: andreacampi

eden/mononoke/modern_sync/src/commands/benchmark.rs

@@ -11,7 +11,11 @@ use std::sync::Arc;
 use anyhow::Result;
 use async_trait::async_trait;
 use clap::Parser;
+use clientinfo::ClientEntryPoint;
+use clientinfo::ClientInfo;
 use context::CoreContext;
+use context::SessionContainer;
+use metadata::Metadata;
 use mononoke_app::MononokeApp;
 use mutable_counters::MutableCounters;
 use slog::info;
@@ -28,6 +32,7 @@ pub struct CommandArgs {
     chunk_size: Option<u64>,
 }
 
+#[derive(Clone, Default)]
 struct MemoryMutableCounters {
     counters: Arc<std::sync::RwLock<std::collections::HashMap<String, i64>>>,
 }
@@ -79,24 +84,57 @@ impl MutableCounters for MemoryMutableCounters {
 pub async fn run(app: MononokeApp, args: CommandArgs) -> Result<()> {
     let app = Arc::new(app);
     let app_args = &app.args::<ModernSyncArgs>()?;
-    let (source_repo_args, dest_repo_name) = get_unsharded_repo_args(app.clone(), app_args).await?;
+    let (source_repo_args, source_repo_name, dest_repo_name) =
+        get_unsharded_repo_args(app.clone(), app_args).await?;
+    let ctx = new_context(&app);
+    let logger = app.logger().clone();
 
     let mc = MemoryMutableCounters::new();
 
-    info!(app.logger(), "Running sync-once loop");
+    let now = std::time::Instant::now();
     crate::sync::sync(
         app,
         Some(0),
-        source_repo_args,
-        dest_repo_name,
+        source_repo_args.clone(),
+        dest_repo_name.clone(),
         ExecutionType::SyncOnce,
         false,
         args.chunk_size.clone().unwrap_or(CHUNK_SIZE_DEFAULT),
         PathBuf::from(""),
         true,
-        Some(Arc::new(mc)),
+        Some(Arc::new(mc.clone())),
     )
     .await?;
+    let elapsed = now.elapsed();
+
+    info!(
+        logger,
+        "Benchmark: Sync {} to {:?} took {}ms",
+        elapsed.as_millis(),
+        &source_repo_name,
+        dest_repo_name,
+    );
+
+    info!(logger, "Counters:");
+    let mut counters = mc.get_all_counters(&ctx).await?;
+    counters.sort_by(|a, b| a.0.cmp(&b.0));
+    for (k, v) in counters {
+        info!(logger, "{}={}", k, v);
+    }
 
     Ok(())
 }
+
+fn new_context(app: &MononokeApp) -> CoreContext {
+    let mut metadata = Metadata::default();
+    metadata.add_client_info(ClientInfo::default_with_entry_point(
+        ClientEntryPoint::ModernSync,
+    ));
+
+    let scuba = app.environment().scuba_sample_builder.clone();
+    let session_container = SessionContainer::builder(app.fb)
+        .metadata(Arc::new(metadata))
+        .build();
+
+    session_container.new_context(app.logger().clone(), scuba)
+}

eden/mononoke/modern_sync/src/commands/sync_once.rs

@@ -32,7 +32,8 @@ pub struct CommandArgs {
 pub async fn run(app: MononokeApp, args: CommandArgs) -> Result<()> {
     let app = Arc::new(app);
     let app_args = &app.args::<ModernSyncArgs>()?;
-    let (source_repo_args, dest_repo_name) = get_unsharded_repo_args(app.clone(), app_args).await?;
+    let (source_repo_args, _, dest_repo_name) =
+        get_unsharded_repo_args(app.clone(), app_args).await?;
 
     info!(app.logger(), "Running sync-once loop");
     crate::sync::sync(

eden/mononoke/modern_sync/src/sync.rs

@@ -874,7 +874,7 @@ fn classify_entries(
 pub(crate) async fn get_unsharded_repo_args(
     app: Arc<MononokeApp>,
     app_args: &ModernSyncArgs,
-) -> Result<(SourceRepoArgs, String)> {
+) -> Result<(SourceRepoArgs, String, String)> {
     let source_repo: Repo = app.open_repo(&app_args.repo).await?;
     let source_repo_name = source_repo.repo_identity.name().to_string();
     let target_repo_name = app_args
@@ -883,7 +883,8 @@ pub(crate) async fn get_unsharded_repo_args(
         .unwrap_or(source_repo_name.clone());
 
     Ok((
-        SourceRepoArgs::with_name(source_repo_name),
+        SourceRepoArgs::with_name(source_repo_name.clone()),
+        source_repo_name,
         target_repo_name,
     ))
 }