enhancement: add log level override subcommand to ADP binary (#767)

## Summary
This PR adds two debug subcommands for dynamically configuring the log
level of the ADP process: `debug set-log-level` and `debug
reset-log-level`.

For setting the log level, this should be `agent-data-plane debug
set-log-level`. The subcommand would call out to /logging/override. For
resetting the log level, this should be `agent-data-plane debug
reset-log-level`. The subcommand would call out to /logging/reset.

## Change Type
- [ ] Bug fix
- [x] New feature
- [ ] Non-functional (chore, refactoring, docs)
- [ ] Performance


## How did you test this PR?

Manually tested by running the run-adp-standalone and testing various
set and reset commands

## References

Closes #731.

Author: brianna-wang-dd

Cargo.lock

@@ -23,6 +23,7 @@ memory-accounting = { workspace = true }
 papaya = { workspace = true }
 rand = { workspace = true }
 rand_distr = { workspace = true } 
+reqwest = { workspace = true, features = ["rustls-tls-native-roots-no-provider", "json"] }
 saluki-app = { workspace = true, features = ["full"] }
 saluki-common = { workspace = true }
 saluki-components = { workspace = true }

bin/agent-data-plane/Cargo.toml

@@ -5,7 +5,7 @@ use clap::{Args, Parser, Subcommand};
 #[derive(Parser)]
 #[command(about)]
 pub struct Cli {
-    /// Subcommand to run.    
+    /// Subcommand to run.   
     #[command(subcommand)]
     pub action: Option<Action>,
 }
@@ -15,6 +15,10 @@ pub enum Action {
     /// Runs the data plane.
     #[command(name = "run")]
     Run(RunConfig),
+
+    /// Various debugging commands.
+    #[command(subcommand)]
+    Debug(DebugConfig),
 }
 
 /// Run subcommand configuration.
@@ -24,3 +28,27 @@ pub struct RunConfig {
     #[arg(short = 'c', long = "config", default_value = "/etc/datadog-agent/datadog.yaml")]
     pub config: PathBuf,
 }
+
+/// Debug subcommand configuration.
+#[derive(Subcommand, Debug)]
+pub enum DebugConfig {
+    /// Resets log level.
+    #[command(name = "reset-log-level")]
+    ResetLogLevel,
+
+    /// Overrides the current log level.
+    #[command(name = "set-log-level")]
+    SetLogLevel(SetLogLevelConfig),
+}
+
+/// Set log level configuration.
+#[derive(Args, Debug)]
+pub struct SetLogLevelConfig {
+    /// Filter directives to apply.
+    #[arg(required = true)]
+    pub filter_directives: String,
+
+    /// Amount of time to apply the log level override, in seconds.
+    #[arg(required = true)]
+    pub duration_secs: u64,
+}

bin/agent-data-plane/src/config.rs

@@ -29,7 +29,7 @@ use tracing::{error, info, warn};
 mod components;
 
 mod config;
-use self::config::{Action, Cli, RunConfig};
+use self::config::{Action, Cli, DebugConfig, RunConfig};
 
 mod env_provider;
 use self::env_provider::ADPEnvironmentProvider;
@@ -70,19 +70,69 @@ async fn main() {
         fatal_and_exit(format!("failed to initialize TLS: {}", e));
     }
 
-    // Use default configuration path when no subcommand is provided.
-    let run_config = match cli.action {
-        Some(Action::Run(config)) => config,
-        None => RunConfig {
-            config: std::path::PathBuf::from("/etc/datadog-agent/datadog.yaml"),
+    let client = reqwest::Client::builder()
+        .danger_accept_invalid_certs(true)
+        .build()
+        .unwrap();
+
+    match cli.action {
+        Some(Action::Run(config)) => match run(started, config).await {
+            Ok(()) => info!("Agent Data Plane stopped."),
+            Err(e) => {
+                error!("{:?}", e);
+                std::process::exit(1);
+            }
         },
-    };
-
-    match run(started, run_config).await {
-        Ok(()) => info!("Agent Data Plane stopped."),
-        Err(e) => {
-            error!("{:?}", e);
-            std::process::exit(1);
+        Some(Action::Debug(DebugConfig::ResetLogLevel)) => {
+            let response = match client.post("https://localhost:5101/logging/reset").send().await {
+                Ok(resp) => resp,
+                Err(e) => {
+                    error!("Failed to send request: {}", e);
+                    std::process::exit(1);
+                }
+            };
+
+            if response.status().is_success() {
+                println!("Log level reset successful");
+            } else {
+                eprintln!("Failed to reset log level: {}", response.status());
+            }
+        }
+        Some(Action::Debug(DebugConfig::SetLogLevel(config))) => {
+            let filter_directives = config.filter_directives;
+            let duration_secs = config.duration_secs;
+
+            let response = match client
+                .post("https://localhost:5101/logging/override")
+                .query(&[("time_secs", duration_secs)])
+                .body(filter_directives)
+                .send()
+                .await
+            {
+                Ok(resp) => resp,
+                Err(e) => {
+                    error!("Failed to send request: {}", e);
+                    std::process::exit(1);
+                }
+            };
+
+            if response.status().is_success() {
+                println!("Log level override successful");
+            } else {
+                eprintln!("Failed to override log level: {}", response.status());
+            }
+        }
+        None => {
+            let default_config = RunConfig {
+                config: std::path::PathBuf::from("/etc/datadog-agent/datadog.yaml"),
+            };
+            match run(started, default_config).await {
+                Ok(()) => info!("Agent Data Plane stopped."),
+                Err(e) => {
+                    error!("{:?}", e);
+                    std::process::exit(1);
+                }
+            }
         }
     }
 }
@@ -217,8 +267,8 @@ async fn create_topology(
         }
         Err(_) => {
             info!(
-                "Dynamic configuration refreshing will be unavailable due to failure to configure refresher configuration."
-            )
+               "Dynamic configuration refreshing will be unavailable due to failure to configure refresher configuration."
+           )
         }
     }