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并发编程的三种特性:原子类型、线程间通信和线程池的使用。

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

原文地址:https://www.cnblogs.com/Jackeyzhe/p/12563384.html

时间: 2024-11-06 23:24:11

Rust入坑指南:齐头并进(下)的相关文章

Rust入坑指南:亡羊补牢

如果你已经开始学习Rust,相信你已经体会过Rust编译器的强大.它可以帮助你避免程序中的大部分错误,但是编译器也不是万能的,如果程序写的不恰当,还是会发生错误,让程序崩溃.所以今天我们就来聊一聊Rust中如何处理程序错误,也就是所谓的"亡羊补牢". 基础概念 在编程中遇到的非正常情况通常可以分为三类:失败.错误.异常. Rust中用两种方式来消除失败:强大的类型系统和断言. 对于类型系统,熟悉Java的同学应该比较清楚.例如我们给一个接收参数为int的函数传入了字符串类型的变量.这是

Rust入坑指南:坑主驾到

欢迎大家和我一起入坑Rust,以后我就是坑主,我主要负责在前面挖坑,各位可以在上面看,有手痒的也可以和我一起挖.这个坑到底有多深?我也不知道,我是抱着有多深就挖多深的心态来的,下面我先跳了,各位请随意. Rust简介 众所周知,在编程语言中,更易读的高级语言和控制底层资源的低级语言是一对矛盾体.Rust想要挑战这一现状,它尝试为开发者提供更好的体验的同时给予开发者控制底层细节的权限(比如内存使用). 低级语言在开发过程中很容易出现各种细微的错误,它们难以发现但是可能影响巨大.其他大部分低级语言只

Rust入坑指南:海纳百川

今天来聊Rust中两个重要的概念:泛型和trait.很多编程语言都支持泛型,Rust也不例外,相信大家对泛型也都比较熟悉,它可以表示任意一种数据类型.trait同样不是Rust所特有的特性,它借鉴于Haskell中的Typeclass.简单来讲,Rust中的trait就是对类型行为的抽象,你可以把它理解为Java中的接口. 泛型 在前面的文章中,我们其实已经提及了一些泛型类型.例如Option.Vec和Result<T, E>.泛型可以在函数.数据结构.Enum和方法中进行定义.在Rust中,

Rust入坑指南:步步为营

俗话说:"测试写得好,奖金少不了." 有经验的开发人员通常会通过单元测试来保证代码基本逻辑的正确性.如果你是一名新手开发者,并且还没体会到单元测试的好处,那么建议你先读一下我之前的一篇文章代码洁癖系列(七):单元测试的地位. 写单元测试一般需要三个步骤: 准备测试用例,测试用例要能覆盖尽可能多的代码 执行需要测试的代码 判断结果,是否是你希望得到的结果 了解了这些以后,我们就来看看在Rust中应该怎么写单元测试. 首先我们建立一个library项目 $ cargo new adder

C语言入坑指南-被遗忘的初始化

前言 什么是初始化?为什么要初始化?静态变量和局部变量的初始化又有什么区别?实际应用中应该怎么做?本文将一一回答这些问题. 什么是初始化 初始化指的是对数据对象或者变量赋予初始值.例如: int value = 8; //声明整型变量并初始化为8int arr[] = {1,2,3}; //声明整型数组arr,并初始化其值为1,2,3 为什么要初始化 我们来看一个示例程序.test0.c程序清单如下: #include <stdio.h>#include <stdlib.h>int

eclipse中导入外部包却无法查看对应源码或Javadoc的 入坑指南

eclipse中导入外部包却无法查看对应源码或Javadoc的 入坑指南 出现这个错误的原因是,你虽然导入了.jar包,但没有配置对应的Javadoc或源码路径,所以在编辑器中无法查看源 码和对应API.接下来我们一起解决这个问题... 在项目名称上右击→ 新建→ 文件夹→ 文件名写lib→ 点击完成 然后把你下载的jar包,复制黏贴到这个lib文件夹 右击lib中的源码包→ 构建路径→ 添加至构建路径,自动生成一个"引用的库" 右击 "引用的库" 中的jar包→

Kotlin快速入坑指南(干货型文档)

<p style="text-align:center;color:#42A5F5;font-size:2em;font-weight: bold;">前言 即使每天10点下班,其实需求很多,我也要用这腐朽的声带喊出:我要学习,我要写文章!! 又是一篇Kotlin的文章,为啥...还不是因为工作需要.毫无疑问,最好的学习方式是通过官方文档去学习.不过个人觉得官方文档多多少少有一些不够高效. 中文官方文档 因此这篇是从我学习的个人视角以文档的形式去输出Kotlin语言基础的学

猿说摄影(上)--入坑指南

最近师弟师妹们以及复读的童鞋临近毕业,有的想买相机拍拍毕业照,记录一下旅行毕业游之类的.五一放假,咱就先不聊技术,聊一下摄影,不过摄影也是一个技术活,而且烧钱.摄影穷三代,单反毁一生.相机贵吗?贵,但贵的不只是相机,还有镜头.为什么这么说呢?大家也知道,单反和微单都是可以更换镜头的.一旦入坑,除了买相机同时买的套头(标准变焦镜头)之外,你很可能会接下来陆陆续续地买其它镜头→_→想拍漂亮的人物,你需要大光圈的定焦镜头:想拍壮阔的风景,你需要广角镜头:想拍飞禽走兽,你需要长焦镜头:你可能还要拍点小花

Docker入坑指南之RUN

总有一些场景,我们需要自己制作一个镜像,可以快速还原环境,又不想被其他因素干扰镜像的纯净,这个时候,就可以选择Docker了,启动便捷,镜像还原很快捷,除了上手不容易. 最近入坑研究了一番,小有心得,故写一篇杂文,记录自己的踩坑经历. 安装Docker的过程可以参考其他前辈的文章,不再赘述,从实战角度说,如何构建一个自用的Docker镜像. 首选说一下Docker的几个名词,仓库是管理镜像的,容器是镜像启动后的,镜像就是最干净的环境,镜像启动之后变成容器. docker的run是启动镜像的介质,