From da81051446645d993966b955e007834a00717f64 Mon Sep 17 00:00:00 2001
From: andylokandy <andylokandy@hotmail.com>
Date: Thu, 19 Sep 2024 16:32:58 +0800
Subject: [PATCH] chore: add OpentelemetryLogBuilder::with_labels

---
 src/append/opentelemetry.rs | 27 ++++++++++++++++++++++++---
 src/layout/kv.rs            |  4 ++++
 2 files changed, 28 insertions(+), 3 deletions(-)

diff --git a/src/append/opentelemetry.rs b/src/append/opentelemetry.rs
index 7a84d4a..147d215 100644
--- a/src/append/opentelemetry.rs
+++ b/src/append/opentelemetry.rs
@@ -80,16 +80,37 @@ impl OpentelemetryLogBuilder {
     }
 
     /// Add a label to the resource.
-    pub fn add_label(mut self, key: impl Into<Cow<'static, str>>, value: impl Into<Cow<'static, str>>) -> Self {
+    pub fn with_label(
+        mut self,
+        key: impl Into<Cow<'static, str>>,
+        value: impl Into<Cow<'static, str>>,
+    ) -> Self {
         self.labels.push((key.into(), value.into()));
         self
     }
 
+    /// Add multiple labels to the resource.
+    pub fn with_labels<K, V>(mut self, labels: impl IntoIterator<Item = (K, V)>) -> Self
+    where
+        K: Into<Cow<'static, str>>,
+        V: Into<Cow<'static, str>>,
+    {
+        self.labels
+            .extend(labels.into_iter().map(|(k, v)| (k.into(), v.into())));
+        self
+    }
+
     /// Build the [`OpentelemetryLog`] appender.
     pub fn build(self) -> Result<OpentelemetryLog, opentelemetry::logs::LogError> {
-        let OpentelemetryLogBuilder { name, endpoint, protocol, labels } = self;
+        let OpentelemetryLogBuilder {
+            name,
+            endpoint,
+            protocol,
+            labels,
+        } = self;
 
-        let collector_timeout = Duration::from_secs(opentelemetry_otlp::OTEL_EXPORTER_OTLP_TIMEOUT_DEFAULT);
+        let collector_timeout =
+            Duration::from_secs(opentelemetry_otlp::OTEL_EXPORTER_OTLP_TIMEOUT_DEFAULT);
         let exporter = match protocol {
             Protocol::Grpc => opentelemetry_otlp::new_exporter()
                 .tonic()
diff --git a/src/layout/kv.rs b/src/layout/kv.rs
index 8f9685b..cc32a5c 100644
--- a/src/layout/kv.rs
+++ b/src/layout/kv.rs
@@ -13,6 +13,8 @@
 // limitations under the License.
 
 /// A helper struct to format log's key-value pairs.
+///
+/// This is useful when you want to display log's key-value pairs in a log message.
 pub struct KvDisplay<'kvs> {
     kv: &'kvs dyn log::kv::Source,
 }
@@ -47,6 +49,8 @@ impl<'a, 'kvs> log::kv::Visitor<'kvs> for KvWriter<'a, 'kvs> {
 }
 
 /// A helper to collect log's key-value pairs.
+/// 
+/// This is useful when you want to collect log's key-value pairs for further processing.
 pub fn collect_kvs(kv: &dyn log::kv::Source) -> Vec<(String, String)> {
     let mut collector = KvCollector { kv: Vec::new() };
     kv.visit(&mut collector).ok();