Merge pull request #21 from logdna/extend-lineMetaMut-trait-with-addi…

…tional-accessors

Extend line meta mut trait with additional accessors

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "logdna-client"
-version = "0.6.0"
+version = "0.7.0"
 authors = ["[email protected]"]
 edition = "2018"
 license = "MIT"

src/body.rs

@@ -183,6 +183,14 @@ pub trait LineMetaMut: LineMeta {
     fn set_labels(&mut self, labels: KeyValueMap) -> Result<(), LineMetaError>;
     fn set_level(&mut self, level: String) -> Result<(), LineMetaError>;
     fn set_meta(&mut self, meta: Value) -> Result<(), LineMetaError>;
+    fn get_annotations_mut(&mut self) -> &mut Option<KeyValueMap>;
+    fn get_app_mut(&mut self) -> &mut Option<String>;
+    fn get_env_mut(&mut self) -> &mut Option<String>;
+    fn get_file_mut(&mut self) -> &mut Option<String>;
+    fn get_host_mut(&mut self) -> &mut Option<String>;
+    fn get_labels_mut(&mut self) -> &mut Option<KeyValueMap>;
+    fn get_level_mut(&mut self) -> &mut Option<String>;
+    fn get_meta_mut(&mut self) -> &mut Option<Value>;
 }
 
 /// Represents a line that provides accessor to the underlying data buffer for manipulation.
@@ -589,6 +597,30 @@ impl LineMetaMut for LineBuilder {
         self.meta = Some(meta);
         Ok(())
     }
+    fn get_annotations_mut(&mut self) -> &mut Option<KeyValueMap> {
+        &mut self.annotations
+    }
+    fn get_app_mut(&mut self) -> &mut Option<String> {
+        &mut self.app
+    }
+    fn get_env_mut(&mut self) -> &mut Option<String> {
+        &mut self.env
+    }
+    fn get_file_mut(&mut self) -> &mut Option<String> {
+        &mut self.file
+    }
+    fn get_host_mut(&mut self) -> &mut Option<String> {
+        &mut self.host
+    }
+    fn get_labels_mut(&mut self) -> &mut Option<KeyValueMap> {
+        &mut self.labels
+    }
+    fn get_level_mut(&mut self) -> &mut Option<String> {
+        &mut self.level
+    }
+    fn get_meta_mut(&mut self) -> &mut Option<Value> {
+        &mut self.meta
+    }
 }
 
 impl LineBufferMut for LineBuilder {
@@ -639,6 +671,30 @@ impl LineMetaMut for &mut LineBuilder {
         self.meta = Some(meta);
         Ok(())
     }
+    fn get_annotations_mut(&mut self) -> &mut Option<KeyValueMap> {
+        &mut self.annotations
+    }
+    fn get_app_mut(&mut self) -> &mut Option<String> {
+        &mut self.app
+    }
+    fn get_env_mut(&mut self) -> &mut Option<String> {
+        &mut self.env
+    }
+    fn get_file_mut(&mut self) -> &mut Option<String> {
+        &mut self.file
+    }
+    fn get_host_mut(&mut self) -> &mut Option<String> {
+        &mut self.host
+    }
+    fn get_labels_mut(&mut self) -> &mut Option<KeyValueMap> {
+        &mut self.labels
+    }
+    fn get_level_mut(&mut self) -> &mut Option<String> {
+        &mut self.level
+    }
+    fn get_meta_mut(&mut self) -> &mut Option<Value> {
+        &mut self.meta
+    }
 }
 
 impl LineBufferMut for &mut LineBuilder {
@@ -761,11 +817,10 @@ impl Default for KeyValueMap {
         Self::new()
     }
 }
+
 impl From<BTreeMap<String, String>> for KeyValueMap {
     fn from(map: BTreeMap<String, String>) -> Self {
-        Self {
-            0: HashMap::from_iter(map),
-        }
+        Self(HashMap::from_iter(map))
     }
 }