imp(examples): update multi-threading examples

Author: omarabid

examples/multi-thread.rs

@@ -1,6 +1,6 @@
 extern crate pyroscope;
 
-use pyroscope::{PyroscopeAgent, Result};
+use pyroscope::{backend::Tag, PyroscopeAgent, Result};
 use pyroscope_pprofrs::{pprof_backend, PprofConfig};
 use std::{
     collections::hash_map::DefaultHasher,
@@ -36,6 +36,34 @@ fn hash_rounds2(n: u64) -> u64 {
     n
 }
 
+fn extra_rounds1(n: u64) -> u64 {
+    let hash_str = "Some string to hash";
+    let mut default_hasher = DefaultHasher::new();
+
+    for _ in 0..n {
+        for _ in 0..1000 {
+            default_hasher.write(hash_str.as_bytes());
+        }
+        hash_str.hash(&mut default_hasher);
+    }
+
+    n
+}
+
+fn extra_rounds2(n: u64) -> u64 {
+    let hash_str = "Some string to hash";
+    let mut default_hasher = DefaultHasher::new();
+
+    for _ in 0..n {
+        for _ in 0..1000 {
+            default_hasher.write(hash_str.as_bytes());
+        }
+        hash_str.hash(&mut default_hasher);
+    }
+
+    n
+}
+
 fn main() -> Result<()> {
     let mut agent = PyroscopeAgent::builder("http://localhost:4040", "example.multithread")
         .tags([("Host", "Rust")].to_vec())
@@ -52,15 +80,25 @@ fn main() -> Result<()> {
     // Start Agent
     agent.start()?;
 
+    let (add_tag, remove_tag) = agent.tag_wrapper();
+
     let handle_1 = thread::Builder::new()
         .name("thread-1".to_string())
-        .spawn(|| {
+        .spawn(move || {
             hash_rounds1(300_000);
+            add_tag("extra".to_string(), "round-1".to_string()).unwrap();
+            extra_rounds1(200_000);
+            remove_tag("extra".to_string(), "round-1".to_string()).unwrap();
         })?;
 
+    let (add_tag, remove_tag) = agent.tag_wrapper();
+
     let handle_2 = thread::Builder::new()
         .name("thread-2".to_string())
-        .spawn(|| {
+        .spawn(move || {
+            add_tag("extra".to_string(), "round-2".to_string()).unwrap();
+            extra_rounds2(100_000);
+            remove_tag("extra".to_string(), "round-2".to_string()).unwrap();
             hash_rounds2(500_000);
         })?;