当前位置：首页 > article >正文

Rust 并发编程：Futures、Tasks 和 Threads 的结合使用

article 2025/3/6 2:51:19

一、线程（Threads）与异步（Async）的对比

1.1. 线程的优势与限制

线程是一种广泛使用的并发模型，几乎所有现代操作系统都支持。Rust 的标准库提供了 std::thread API，使得线程编程变得直观。然而，线程也有一些限制：

每个线程占用较多内存（栈空间一般为 2MB 左右）。
线程的创建和销毁有一定的性能开销。
在没有操作系统支持的情况下（如某些嵌入式系统），线程模型无法使用。

1.2. 异步的优势与限制

异步编程基于 async 和 await，提供了一种不同的并发方式，它的核心概念是 Future。Rust 的 Future 由运行时管理，而非操作系统线程。其主要优势包括：

任务（Task）比线程更轻量级，适合大量并发操作。
适用于 IO 密集型任务，如处理多个网络请求、消息队列等。
任务之间可以高效地共享资源。

但异步编程也有一定的限制：

需要一个异步运行时（如 tokio 或 async-std）。
代码复杂度较高，尤其是涉及 Pin、Send 和 Sync 等特性的情况。
适用于 IO 绑定任务，而 CPU 密集型任务可能仍然需要线程。

二、线程与异步任务的结合使用

在许多实际场景中，我们可以同时使用线程和异步任务，以发挥各自的优势。例如，我们可以在后台线程中执行 CPU 密集型任务，并使用异步任务来管理 IO 任务。

2.1.线程和异步的对比示例

让我们来看一个例子：

use std::thread;
use std::time::Duration;
use async_channel::unbounded;
use tokio::task;

#[tokio::main]
async fn main() {
    let (tx, rx) = unbounded();

    thread::spawn(move || {
        for i in 1..=10 {
            tx.send(i).unwrap();
            thread::sleep(Duration::from_secs(1));
        }
    });

    while let Ok(value) = rx.recv().await {
        println!("Received: {}", value);
    }
}

在这个例子中：

我们创建了一个 async_channel 以支持异步通信。
使用 thread::spawn 启动一个 OS 线程，在其中发送数据。
在主 async 任务中等待接收消息，并异步处理它们。

三、什么时候使用线程，什么时候使用异步？

在选择并发模型时，可以遵循以下原则：

线程适用于高并行计算（Parallelism）：如 CPU 密集型任务，例如视频编码、图像处理、密码学计算等。
异步适用于高并发（Concurrency）：如 IO 绑定任务，例如 Web 服务器、消息队列、数据库操作等。
结合使用：
- 如果某个任务主要是计算密集型，但仍然需要异步处理结果，使用 spawn_blocking。
- 如果任务主要是异步的，但某些部分需要更高的并行度，可以在异步任务中启动线程。

3.1. 结合线程和异步的应用场景

以下是几个实际应用示例：

Web 服务器：在 async 代码中处理 HTTP 请求，但将 CPU 密集型任务交给线程池。
数据库访问：使用 async 处理数据库连接，但在后台线程执行复杂查询。
游戏引擎：使用 async 处理网络 IO，但使用多线程进行物理计算。

四、Rust 运行时的多线程支持

许多 Rust 异步运行时（如 tokio）本身是多线程的。它们采用 工作窃取（Work Stealing） 机制，使任务能够在多个线程之间调度，从而提高 CPU 利用率。

use tokio::task;

#[tokio::main]
async fn main() {
    let handle = task::spawn_blocking(|| {
        // 在专门的线程池中运行
        heavy_computation()
    });
    let result = handle.await.unwrap();
    println!("Computation result: {}", result);
}

fn heavy_computation() -> i32 {
    // 执行 CPU 密集型任务
    42
}