wx, feat: first try on rust thread

Author: sword-demon

Cargo.lock

@@ -6,3 +6,5 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+anyhow = "1.0.86"
+rand = "0.8.5"

Cargo.toml

@@ -0,0 +1,66 @@
+use anyhow::{anyhow, Ok, Result};
+use std::{sync::mpsc, thread, time::Duration};
+
+const NUM_PRODUCERS: usize = 4;
+
+#[derive(Debug)]
+#[allow(dead_code)]
+struct Msg {
+    idx: usize,
+    value: usize,
+}
+
+fn main() -> Result<()> {
+    let (tx, rx) = mpsc::channel();
+
+    // 创建 producers
+    for i in 0..NUM_PRODUCERS {
+        let tx = tx.clone();
+        thread::spawn(move || producer(i, tx));
+    }
+
+    // 释放 tx 否则 rx 无法结束
+    drop(tx);
+
+    // 创建 consumer
+    let consumer = thread::spawn(move || {
+        for msg in rx {
+            println!("consumer {:?}", msg);
+        }
+        println!("consumer exit");
+
+        42
+    });
+
+    let secret = consumer
+        .join()
+        .map_err(|e| anyhow!("Thread join error: {:?}", e))?;
+
+    println!("secret = {}", secret);
+
+    Ok(())
+}
+
+fn producer(idx: usize, tx: mpsc::Sender<Msg>) -> Result<()> {
+    loop {
+        let value = rand::random::<usize>();
+        tx.send(Msg::new(idx, value))?;
+        let sleep_time = rand::random::<u8>() as u64 * 10;
+        thread::sleep(Duration::from_millis(sleep_time));
+
+        // random exit the producer
+        if rand::random::<u8>() % 5 == 0 {
+            println!("producer {} exit", idx);
+            break;
+        }
+    }
+
+    // more things to do
+    Ok(())
+}
+
+impl Msg {
+    fn new(idx: usize, value: usize) -> Self {
+        Self { idx, value }
+    }
+}