喜欢的话别忘了点赞、收藏加关注哦(加关注即可阅读全文),对接下来的教程有兴趣的可以关注专栏。谢谢喵!(=・ω・=)
16.3.1. 使用共享来实现并发
还记得Go语言有一句名言是这么说的:Do not communicate by sharing memory; instead, share memory by communicating.(不要用共享内存来通信,要用通信来共享内存)
上一篇文章就是使用通信的方式来实现并发的。这一篇文章讲一下如何使用共享内存的方式来实现并发。Go语言不建议使用这种方式,Rust支持通过共享状态来实现并发。
上一篇文章讲的Channel
类似单所有权:一旦值的所有权转移至Channel
,就无法使用它了。共享内存并发类似于多所有权:多个线程可以同时访问同一块内存。
16.3.2. 使用Mutex
来只允许一个线程来访问数据
Mutex
是mutual exclusion(互斥锁)的简写。
在同一时刻,Mutex
只允许一个线程来访问某些数据。
想要访问数据,线程必须首先获取互斥锁(lock),在Rust里就是调用lock
方法获得。lock
数据结构是Mutex
的一部分,它能跟踪谁对数据拥有独占访问权。Mutex
通常被描述为:通过锁定系统来保护它所持有的数据。
16.3.3. Mutex
的两条规则
- 在使用数据之前,必须尝试获取锁(lock)。
- 使用完
Mutex
所保护的数据,必须对数据进行解锁,以便其他线程可以获取锁。
16.3.4. Mutex<T>
的API
通过Mutex::new
函数来创建Mutex<T>
,其参数就是要保护的数据。Mutex<T>
实际上是一个智能指针。
在访问数据前,通过lock
方法来获取锁,这个方法会阻塞当前线程的运行。lock
方法也可能会失败,所以返回的值被Result
包裹,如果成功其值,Ok
变体附带的值的类型就为MutexGuard
(智能指针,实现了Deref
和Drop
)。
看个例子:
use std::sync::Mutex;fn main() {let m = Mutex::new(5);{let mut num = m.lock().unwrap();*num = 6;}println!("m = {m:?}");
}
- 使用
Mutex::new
创建了一个互斥锁,其保护的数据是5,赋给m
。所以m
的类型是MutexGuard<i32>
。 - 后面使用
{}
创建了新的小作用域,在小作用域里使用lock
方法获取值,使用unwrap
进行错误处理。由于MutexGuard
实现了Deref
trait,我们就可以获得内部数据的引用。所以num
是一个可变引用。 - 在小作用域内还使用了解引用
*
来修改数据的值为6。 - 由于
MutexGuard
实现了Drop
trait,所以在小作用域结束后会自动解锁。 - 最后打印了修改后的互斥锁内的内容。
输出:
m = Mutex { data: 6 }
16.3.5. 多线程共享Mutex<T>
看个例子:
use std::sync::Mutex;
use std::thread;fn main() {let counter = Mutex::new(0);let mut handles = vec![];for _ in 0..10 {let handle = thread::spawn(move || {let mut num = counter.lock().unwrap();*num += 1;});handles.push(handle);}for handle in handles {handle.join().unwrap();}println!("Result: {}", *counter.lock().unwrap());
}
counter
实际上就是一个计数器,只是使用了Mutex
包裹以更好地在多线程中调用,刚开始的值是0handle
目前是一个空Vector
- 下面通过从0到10(不包括10)的循环创建了10个线程,把每个线程得到的
handle
放到空集合handles
里。 - 在线程的闭包里,我们的意图是把
counter
这个互斥锁转移到闭包里(所以使用了move
关键字),然后获取互斥锁,然后修改它的值,每个线程都加1。当线程执行完后,num
会离开作用域,互斥锁被释放,其他线程就可以使用了。 - 从0到10(不包括10)的循环里还遍历了
handles
,使用join
方法,这样等每个handle
所对应的线程都结束后才会继续执行。 - 最后在主线程里尝试获得
counter
的互斥锁,然后把它打印出来。
输出:
$ cargo runCompiling shared-state v0.1.0 (file:///projects/shared-state)
error[E0382]: borrow of moved value: `counter`--> src/main.rs:21:29|
5 | let counter = Mutex::new(0);| ------- move occurs because `counter` has type `Mutex<i32>`, which does not implement the `Copy` trait
...
8 | for _ in 0..10 {| -------------- inside of this loop
9 | let handle = thread::spawn(move || {| ------- value moved into closure here, in previous iteration of loop
...
21 | println!("Result: {}", *counter.lock().unwrap());| ^^^^^^^ value borrowed here after move|
help: consider moving the expression out of the loop so it is only moved once|
8 ~ let mut value = counter.lock();
9 ~ for _ in 0..10 {
10 | let handle = thread::spawn(move || {
11 ~ let mut num = value.unwrap();|For more information about this error, try `rustc --explain E0382`.
error: could not compile `shared-state` (bin "shared-state") due to 1 previous error
错误是在前一次循环中已经把所有权移到前一次的那个线程里了,而这一次循环就没发再获得所有权了。
那么如何把counter
放到多个线程,也就是让多个线程拥有它的所有权呢?
16.3.6. 多线程的多重所有权
在15章讲了一个多重所有权的智能指针叫Rc<T>
,把counter
用Rc
包裹即可:
let counter = Rc::new(Mutex::new(0));
在循环里,需要把克隆传进线程,这里用了类型遮蔽把新counter
值设为旧counter
的引用:
let counter = Rc::clone(&counter);
修改后的代码(记得在使用前引入Rc
):
use std::rc::Rc;
use std::sync::Mutex;
use std::thread;fn main() {let counter = Rc::new(Mutex::new(0));let mut handles = vec![];for _ in 0..10 {let counter = Rc::clone(&counter);let handle = thread::spawn(move || {let mut num = counter.lock().unwrap();*num += 1;});handles.push(handle);}for handle in handles {handle.join().unwrap();}println!("Result: {}", *counter.lock().unwrap());
}
输出:
error[E0277]: `Rc<Mutex<i32>>` cannot be sent between threads safely--> src/main.rs:11:36|
11 | let handle = thread::spawn(move || {| ------------- ^------| | || ______________________|_____________within this `{closure@src/main.rs:11:36: 11:43}`| | || | required by a bound introduced by this call
12 | | let mut num = counter.lock().unwrap();
13 | |
14 | | *num += 1;
15 | | });| |_________^ `Rc<Mutex<i32>>` cannot be sent between threads safely|= help: within `{closure@src/main.rs:11:36: 11:43}`, the trait `Send` is not implemented for `Rc<Mutex<i32>>`, which is required by `{closure@src/main.rs:11:36: 11:43}: Send`
note: required because it's used within this closure--> src/main.rs:11:36|
11 | let handle = thread::spawn(move || {| ^^^^^^^
note: required by a bound in `spawn`--> /rustc/eeb90cda1969383f56a2637cbd3037bdf598841c/library/std/src/thread/mod.rs:688:1For more information about this error, try `rustc --explain E0277`.
error: could not compile `shared-state` (bin "shared-state") due to 1 previous error
看报错信息的这部分:`Rc<Mutex<i32>>` cannot be sent between threads safely
,Rc<Mutex<i32>>
不能在线程间安全地传递。编译器也告诉我们了原因:the trait `Send` is not implemented for `Rc<Mutex<i32>>`
,Rc<Mutex<i32>>
没有实现send
trait(下一篇文章会讲到)。只有实现send
的类型才能在线程间安全地传递。
其实在第15章讲Rc<T>
也说到了它不能用于多线程场景:Rc<T>
不能安全地跨线程共享。它不能确保计数的更改不会被另一个线程中断。这可能会导致错误的计数,进而导致内存泄漏或在我们完成之前删除某个值。我们需要的是一种与Rc<T>
完全相同的类型,但它以线程安全的方式更改引用计数。
那么多线程应该用什么呢?有一个智能指针叫做Arc<T>
可以胜任这个场景。
16.3.7. 使用Arc<T>
来进行原子引用计数
Arc<T>
和Rc<T>
类似,但是它可以用于并发场景。Arc
的A指的是Atomic(原子的),这意味着它是一个原子引用计数类型,原子是另一种并发原语。这里不对Arc<T>
做过于详细的介绍,只需要知道原子像原始类型一样工作,但可以安全地跨线程共享,其余信息详见Rust官方文档。
那么为什么所有的基础类型都不是原子的?为什么标准库不默认使用Arc<T>
?这是因为:
Arc<T>
的功能需要以性能作为代价Arc<T>
和Rc<T>
的API都是相同的
既然Arc<T>
和Rc<T>
的API都是相同的,那么先前的代码就很好改了(记得在使用前引入Arc
):
use std::sync::{Arc, Mutex};
use std::thread;fn main() {let counter = Arc::new(Mutex::new(0));let mut handles = vec![];for _ in 0..10 {let counter = Arc::clone(&counter);let handle = thread::spawn(move || {let mut num = counter.lock().unwrap();*num += 1;});handles.push(handle);}for handle in handles {handle.join().unwrap();}println!("Result: {}", *counter.lock().unwrap());
}
16.3.8. RefCell<T>
/Rc<T>
vs. Mutex<T>
/Arc<T>
Mutex<T>
提供了内部可变性,和Cell
家族一样。我们一般使用RefCell<T>
包裹Rc<T>
以获得一个有内部可变性的共享所有权数据类型。同样的,使用Mutex<T>
可以改变Arc<T>
里面的内容。
当使用Mutex<T>
时,Rust 无法保护您免受各种逻辑错误的影响。使用Rc<T>
会带来创建引用循环的风险,其中两个Rc<T>
值相互引用,从而导致内存泄漏。同样, Mutex<T>
也存在产生死锁(deadlock) 的风险。当一个操作需要锁定两个资源并且两个线程各自获取其中一个锁,导致它们永远等待对方时,就会发生这种情况。Mutex<T>
和MutexGuard
的标准库API文档提供了有用的信息。详见:Mutex<T>
API文档和MutexGuard
API文档。