Flatten the telemetry config section

Author: sandhose

crates/cli/src/telemetry.rs

@@ -16,7 +16,10 @@ use std::time::Duration;
 
 use anyhow::Context as _;
 use hyper::{header::CONTENT_TYPE, Body, Response};
-use mas_config::{MetricsExporterConfig, Propagator, TelemetryConfig, TracingExporterConfig};
+use mas_config::{
+    MetricsConfig, MetricsExporterKind, Propagator, TelemetryConfig, TracingConfig,
+    TracingExporterKind,
+};
 use opentelemetry::{
     global,
     propagation::{TextMapCompositePropagator, TextMapPropagator},
@@ -52,9 +55,9 @@ pub fn setup(config: &TelemetryConfig) -> anyhow::Result<Option<Tracer>> {
     mas_http::set_propagator(&propagator);
     global::set_text_map_propagator(propagator);
 
-    let tracer = tracer(&config.tracing.exporter).context("Failed to configure traces exporter")?;
+    let tracer = tracer(&config.tracing).context("Failed to configure traces exporter")?;
 
-    init_meter(&config.metrics.exporter).context("Failed to configure metrics exporter")?;
+    init_meter(&config.metrics).context("Failed to configure metrics exporter")?;
 
     Ok(tracer)
 }
@@ -114,13 +117,13 @@ fn otlp_tracer(endpoint: Option<&Url>) -> anyhow::Result<Tracer> {
     Ok(tracer)
 }
 
-fn tracer(config: &TracingExporterConfig) -> anyhow::Result<Option<Tracer>> {
-    let tracer_provider = match config {
-        TracingExporterConfig::None => return Ok(None),
-        TracingExporterConfig::Stdout => stdout_tracer_provider(),
-        TracingExporterConfig::Otlp { endpoint } => {
+fn tracer(config: &TracingConfig) -> anyhow::Result<Option<Tracer>> {
+    let tracer_provider = match config.exporter {
+        TracingExporterKind::None => return Ok(None),
+        TracingExporterKind::Stdout => stdout_tracer_provider(),
+        TracingExporterKind::Otlp => {
             // The OTLP exporter already creates a tracer and installs it
-            return Ok(Some(otlp_tracer(endpoint.as_ref())?));
+            return Ok(Some(otlp_tracer(config.endpoint.as_ref())?));
         }
     };
 
@@ -208,15 +211,15 @@ fn prometheus_metric_reader() -> anyhow::Result<PrometheusExporter> {
     Ok(exporter)
 }
 
-fn init_meter(config: &MetricsExporterConfig) -> anyhow::Result<()> {
+fn init_meter(config: &MetricsConfig) -> anyhow::Result<()> {
     let meter_provider_builder = SdkMeterProvider::builder();
-    let meter_provider_builder = match config {
-        MetricsExporterConfig::None => meter_provider_builder.with_reader(ManualReader::default()),
-        MetricsExporterConfig::Stdout => meter_provider_builder.with_reader(stdout_metric_reader()),
-        MetricsExporterConfig::Otlp { endpoint } => {
-            meter_provider_builder.with_reader(otlp_metric_reader(endpoint.as_ref())?)
+    let meter_provider_builder = match config.exporter {
+        MetricsExporterKind::None => meter_provider_builder.with_reader(ManualReader::default()),
+        MetricsExporterKind::Stdout => meter_provider_builder.with_reader(stdout_metric_reader()),
+        MetricsExporterKind::Otlp => {
+            meter_provider_builder.with_reader(otlp_metric_reader(config.endpoint.as_ref())?)
         }
-        MetricsExporterConfig::Prometheus => {
+        MetricsExporterKind::Prometheus => {
             meter_provider_builder.with_reader(prometheus_metric_reader()?)
         }
     };

crates/config/src/sections/mod.rs

@@ -46,8 +46,8 @@ pub use self::{
     policy::PolicyConfig,
     secrets::SecretsConfig,
     telemetry::{
-        MetricsConfig, MetricsExporterConfig, Propagator, TelemetryConfig, TracingConfig,
-        TracingExporterConfig,
+        MetricsConfig, MetricsExporterKind, Propagator, TelemetryConfig, TracingConfig,
+        TracingExporterKind,
     },
     templates::TemplatesConfig,
     upstream_oauth2::{

crates/config/src/sections/telemetry.rs

@@ -35,83 +35,74 @@ pub enum Propagator {
     Jaeger,
 }
 
-fn otlp_endpoint_example() -> &'static str {
-    "https://localhost:4318"
+#[allow(clippy::unnecessary_wraps)]
+fn otlp_endpoint_default() -> Option<String> {
+    Some("https://localhost:4318".to_owned())
 }
 
 /// Exporter to use when exporting traces
 #[skip_serializing_none]
-#[derive(Clone, Debug, Serialize, Deserialize, JsonSchema)]
-#[serde(tag = "exporter", rename_all = "lowercase", deny_unknown_fields)]
-pub enum TracingExporterConfig {
+#[derive(Clone, Copy, Debug, Serialize, Deserialize, JsonSchema, Default)]
+#[serde(rename_all = "lowercase")]
+pub enum TracingExporterKind {
     /// Don't export traces
+    #[default]
     None,
 
     /// Export traces to the standard output. Only useful for debugging
     Stdout,
 
     /// Export traces to an OpenTelemetry protocol compatible endpoint
-    Otlp {
-        /// OTLP compatible endpoint
-        #[schemars(url, example = "otlp_endpoint_example")]
-        #[serde(default)]
-        endpoint: Option<Url>,
-    },
-}
-
-impl Default for TracingExporterConfig {
-    fn default() -> Self {
-        Self::None
-    }
+    Otlp,
 }
 
 /// Configuration related to exporting traces
 #[derive(Clone, Debug, Default, Serialize, Deserialize, JsonSchema)]
 pub struct TracingConfig {
     /// Exporter to use when exporting traces
-    #[serde(default, flatten)]
-    pub exporter: TracingExporterConfig,
+    #[serde(default)]
+    pub exporter: TracingExporterKind,
+
+    /// OTLP exporter: OTLP over HTTP compatible endpoint
+    #[serde(skip_serializing_if = "Option::is_none")]
+    #[schemars(url, default = "otlp_endpoint_default")]
+    pub endpoint: Option<Url>,
 
     /// List of propagation formats to use for incoming and outgoing requests
     pub propagators: Vec<Propagator>,
 }
 
 /// Exporter to use when exporting metrics
 #[skip_serializing_none]
-#[derive(Clone, Debug, Serialize, Deserialize, JsonSchema)]
-#[serde(tag = "exporter", rename_all = "lowercase")]
-pub enum MetricsExporterConfig {
+#[derive(Clone, Copy, Debug, Serialize, Deserialize, JsonSchema, Default)]
+#[serde(rename_all = "lowercase")]
+pub enum MetricsExporterKind {
     /// Don't export metrics
+    #[default]
     None,
 
     /// Export metrics to stdout. Only useful for debugging
     Stdout,
 
     /// Export metrics to an OpenTelemetry protocol compatible endpoint
-    Otlp {
-        /// OTLP compatible endpoint
-        #[schemars(url, example = "otlp_endpoint_example")]
-        #[serde(default)]
-        endpoint: Option<Url>,
-    },
+    Otlp,
 
     /// Export metrics via Prometheus. An HTTP listener with the `prometheus`
     /// resource must be setup to expose the Promethes metrics.
     Prometheus,
 }
 
-impl Default for MetricsExporterConfig {
-    fn default() -> Self {
-        Self::None
-    }
-}
-
 /// Configuration related to exporting metrics
 #[derive(Clone, Debug, Default, Serialize, Deserialize, JsonSchema)]
 pub struct MetricsConfig {
     /// Exporter to use when exporting metrics
-    #[serde(default, flatten)]
-    pub exporter: MetricsExporterConfig,
+    #[serde(default)]
+    pub exporter: MetricsExporterKind,
+
+    /// OTLP exporter: OTLP over HTTP compatible endpoint
+    #[serde(skip_serializing_if = "Option::is_none")]
+    #[schemars(url, default = "otlp_endpoint_default")]
+    pub endpoint: Option<Url>,
 }
 
 fn sentry_dsn_example() -> &'static str {
@@ -123,7 +114,7 @@ fn sentry_dsn_example() -> &'static str {
 pub struct SentryConfig {
     /// Sentry DSN
     #[schemars(url, example = "sentry_dsn_example")]
-    #[serde(default)]
+    #[serde(skip_serializing_if = "Option::is_none")]
     pub dsn: Option<String>,
 }