百度360必应搜狗淘宝本站头条
python判断字典是否为空crontab每周一执行aes和des区别安装指定版本npmadb pull
当前位置：网站首页 > IT知识 > 正文

Rust语言从入门到精通系列 - 深入理解Tokio的Stream

liuian 2025-03-04 13:07 17 浏览

在 Rust 语言中，Tokio 是一个非常流行的异步编程框架。它提供了一系列的模块，其中最常用的就是 Stream 模块。Stream 模块允许我们以异步的方式处理数据流，这在很多情况下非常有用。在本教程中，我们将介绍 Stream 模块的基础用法和进阶用法，并提供示例。

基础用法

在本节中，我们将介绍 Stream 模块的基础用法，并提供基础示例。

从 Vec 中创建 Stream

首先，我们将从一个 Vec 中创建一个 Stream。假设我们有一个包含数字 1 到 10 的 Vec，我们可以使用stream::iter函数来创建一个 Stream。

use tokio::stream::StreamExt;

#[tokio::main]
async fn main() {
    let vec = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
    let mut stream = tokio::stream::iter(vec);

    while let Some(num) = stream.next().await {
        println!("{}", num);
    }
}

在上面的代码中，我们使用了StreamExt trait 中的next方法来遍历 Stream 中的每个元素。注意，我们需要使用await关键字来等待每个元素的到来。

从文件中创建 Stream

接下来，我们将介绍如何从文件中创建一个 Stream。假设我们有一个名为data.txt的文件，其中包含一些文本行。我们可以使用tokio::fs::File::open方法来打开文件，并使用tokio::io::BufReader来读取文件中的每一行。

use tokio::io::{AsyncBufReadExt, BufReader};
use tokio::fs::File;

#[tokio::main]
async fn main() {
    let file = File::open("data.txt").await.unwrap();
    let mut reader = BufReader::new(file).lines();

    while let Some(line) = reader.next_line().await.unwrap() {
        println!("{}", line);
    }
}

在上面的代码中，我们使用了AsyncBufReadExt trait 中的next_line方法来遍历 Stream 中的每个元素。注意，我们需要使用await关键字来等待每个元素的到来。

使用 Stream 的 map 方法

接下来，我们将介绍如何使用 Stream 的map方法来对 Stream 中的元素进行转换。假设我们有一个包含数字 1 到 10 的 Vec，我们可以使用stream::iter函数来创建一个 Stream，并使用map方法将每个数字乘以 2。

use tokio::stream::StreamExt;

#[tokio::main]
async fn main() {
    let vec = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
    let mut stream = tokio::stream::iter(vec).map(|x| x * 2);

    while let Some(num) = stream.next().await {
        println!("{}", num);
    }
}

在上面的代码中，我们使用了map方法将每个数字乘以 2。这种方式非常适合对 Stream 中的元素进行转换。

使用 Stream 的 filter 方法

接下来，我们将介绍如何使用 Stream 的filter方法来过滤 Stream 中的元素。假设我们有一个包含数字 1 到 10 的 Vec，我们可以使用stream::iter函数来创建一个 Stream，并使用filter方法将大于 5 的数字过滤出来。

use tokio::stream::StreamExt;

#[tokio::main]
async fn main() {
    let vec = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
    let mut stream = tokio::stream::iter(vec).filter(|x| *x > 5);

    while let Some(num) = stream.next().await {
        println!("{}", num);
    }
}

在上面的代码中，我们使用了filter方法将大于 5 的数字过滤出来。这种方式非常适合对 Stream 中的元素进行过滤。

使用 Stream 的 take 方法

接下来，我们将介绍如何使用 Stream 的take方法来限制 Stream 中的元素数量。假设我们有一个包含数字 1 到 10 的 Vec，我们可以使用stream::iter函数来创建一个 Stream，并使用take方法限制只输出前 3 个数字。

use tokio::stream::StreamExt;

#[tokio::main]
async fn main() {
    let vec = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
    let mut stream = tokio::stream::iter(vec).take(3);

    while let Some(num) = stream.next().await {
        println!("{}", num);
    }
}

在上面的代码中，我们使用了take方法限制只输出前 3 个数字。这种方式非常适合对 Stream 中的元素数量进行限制。

使用 Stream 的 fold 方法

最后，我们将介绍如何使用 Stream 的fold方法来对 Stream 中的元素进行累加。假设我们有一个包含数字 1 到 10 的 Vec，我们可以使用stream::iter函数来创建一个 Stream，并使用fold方法将每个数字相加。

use tokio::stream::StreamExt;

#[tokio::main]
async fn main() {
    let vec = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
    let sum = tokio::stream::iter(vec).fold(0, |acc, x| async move { acc + x }).await;

    println!("{}", sum);
}

在上面的代码中，我们使用了fold方法将每个数字相加。注意，我们需要使用async move关键字来让闭包具有异步能力。

进阶用法

在本节中，我们将介绍 Stream 模块的进阶用法，并提供进阶示例。

使用 Stream 的 buffer_unordered 方法

首先，我们将介绍如何使用 Stream 的buffer_unordered方法来并发处理 Stream 中的元素。假设我们有一个包含数字 1 到 10 的 Vec，我们可以使用stream::iter函数来创建一个 Stream，并使用buffer_unordered方法并发处理每个数字。

use tokio::stream::StreamExt;

#[tokio::main]
async fn main() {
    let vec = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
    let mut stream = tokio::stream::iter(vec).buffer_unordered(4);

    while let Some(num) = stream.next().await {
        println!("{}", num);
    }
}

在上面的代码中，我们使用了buffer_unordered方法并发处理每个数字。注意，我们需要使用await关键字来等待每个元素的到来。

使用 Stream 的 zip 方法

接下来，我们将介绍如何使用 Stream 的zip方法将两个 Stream 合并为一个 Stream。假设我们有两个包含数字 1 到 5 的 Vec，我们可以使用stream::iter函数来创建两个 Stream，并使用zip方法将它们合并为一个 Stream。

use tokio::stream::StreamExt;

#[tokio::main]
async fn main() {
    let vec1 = vec![1, 2, 3, 4, 5];
    let vec2 = vec![6, 7, 8, 9, 10];
    let mut stream1 = tokio::stream::iter(vec1);
    let mut stream2 = tokio::stream::iter(vec2);
    let mut stream = stream1.zip(stream2);

    while let Some((num1, num2)) = stream.next().await {
        println!("{} {}", num1, num2);
    }
}

在上面的代码中，我们使用了zip方法将两个 Stream 合并为一个 Stream。注意，我们需要使用await关键字来等待每个元素的到来。

使用 Stream 的 forward 方法

最后，我们将介绍如何使用 Stream 的forward方法将一个 Stream 转发到另一个 Stream。假设我们有一个名为data.txt的文件，其中包含一些文本行。我们可以使用tokio::fs::File::open方法来打开文件，并使用tokio::io::BufReader来读取文件中的每一行。然后，我们可以使用forward方法将读取的每一行转发到标准输出。

use tokio::io::{AsyncBufReadExt, BufReader};
use tokio::fs::File;
use tokio::stream::StreamExt;

#[tokio::main]
async fn main() {
    let file = File::open("data.txt").await.unwrap();
    let mut reader = BufReader::new(file).lines();
    let stdout = tokio::io::stdout();
    let mut writer = tokio::io::BufWriter::new(stdout);

    reader.forward(&mut writer).await.unwrap();
}

在上面的代码中，我们使用了forward方法将读取的每一行转发到标准输出。注意，我们需要使用await关键字来等待每个元素的到来。

结论

在本教程中，我们介绍了 Rust 语言中的 Tokio 模块 Stream 的基础用法和进阶用法。Stream 模块提供了一种非常方便的方式来处理数据流，这在异步编程中非常有用。我们希望这个教程可以帮助你更好地理解 Stream 模块的用法和特性。

相关推荐

教你把多个视频合并成一个视频的方法

一.情况介绍当你有一个m3u8文件和一个目录，目录中有连续的视频片段，这些片段可以连成一段完整的视频。m3u8文件打开后像这样：m3u8文件,可以理解为播放列表，里面是播放视频片段的顺序。视频片段像这...

零代码编程:用kimichat合并一个文件夹下的多个文件

一个文件夹里面有很多个srt字幕文件，如何借助kimichat来自动批量合并呢？在kimichat对话框中输入提示词：你是一个Python编程专家，完成如下的编程任务：这个文件夹：D:\downloa...

Java APT_java APT 生成代码

JavaAPT（AnnotationProcessingTool）是一种在Java编译阶段处理注解的工具。APT会在编译阶段扫描源代码中的注解，并根据这些注解生成代码、资源文件或其他输出，...

Unit Runtime:一键运行 AI 生成的代码，或许将成为你的复制 + 粘贴神器

在我们构建了UnitMesh架构之后，以及对应的demo之后，便着手于实现UnitMesh架构。于是，我们就继续开始UnitRuntime，以用于直接运行AI生成的代码。PS：...

挣脱臃肿的枷锁:为什么说Vert.x是Java开发者手中的一柄利剑?

如果你是一名Java开发者，那么你的职业生涯几乎无法避开Spring。它如同一位德高望重的老国王，统治着企业级应用开发的大片疆土。SpringBoot的约定大于配置、SpringCloud的微服务...

五年后，谷歌还在全力以赴发展 Kotlin

作者|FredericLardinois译者|Sambodhi策划|Tina自2017年谷歌I/O全球开发者大会上，谷歌首次宣布将Kotlin（JetBrains开发的Ja...

kotlin和java开发哪个好，优缺点对比

Kotlin和Java都是常见的编程语言，它们有各自的优缺点。Kotlin的优点：简洁：Kotlin程序相对于Java程序更简洁，可以减少代码量。安全：Kotlin在类型系统和空值安全...

移动端架构模式全景解析:从MVC到MVVM，如何选择最佳设计方案?

掌握不同架构模式的精髓，是构建可维护、可测试且高效移动应用的关键。在移动应用开发中，选择合适的软件架构模式对项目的可维护性、可测试性和团队协作效率至关重要。随着应用复杂度的增加，一个良好的架构能够帮助...

颜值非常高的XShell替代工具Termora，不一样的使用体验!

Termora是一款面向开发者和运维人员的跨平台SSH终端与文件管理工具，支持Windows、macOS及Linux系统，通过一体化界面简化远程服务器管理流程。其核心定位是解决多平台环境下远程连接、文...

预处理的底层原理和预处理编译运行异常的解决方案

若文章对您有帮助，欢迎关注程序员小迷。助您在编程路上越走越好！[Mac-10.7.1LionIntel-based]Q：预处理到底干了什么事情？A：预处理，顾名思义，预先做的处理。源代码中...

为“架构”再建个模:如何用代码描述软件架构?

在架构治理平台ArchGuard中，为了实现对架构的治理，我们需要代码+模型描述所要处理的内容和数据。所以，在ArchGuard中，我们有了代码的模型、依赖的模型、变更的模型等，剩下的两个...

深度解析:Google Gemma 3n —— 移动优先的轻量多模态大模型

2025年6月，Google正式发布了Gemma3n，这是一款能够在2GB内存环境下运行的轻量级多模态大模型。它延续了Gemma家族的开源基因，同时在架构设计上大幅优化，目标是让...

比分网开发技术栈与功能详解_比分网有哪些

一、核心功能模块一个基本的比分网通常包含以下模块：首页/总览实时比分看板：滚动展示所有正在进行的比赛，包含比分、比赛时间、红黄牌等关键信息。热门赛事/焦点战：突出显示重要的、关注度高的比赛。赛事导航...

设计模式之-生成器_一键生成设计

一、【概念定义】——“分步构建复杂对象，隐藏创建细节”生成器模式（BuilderPattern）：一种“分步构建型”创建型设计模式，它将一个复杂对象的构建与其表示分离，使得同样的构建过程可以创建...

构建第一个 Kotlin Android 应用_kotlin简介

第一步：安装AndroidStudio（推荐IDE）AndroidStudio是官方推荐的Android开发集成开发环境（IDE），内置对Kotlin的完整支持。1.下载And...

一周热门
最近发表
标签列表