并发

concurrency.md
commit 23a7a7bdb6a6a43cd7efdd9176b1d3f75d9d0e70

并发与并行是计算机科学中相当重要的两个主题,并且在当今生产环境中也十分热门。计算机正拥有越来越多的核心,然而很多程序员还没有准备好去完全的利用它们。

Rust 的内存安全功能也适用于并发环境。甚至并发的 Rust 程序也会是内存安全的,并且没有数据竞争。Rust 的类型系统也能胜任,并且在编译时能提供你强大的方式去推论并发代码。

在我们讨论 Rust 提供的并发功能之前,理解一些问题是很重要的:Rust 非常底层以至于所有这些都是由标准库,而不是由语言提供的。这意味着如果你在某些方面不喜欢 Rust 处理并发的方式,你可以自己实现一个。mio是关于这个原则实践的一个实际的例子。

背景:SendSync

并发难以推理。在 Rust 中,我们有一个强大、静态类型系统来帮助我们推理我们的代码。Rust 自身提供了两个特性来帮助我们理解可能是并发的代码的意思。

Send

第一个我们要谈到的特性是Send。当一个T类型实现了Send,它向编译器指示这个类型的所有权可以在线程间安全的转移。

强制实施一些通用的限制是很重要的。例如,我们有一个连接两个线程的通道,我们想要能够向通道发送些数据到另一个线程。因此,我们要确保这个类型实现了Send

相反的,如果我们通过 FFI 封装了一个不是线程安全的库,我们并不想实现Send,那么编译器会帮助我们强制确保它不会离开当前线程。

Sync

第二个特性是Sync。当一个类型T实现了Sync,它向编译器指示这个类型在多线程并发时没有导致内存不安全的可能性。这隐含了没有内部可变性的类型天生是Sync的,这包含了基本类型(如 u8)和包含他们的聚合类型。

为了在线程间共享引用,Rust 提供了一个叫做Arc<T>的 wrapper 类型。Arc<T>实现了SendSync当且仅当T实现了SendSync。例如,一个Arc<RefCell<U>>类型的对象不能在线程间传送因为RefCell并没有实现Sync,因此Arc<RefCell<U>>并不会实现Send

这两个特性允许你使用类型系统来确保你代码在并发环境的特性。在我们演示为什么之前,我们需要先学会如何创建一个并发 Rust 程序!

线程

Rust标准库提供了一个“线程”库,它允许你并行的执行 Rust 代码。这是一个使用std::thread的基本例子:

  1. use std::thread;
  2. fn main() {
  3. thread::spawn(|| {
  4. println!("Hello from a thread!");
  5. });
  6. }

thread::spawn()方法接受一个闭包,它将会在一个新线程中执行。它返回一线程的句柄,这个句柄可以用来等待子线程结束并提取它的结果:

  1. use std::thread;
  2. fn main() {
  3. let handle = thread::spawn(|| {
  4. "Hello from a thread!"
  5. });
  6. println!("{}", handle.join().unwrap());
  7. }

正如闭包可以从它的环境中获取变量,我们也可以把一些数据带到其他线程中:

  1. use std::thread;
  2. fn main() {
  3. let x = 1;
  4. thread::spawn(|| {
  5. println!("x is {}", x);
  6. });
  7. }

然而,这会给我们一个错误:

  1. 5:19: 7:6 error: closure may outlive the current function, but it
  2. borrows `x`, which is owned by the current function
  3. ...
  4. 5:19: 7:6 help: to force the closure to take ownership of `x` (and any other referenced variables),
  5. use the `move` keyword, as shown:
  6. thread::spawn(move || {
  7. println!("x is {}", x);
  8. });

这是因为默认闭包通过引用获取变量,因此闭包只会获取一个x的引用。这样有一个问题,因为线程可能会存在超过x的作用域,导致产生一个悬垂指针。

为了解决这个问题,我们使用错误中提示的move闭包。move闭包将在这里详细讲解。从基本上讲,它把变量从环境中移动到自身。这意味着x现在归闭包所有,并不能在调用spawn()后的main()中使用。

  1. use std::thread;
  2. fn main() {
  3. let x = 1;
  4. thread::spawn(move || {
  5. println!("x is {}", x);
  6. });
  7. }

很多语言有执行多线程的能力,不过是很不安全的。有完整的书籍是关于如何避免在共享可变状态下出现错误的。在此,借助类型系统,Rust也通过在编译时避免数据竞争来帮助我们。让我们具体讨论下如何在线程间共享数据。

安全共享的可变状态(Safe Shared Mutable State)

根据 Rust的 类型系统,我们有个听起来类似谎言的概念叫做:“安全共享的可变状态”。很多程序员都同意共享可变状态是非常,非常不好的。

有人曾说道:

共享可变状态是一切罪恶的根源。大部分语言尝试解决这个问题的“可变”部分,而Rust则尝试解决“共享”部分。

同样所有权系统也通过防止不当的使用指针来帮助我们排除数据竞争,最糟糕的并发bug之一。

作为一个例子,这是一个在很多语言中可能会产生数据竞争的 Rust 版本程序。它不能编译:

  1. use std::thread;
  2. use std::time::Duration;
  3. fn main() {
  4. let mut data = vec![1, 2, 3];
  5. for i in 0..3 {
  6. thread::spawn(move || {
  7. data[0] += i;
  8. });
  9. }
  10. thread::sleep(Duration::from_millis(50));
  11. }

这会给我们一个错误:

  1. 8:17 error: capture of moved value: `data`
  2. data[0] += i;
  3. ^~~~

Rust 知道这并不是安全的!如果每个线程中都有一个data的引用,并且这些线程获取了引用的所有权,我们就有了 3 个所有者!data在第一次调用spawn()时被移出了main,所以循环中接下来的调用不能使用这个变量。

所以,我们需要一些类型可以让我们拥有一个值的多个有所有权的引用。通常,我们使用Rc<T>,它是一个引用计数类型用以提供共享的所有权。它有一些运行时记录来跟踪引用它的数量,也就是“引用计数”。

调用Rc<T>clone()方法会返回一个有所有权的引用并增加其内部引用计数。我们为每一个线程创建一个:

  1. use std::thread;
  2. use std::time::Duration;
  3. use std::rc::Rc;
  4. fn main() {
  5. let mut data = Rc::new(vec![1, 2, 3]);
  6. for i in 0..3 {
  7. // Create a new owned reference:
  8. let data_ref = data.clone();
  9. // Use it in a thread:
  10. thread::spawn(move || {
  11. data_ref[0] += i;
  12. });
  13. }
  14. thread::sleep(Duration::from_millis(50));
  15. }

这并不能运行,不过它会给我们这个错误:

  1. 13:9: 13:22 error: the trait bound `alloc::rc::Rc<collections::vec::Vec<i32>> : core::marker::Send`
  2. is not satisfied
  3. ...
  4. 13:9: 13:22 note: `alloc::rc::Rc<collections::vec::Vec<i32>>`
  5. cannot be sent between threads safely

如错误中提到的,Rc并不能在线程间安全的传递。这是因为其内部的引用计数并不是通过一个线程安全的方式维护的(非原子性操作)并可能产生数据竞争。

为了解决这个问题,我们使用Arc<T>,Rust 标准的原子引用计数类型。

Arc<T>的原子部分可以在多线程中安全的访问。为此编译器确保了内部计数的改变都是不可分割的操作这样就不会产生数据竞争。

本质上,Arc<T>是一个可以让我们在线程间安全的共享所有权的类型。

  1. use std::thread;
  2. use std::sync::Arc;
  3. use std::time::Duration;
  4. fn main() {
  5. let mut data = Arc::new(vec![1, 2, 3]);
  6. for i in 0..3 {
  7. let data = data.clone();
  8. thread::spawn(move || {
  9. data[0] += i;
  10. });
  11. }
  12. thread::sleep(Duration::from_millis(50));
  13. }

与之前类似,我们使用clone()来创建一个新的有所有权的句柄。接着这个句柄被移动到了新线程。

不过。。。仍然出错了。

  1. <anon>:11:24 error: cannot borrow immutable borrowed content as mutable
  2. <anon>:11 data[0] += i;
  3. ^~~~

Arc<T>默认是不可变的。它允许在线程间共享数据,不过可变的共享数据是不安全的并且在涉及到多线程时会造成数据竞争!

通常当我们希望让某个不可变的东西变成可变时,我们使用Cell<T>或者RefCell<T>,它们通过运行时检查或其他手段提供安全的可变性。然而,与Rc类似,它们不是线程安全的。如果我们尝试使用它们,我们将会得到一个错误说这些类型不能被Sync,代码会编译失败。

看起来我们需要一些允许我们安全的在线程间改变共享值的类型,例如同一时刻只允许一个线程能够改变它内部值的类型。

为此,我们可以使用Mutex<T>类型!

下面是一个可以工作的版本:

  1. use std::sync::{Arc, Mutex};
  2. use std::thread;
  3. use std::time::Duration;
  4. fn main() {
  5. let data = Arc::new(Mutex::new(vec![1, 2, 3]));
  6. for i in 0..3 {
  7. let data = data.clone();
  8. thread::spawn(move || {
  9. let mut data = data.lock().unwrap();
  10. data[0] += i;
  11. });
  12. }
  13. thread::sleep(Duration::from_millis(50));
  14. }

注意i的值被限制(拷贝)到了闭包里并不是在线程间共享。

这里我们“锁定”了互斥锁(mutex)。一个互斥锁,正如其名,同时只允许一个线程访问一个值。当我们想要访问一个值时,我们lock()它。这会“锁定” mutex,并且其他线程不能锁定它(也就是改变它的值),直到我们处理完之后。如果一个线程尝试锁定一个已经被锁定的 mutex,它将会等待直到其他线程释放这个锁为止。

这里锁的“释放”是隐式的;当锁的结果(在这里是data)离开作用域,锁将会被自动释放。

注意Mutexlock方法有如下签名:

  1. fn lock(&self) -> LockResult<MutexGuard<T>>

并且因为MutexGuard<T>并没有实现Send,guard 并不能跨线程,确保了线程局部性锁的获取和释放。

让我们更仔细的检查一个线程代码:

  1. # use std::sync::{Arc, Mutex};
  2. # use std::thread;
  3. # use std::time::Duration;
  4. # fn main() {
  5. # let data = Arc::new(Mutex::new(vec![1, 2, 3]));
  6. # for i in 0..3 {
  7. # let data = data.clone();
  8. thread::spawn(move || {
  9. let mut data = data.lock().unwrap();
  10. data[0] += i;
  11. });
  12. # }
  13. # thread::sleep(Duration::from_millis(50));
  14. # }

首先,我们调用lock(),它获取了互斥锁。因为这可能失败,它返回一个Result<T, E>,并且因为这仅仅是一个例子,我们unwrap()结果来获得一个数据的引用。现实中的代码在这里应该有更健壮的错误处理。下面我们可以随意修改它,因为我们持有锁。

最后,在线程运行的同时,我们等待在一个较短的定时器上。不过这并不理想:我们可能选择等待了一个合理的时间不过它更可能比所需的时间要久或并不足够长,这依赖程序运行时线程完成它的计算所需的时间。

一个比定时器更精确的替代是使用一个 Rust 标准库提供的用来同步各个线程的机制。让我们聊聊其中一个:通道。

通道(Channels)

下面是我们代码使用通道同步的版本,而不是等待特定时间:

  1. use std::sync::{Arc, Mutex};
  2. use std::thread;
  3. use std::sync::mpsc;
  4. fn main() {
  5. let data = Arc::new(Mutex::new(0));
  6. // `tx` is the "transmitter" or "sender".
  7. // `rx` is the "receiver".
  8. let (tx, rx) = mpsc::channel();
  9. for _ in 0..10 {
  10. let (data, tx) = (data.clone(), tx.clone());
  11. thread::spawn(move || {
  12. let mut data = data.lock().unwrap();
  13. *data += 1;
  14. tx.send(()).unwrap();
  15. });
  16. }
  17. for _ in 0..10 {
  18. rx.recv().unwrap();
  19. }
  20. }

我们使用mpsc::channel()方法创建了一个新的通道。我们仅仅向通道中send了一个简单的(),然后等待它们 10 个都返回。

因为这个通道只是发送了一个通用信号,我们也可以通过通道发送任何实现了Send的数据!

  1. use std::thread;
  2. use std::sync::mpsc;
  3. fn main() {
  4. let (tx, rx) = mpsc::channel();
  5. for i in 0..10 {
  6. let tx = tx.clone();
  7. thread::spawn(move || {
  8. let answer = i * i;
  9. tx.send(answer).unwrap();
  10. });
  11. }
  12. for _ in 0..10 {
  13. println!("{}", rx.recv().unwrap());
  14. }
  15. }

这里我们创建了 10 个线程,分别计算一个数字的平方(spawn()时的i),接着通过通道把结果send()回主线程。

恐慌(Panics)

panic!会使当前执行线程崩溃。你可以使用 Rust 的线程来作为一个简单的隔离机制:

  1. use std::thread;
  2. let handle = thread::spawn(move || {
  3. panic!("oops!");
  4. });
  5. let result = handle.join();
  6. assert!(result.is_err());

我们的Thread返回一个Result,它允许我们检查我们的线程是否发生了恐慌。