Add config track_query_stats (#1411)

edgedb · Nov 27, 2024 · 45187bf · 45187bf
1 parent c9104a5
commit 45187bf
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/commands/configure.rs b/src/commands/configure.rs
@@ -209,6 +209,9 @@ pub async fn configure(
         C::Set(Set {
             parameter: S::WarnOldScoping(ConfigStr { value }),
         }) => set(cli, "warn_old_scoping", None, value).await,
+        C::Set(Set {
+            parameter: S::TrackQueryStats(ConfigStr { value }),
+        }) => set(cli, "track_query_stats", None, value).await,
         C::Reset(Res { parameter }) => {
             use crate::commands::parser::ConfigParameter as C;
             let name = match parameter {
@@ -236,6 +239,7 @@ pub async fn configure(
                 C::CurrentEmailProviderName => "current_email_provider_name",
                 C::SimpleScoping => "simple_scoping",
                 C::WarnOldScoping => "warn_old_scoping",
+                C::TrackQueryStats => "track_query_stats",
             };
             let (status, _warnings) = cli
                 .execute(&format!("CONFIGURE INSTANCE RESET {name}"), &())

diff --git a/src/commands/parser.rs b/src/commands/parser.rs
@@ -611,6 +611,9 @@ pub enum ValueParameter {
 
     /// Whether to warn when depending on old scoping behavior.
     WarnOldScoping(ConfigStr),
+
+    /// Select what queries are tracked in sys::QueryStats.
+    TrackQueryStats(ConfigStr),
 }
 
 #[derive(clap::Subcommand, Clone, Debug)]
@@ -665,6 +668,8 @@ pub enum ConfigParameter {
     SimpleScoping,
     /// Whether to warn when depending on old scoping behavior.
     WarnOldScoping,
+    /// Select what queries are tracked in sys::QueryStats.
+    TrackQueryStats,
 }
 
 #[derive(clap::Args, Clone, Debug)]