Rust 多线程并发编程

使用一个进程可以运行多个线程的机制叫做多线程编程，Rust 语言支持多线程并发编程。

一个进程一定有一个主线程，主线程之外创建出来的线程称之为 子线程。

多线程编程，其实就是在主线程之外创建子线程，让子线程和主线程同时运行，完成各自的任务。

1. 创建线程

Rust 语言标准库中的 std::thread 模块用于支持多线程编程。

std::thread 提供很很多方法用于创建线程、管理线程和结束线程。

创建一个新线程，可以使用 std::thread::spawn() 方法。

spawn() 函数的原型如下

pub fn spawn<F, T>(f: F) -> JoinHandle<T> 

参数 f 是一个闭包（closure ） 是线程要执行的代码。

范例

下面的范例，我们使用 spawn() 函数创建了一个线程，用于输出数字 1 到 10

use std::thread;           // 导入线程模块
use std::time::Duration;   // 导入时间模块

fn main() {
   //创建一个新线程
   thread::spawn(|| {
      for i in 1..10 {
         println!("hi number {} from the spawned thread!", i);
         thread::sleep(Duration::from_millis(1));
      }
   });
   // 主线程要执行的代码
   for i in 1..5 {
      println!("hi number {} from the main thread!", i);
      thread::sleep(Duration::from_millis(1));
   }
}

编译运行以上 Rust 范例，输出结果如下

hi number 1 from the main thread!
hi number 1 from the spawned thread!
hi number 2 from the main thread!
hi number 2 from the spawned thread!
hi number 3 from the main thread!
hi number 3 from the spawned thread!
hi number 4 from the spawned thread!
hi number 4 from the main thread!

注意

当主线程执行结束，那么就会自动关闭创建出来的衍生子线程。

上面的代码，我们调用 thread::sleep() 函数强制线程休眠一段时间，这就允许不同的线程交替执行。

虽然某个线程休眠时会自动让出 cpu，但并不保证其它线程会执行。这取决于操作系统如何调度线程。

这个范例的输出结果是随机的，主线程一旦执行完成程序就会自动退出，不会继续等待子线程。这就是子线程的输出结果为什么不全的原因。

2. 加入线程句柄 join()

默认情况下，主线程并不会等待子线程执行完毕。如果主线程创建完衍生线程就立即退出的话，那么子线程可能根本没机会开始运行或者执行完毕，

为了避免这种情况，我们可以让主线程等待子线程执行完毕然后再继续执行。

Rust 标准库提供了 join() 方法用于把子线程加入主线程等待队列。

join() 方法的原型如下

spawn<F, T>(f: F) -> JoinHandle<T>

范例

下面的范例，我们使用 join() 方法把衍生线程加入主线程等待队列，这样主线程会等待子线程执行完毕才能退出。

use std::thread;
use std::time::Duration;

fn main() {
   let handle = thread::spawn(|| {
      for i in 1..10 {
         println!("hi number {} from the spawned thread!", i);
         thread::sleep(Duration::from_millis(1));
      }
   });
   for i in 1..5 {
      println!("hi number {} from the main thread!", i);
      thread::sleep(Duration::from_millis(1));
   }
   handle.join().unwrap();
}

编译运行以上 Rust 范例，输出结果如下

hi number 1 from the main thread!
hi number 1 from the spawned thread!
hi number 2 from the spawned thread!
hi number 2 from the main thread!
hi number 3 from the spawned thread!
hi number 3 from the main thread!
hi number 4 from the main thread!
hi number 4 from the spawned thread!
hi number 5 from the spawned thread!
hi number 6 from the spawned thread!
hi number 7 from the spawned thread!
hi number 8 from the spawned thread!
hi number 9 from the spawned thread!

从输出结果来看，主线程和子线程交替执行。

注意

主线程会等待子线程执行完毕是因为调用了 join() 方法。