Rust入坑指南:齐头并进(下)

前文中我们聊了Rust如何管理线程以及如何利用Rust中的锁进行编程。今天我们继续学习并发编程,

原子类型

许多编程语言都会提供原子类型,Rust也不例外,在前文中我们聊了Rust中锁的使用,有了锁,就要小心死锁的问题,Rust虽然声称是安全并发,但是仍然无法帮助我们解决死锁的问题。原子类型就是编程语言为我们提供的无锁并发编程的最佳手段。熟悉Java的同学应该知道,Java的编译器并不能保证代码的执行顺序,编译器会对我们的代码的执行顺序进行优化,这一操作成为指令重排。而Rust的多线程内存模型不会进行指令重排,它可以保证指令的执行顺序。

通常来讲原子类型会提供以下操作:

  • Load:从原子类型读取值
  • Store:为一个原子类型写入值
  • CAS(Compare-And-Swap):比较并交换
  • Swap:交换
  • Fetch-add(sub/and/or):表示一系列的原子的加减或逻辑运算

Ok,这些基础的概念聊完以后,我们就来看看Rust为我们提供了哪些原子类型。Rust的原子类型定义在标准库std::sync::atomic中,目前它提供了12种原子类型。

原子类型

下面这段代码是Rust演示了如何用原子类型实现一个自旋锁。

use std::sync::Arc;
use std::sync::atomic::{AtomicUsize, Ordering};
use std::thread;

fn main() {
    let spinlock = Arc::new(AtomicUsize::new(1));
    let spinlock_clone = spinlock.clone();
    let thread = thread::spawn(move|| {
        spinlock_clone.store(0, Ordering::SeqCst);
    });
    while spinlock.load(Ordering::SeqCst) != 0 {}
    if let Err(panic) = thread.join() {
        println!("Thread had an error: {:?}", panic);
    }
}

我们利用AtomicUsize的store方法将它的值设置为0,然后用load方法获取到它的值,如果不是0,则程序一直空转。在store和load方法中,我们都用到了一个参数:Ordering::SeqCst,在声明中能看出来它也是属于atomic包。

我们在文档中发现它是一个枚举。其定义为

pub enum Ordering {
    Relaxed,
    Release,
    Acquire,
    AcqRel,
    SeqCst,
}

它的作用是将内存顺序的控制权交给开发者,我们可以自己定义底层的内存排序。下面我们一起来看一下这5种排序分别代表什么意思

  • Relaxed:表示「没有顺序」,也就是开发者不会干预线程顺序,线程只进行原子操作
  • Release:对于使用Release的store操作,在它之前所有使用Acquire的load操作都是可见的
  • Acquire:对于使用Acquire的load操作,在它之前的所有使用Release的store操作也都是可见的
  • AcqRel:它代表读时使用Acquire顺序的load操作,写时使用Release顺序的store操作
  • SeqCst:使用了SeqCst的原子操作都必须先存储,再加载。

一般情况下建议使用SeqCst,而不推荐使用Relaxed。

线程间通信

Go语言文档中有这样一句话:不要使用共享内存来通信,应该使用通信实现共享内存。

Rust标准库选择了CSP并发模型,也就是依赖channel来进行线程间的通信。它的定义是在标准库std::sync::mpsc中,里面定义了三种类型的CSP进程:

  • Sender:发送异步消息
  • SyncSender:发送同步消息
  • Receiver:用于接收消息

我们通过一个栗子来看一下channel是如何创建并收发消息的。

use std::thread;
use std::sync::mpsc;

fn main() {
    let (tx, rx) = mpsc::channel();

    thread::spawn(move || {
        let val = String::from("hi");
        tx.send(val).unwrap();
    });

    let received = rx.recv().unwrap();
    println!("Got: {}", received);
}

首先,我们先是使用了channel()函数来创建一个channel,它会返回一个(Sender, Receiver)元组。它的缓冲区是无界的。此外,我们还可以使用sync_channel()来创建channel,它返回的则是(SyncSender, Receiver)元组,这样的channel发送消息是同步的,并且可以设置缓冲区大小。

接着,在子线程中,我们定义了一个字符串变量,并使用send()函数向channel中发送消息。这里send返回的是一个Result类型,所以使用unwrap来传播错误。

在main函数最后,我们又用recv()函数来接收消息。

这里需要注意的是,send()函数会转移所有权,所以,如果你在发送消息之后再使用val变量时,程序就会报错。

现在我们已经掌握了使用Channel进行线程间通信的方法了,这里还有一段代码,感兴趣的同学可以自己执行一下这段代码看是否能够顺利执行。如果不能,应该怎么修改这段代码呢?

use std::thread;
use std::sync::mpsc;
fn main() {
    let (tx, rx) = mpsc::channel();
    for i in 0..5 {
        let tx = tx.clone();
        thread::spawn(move || {
            tx.send(i).unwrap();
        });
    }

    for rx in rx.iter() {
        println!("{:?}", j);
    }
}

线程池

在实际工作中,如果每次都要创建新的线程,每次创建、销毁线程的开销就会变得非常可观,甚至会成为系统性能的瓶颈。对于这种问题,我们通常使用线程池来解决。

Rust的标准库中没有现成的线程池给我们使用,不过还是有一些第三方库来支持的。这里我使用的是threadpool

首先需要在Cargo.toml中增加依赖threadpool = "1.7.1"。然后就可以使用use threadpool::ThreadPool;将ThreadPool引入我们的程序中了。

use threadpool::ThreadPool;
use std::sync::mpsc::channel;

fn main() {
    let n_workers = 4;
    let n_jobs = 8;
    let pool = ThreadPool::new(n_workers);

    let (tx, rx) = channel();
    for _ in 0..n_jobs {
        let tx = tx.clone();
        pool.execute(move|| {
            tx.send(1).expect("channel will be there waiting for the pool");
        });
    }

    assert_eq!(rx.iter().take(n_jobs).fold(0, |a, b| a + b), 8);
}

这里我们使用ThreadPool::new()来创建一个线程池,初始化4个工作线程。使用时用execute()方法就可以拿出一个线程来进行具体的工作。

总结

今天我们介绍了Rust并发编程的三种特性:原子类型、线程间通信和线程池的使用。

原子类型是我们进行无锁并发的重要手段,线程间通信和线程池也都是工作中所必须使用的。当然并发编程的知识远不止于此,大家有兴趣的可以自行学习也可以与我交流讨论。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,509评论 6 504
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,806评论 3 394
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,875评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,441评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,488评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,365评论 1 302
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,190评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,062评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,500评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,706评论 3 335
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,834评论 1 347
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,559评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,167评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,779评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,912评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,958评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,779评论 2 354

推荐阅读更多精彩内容