Rust那些事之并发Send与Sync

Rust那些事之并发Send与Sync

Send与Sync在Rust中属于marker trait，代码位于marker.rs，在标记模块中还有Copy、Unpin等trait。

在marker.rs中是通过auto trait来实现。

pub unsafe auto trait Sync { }

auto trait又称为opt-in, built-in trait (OIBIT)。这是一种不稳定的特性，每个类型都会自动实现一个特征，除非它们选择退出或包含一个不实现该特征的类型。

换言之，opt-in对应还有个opt-out，可以通过!(negative trait impl)语法来实现。

例如：下面代码中第一行表示类型Wrapper实现了Send，但是却没实现Sync。

unsafe impl Send for Wrapper {}
unsafe impl !Sync for Wrapper {}

本节将会重点讲解Send、Sync相关的并发知识。

1.auto trait

可以通过安装nightly版使用feature特性。

rustup toolchain install nightly

下面以自定义auto trait实现为例：

#![feature(negative_impls)]
#![feature(auto_traits)]
auto trait IsCool {}

impl !IsCool for String {}

struct MyStruct;
struct HasAString(String);

fn check_cool<C: IsCool>(_: C) {}

调用：

check_cool(42);
check_cool(false);
check_cool(MyStruct);
# the trait `IsCool` is not implemented for `std::string::String`
check_cool(String::new());

这里给了一个简单的例子，展示了auto trait的用法，当没有实现(通过!)auto trait时，编译器会在编译阶段报：the trait XXX is not implemented for YYY。

2.Send与Sync

Send含义：跨线程move，ownership。Sync含义：跨线程share data，borrow。

通常在我们编译多线程代码时，会存在所有权转移、数据共享。那么问题来了，Rc与原生指针是否可以在多线程使用呢？

我们打开Rc(Reference Counting, 引用计数)的源码可以看到这里使用了negative trait，并没有实现Send与Sync，因此通过Rc包裹的对象并不是线程安全的，只能用在单线程中。

impl<T: ?Sized> !marker::Send for Rc<T> {}
impl<T: ?Sized> !marker::Sync for Rc<T> {}

如果我们将它用在多线程中，会出什么问题呢？

fn main() {
    let val = std::rc::Rc::new(5);
    let t = std::thread::spawn(move || {
        println!("this is a thread val: {}",val);
    });

    t.join().unwrap();
}

报错：

error[E0277]: `Rc<i32>` cannot be sent between threads safely
...
the trait `Send` is not implemented for `Rc<i32>`
...

与之对应，Arc(Atomic Reference Counted, 原子引用计数)，可以看一下源码实现：

unsafe impl<T: ?Sized + Sync + Send> Send for Arc<T> {}
unsafe impl<T: ?Sized + Sync + Send> Sync for Arc<T> {}

Send与Sync都实现了。

Send可以实现在多线程间安全传递所有权，Sync可以线程安全的共享数据(例如：引用)。

此外，官方文档：当且仅当类型T的引用&T是Send，T是Sync。

大概意思就是如果引用都无法在多线程之前传递，那么底层数据变无法进行数据共享了。

marker.rs中还有段比较重要的代码，表示原生指针不是线程安全的，没有实现Send、Sync trait。

impl<T: ?Sized> !Send for *const T {}
impl<T: ?Sized> !Send for *mut T {}
impl<T: ?Sized> !Sync for *const T {}
impl<T: ?Sized> !Sync for *mut T {}

Mutex与RwLock

Mutex与RwLock相比于其他语言来说，实现了用户友好的接口，通过new即可将类型传递进去。

Arc::new(Mutex::new(Foo{}))

在Go中使用Mutex，张这个样子：

v   map[string]int
mux sync.Mutex

可以看到rust一行便可以知道保护的是哪个数据。Mutex是用来保护共享变量，所以这个变量类型T我们猜测可以是安全的，也可以是不安全的，所以Sync是不被要求的，因此我们看源码：

unsafe impl<T: ?Sized + Send> Send for Mutex<T> {}
unsafe impl<T: ?Sized + Send> Sync for Mutex<T> {}

Mutex会去实现Send与Sync，要求的类型T一定是具有所有权(实现Send)，但是并不要求数据是否是安全的(没实现Sync)。

同理：RwLock是读写锁，需要满足并发读，因此要求T必须实现Sync。

unsafe impl<T: ?Sized + Send> Send for RwLock<T> {}
unsafe impl<T: ?Sized + Send + Sync> Sync for RwLock<T> {}

小知识

前面讲解了raw pointer并不是线程安全的，那么如何实现线程安全呢？

其实也比较简单：可以通过如下多种方法：

• 自定义类型

  将raw pointer包裹起来即可。

struct Wrapper(*mut i32);
unsafe impl Send for Wrapper {}
unsafe impl Sync for Wrapper {}

• Box

  使用智能指针Box。

Box::new(my_num)

本节完~