feat: add shutdown with timeout for metric exporter

Author: mohammadVatandoost

opentelemetry-otlp/src/metric.rs

@@ -21,6 +21,7 @@ use opentelemetry_sdk::metrics::{
     data::ResourceMetrics, exporter::PushMetricExporter, Temporality,
 };
 use std::fmt::{Debug, Formatter};
+use std::time::Duration;
 
 /// Target to which the exporter is going to send metrics, defaults to https://localhost:4317/v1/metrics.
 /// Learn about the relationship between this constant and default/spans/logs at
@@ -163,6 +164,10 @@ impl PushMetricExporter for MetricExporter {
     }
 
     fn shutdown(&self) -> OTelSdkResult {
+        self.shutdown_with_timeout(Duration::from_secs(5))
+    }
+
+    fn shutdown_with_timeout(&self, _timeout: std::time::Duration) -> OTelSdkResult {
         match &self.client {
             #[cfg(feature = "grpc-tonic")]
             SupportedTransportClient::Tonic(client) => client.shutdown(),

opentelemetry-sdk/src/metrics/exporter.rs

@@ -1,5 +1,7 @@
 //! Interfaces for exporting metrics
+
 use crate::error::OTelSdkResult;
+use std::time::Duration;
 
 use crate::metrics::data::ResourceMetrics;
 
@@ -26,6 +28,9 @@ pub trait PushMetricExporter: Send + Sync + 'static {
     ///
     /// After Shutdown is called, calls to Export will perform no operation and
     /// instead will return an error indicating the shutdown state.
+    fn shutdown_with_timeout(&self, timeout: Duration) -> OTelSdkResult;
+
+    /// Shutdown with the default timeout of 5 seconds.
     fn shutdown(&self) -> OTelSdkResult;
 
     /// Access the [Temporality] of the MetricExporter.

opentelemetry-sdk/src/metrics/in_memory_exporter.rs

@@ -7,6 +7,7 @@ use crate::InMemoryExporterError;
 use std::collections::VecDeque;
 use std::fmt;
 use std::sync::{Arc, Mutex};
+use std::time::Duration;
 
 /// An in-memory metrics exporter that stores metrics data in memory.
 ///
@@ -281,6 +282,10 @@ impl PushMetricExporter for InMemoryMetricExporter {
         Ok(())
     }
 
+    fn shutdown_with_timeout(&self, _timeout: Duration) -> OTelSdkResult {
+        Ok(())
+    }
+
     fn temporality(&self) -> Temporality {
         self.temporality
     }

opentelemetry-sdk/src/metrics/periodic_reader.rs

@@ -565,6 +565,10 @@ mod tests {
             Ok(())
         }
 
+        fn shutdown_with_timeout(&self, _timeout: Duration) -> OTelSdkResult {
+            Ok(())
+        }
+
         fn temporality(&self) -> Temporality {
             Temporality::Cumulative
         }
@@ -585,6 +589,10 @@ mod tests {
         }
 
         fn shutdown(&self) -> OTelSdkResult {
+            self.shutdown_with_timeout(Duration::from_secs(5))
+        }
+
+        fn shutdown_with_timeout(&self, _timeout: Duration) -> OTelSdkResult {
             self.is_shutdown.store(true, Ordering::Relaxed);
             Ok(())
         }

opentelemetry-stdout/src/metrics/exporter.rs

@@ -13,6 +13,7 @@ use opentelemetry_sdk::{
 };
 use std::fmt::Debug;
 use std::sync::atomic;
+use std::time::Duration;
 
 /// An OpenTelemetry exporter that writes to stdout on export.
 pub struct MetricExporter {
@@ -64,6 +65,10 @@ impl PushMetricExporter for MetricExporter {
     }
 
     fn shutdown(&self) -> OTelSdkResult {
+        self.shutdown_with_timeout(Duration::from_secs(5))
+    }
+
+    fn shutdown_with_timeout(&self, _timeout: Duration) -> OTelSdkResult {
         self.is_shutdown.store(true, atomic::Ordering::SeqCst);
         Ok(())
     }