百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术分析 > 正文

Rust语言从入门到精通系列 - 零基础掌握prost库和Protobuf

liebian365 2024-10-24 14:38 32 浏览 0 评论


Prost是一个用于序列化和反序列化协议缓冲区数据的Rust语言库。它使用Google Protocol Buffers语言来定义协议,并生成Rust代码以便使用该协议。 Prost具有高性能的特点,并且支持许多protobuf功能,例如嵌套消息、默认值、枚举类型以及变长编码。

Prost支持从protobuf2和protobuf3生成代码,而且可以与其他Rust语言库和框架无缝集成。

模块场景和基础用法

Prost可以用于许多场景,包括网络通信、持久化、日志记录等。在这里,我们将通过一个简单的例子来介绍Prost的基础用法。

首先在Cargo.toml中引入prost模块,示例配置如下:

[dependencies]
prost = "0.11"
# Only necessary if using Protobuf well-known types:
prost-types = "0.11"

假设我们有一个动物园,里面有许多不同种类的动物。我们可以使用Prost来定义一个动物的协议,然后使用该协议来序列化和反序列化动物对象。

首先,我们需要定义动物的protobuf文件。在这里,我们定义了一个动物具有名称、年龄和类型。动物类型是一个枚举类型,它可以是狗、猫或鸟。

syntax = "proto3";

enum AnimalType {
    DOG = 0;
    CAT = 1;
    BIRD = 2;
}

message Animal {
    string name = 1;
    uint32 age = 2;
    AnimalType animal_type = 3;
}

接下来,我们需要使用Prost生成Rust代码。我们可以使用以下命令来执行此操作:

$ protoc --rust_out . animals.proto

这将生成一个名为animals.rs的文件,其中包含与protobuf定义相对应的Rust代码。

接下来,我们可以使用Prost来序列化和反序列化动物对象。以下是一个示例代码:

use prost::{Enumeration, Message};

#[derive(Clone, PartialEq, Message)]
pub struct Animal {
    #[prost(string, tag="1")]
    pub name: String,
    #[prost(uint32, tag="2")]
    pub age: u32,
    #[prost(enumeration="AnimalType", tag="3")]
    pub animal_type: i32,
}

#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, Enumeration)]
pub enum AnimalType {
    Dog = 0,
    Cat = 1,
    Bird = 2,
}

fn main() {
    let mut animal = Animal::default();
    animal.name = "Tom".to_string();
    animal.age = 3;
    animal.animal_type = AnimalType::Cat as i32;

    let mut buf = Vec::new();
    animal.encode(&mut buf).unwrap();

    let decoded_animal = Animal::decode(&buf[..]).unwrap();
    assert_eq!(animal, decoded_animal);
    println!("{:?}", animal);
}
//  输出结果:
//  Animal { name: "Tom", age: 3, animal_type: Cat } 

在这个示例代码中,我们定义了一个名为Animal的结构体,并使用prost宏将其与protobuf定义相关联。我们还定义了一个名为AnimalType的枚举类型,它与protobuf定义中的枚举类型相对应。

main函数中,我们创建了一个Animal对象,并将其序列化为字节数组。然后,我们将字节数组反序列化为另一个Animal对象,并使用assert_eq宏比较这两个对象是否相等。

高级特性

Prost提供了许多高级特性,例如自定义类型、扩展字段、oneof等。在这里,我们将介绍其中一些特性。

自定义类型

有时,我们可能需要在protobuf定义中使用自定义类型。例如,我们可能需要使用自定义类型来表示日期或时间。在这种情况下,我们可以使用prost宏的bytes属性来定义自定义类型。

以下是一个示例代码:

syntax = "proto3";

message Date {
    bytes value = 1 [(prost(bytes_type) = "chrono::NaiveDate")];
}

message Time {
    bytes value = 1 [(prost(bytes_type) = "chrono::NaiveTime")];
}

在这个示例代码中,我们定义了两个消息类型:DateTime。它们都包含一个名为value的字节数组字段,并使用prost宏的bytes_type属性将其与chrono库中的NaiveDateNaiveTime类型相关联。

自定义编解码

Prost支持自定义编解码,可以使用prost::Message trait来实现自定义编解码。

impl Animal {
    pub fn from_bytes(bytes: &[u8]) -> Result<Self, prost::DecodeError> {
        Animal::decode(bytes)
    }

    pub fn to_bytes(&self) -> Result<Vec<u8>, prost::EncodeError> {
        let mut buf = Vec::new();
        self.encode(&mut buf)?;
        Ok(buf)
    }
}
fn main() {
    let mut animal = Animal::default();
    animal.name = "Tom".to_string();
    animal.age = 3;
    animal.animal_type = AnimalType::Cat as i32;

    let bytes = animal.to_bytes();
    println!("{:?}", Animal::from_bytes(&bytes.unwrap()));
}
//  输出结果:
// Ok(Animal { name: "Tom", age: 3, animal_type: Cat })

扩展字段

有时,我们可能需要向protobuf消息添加额外的字段,但是又不想破坏现有的消息格式。在这种情况下,我们可以使用扩展字段。

扩展字段是在protobuf定义中定义的,但是在生成的Rust代码中不会出现。它们可以用来存储任何类型的数据,并且可以与protobuf消息一起序列化和反序列化。

以下是一个示例代码:

syntax = "proto3";

message Animal {
    string name = 1;
    uint32 age = 2;
    AnimalType animal_type = 3;

    map<string, bytes> extensions = 1000;
}

在这个示例代码中,我们添加了一个名为extensions的字段,它是一个map类型,可以存储任何类型的数据。此字段的标签为1000,这意味着它是一个扩展字段。

在Rust代码中,我们可以使用prost::Message trait的extensions方法来访问扩展字段。以下是一个示例代码:

use prost::{Enumeration, Message};
use std::collections::HashMap;

#[derive(Clone, PartialEq, Message)]
pub struct Animal {
    #[prost(string, tag="1")]
    pub name: String,
    #[prost(uint32, tag="2")]
    pub age: u32,
    #[prost(enumeration="AnimalType", tag="3")]
    pub animal_type: i32,
    #[prost(map="string, bytes", tag="1000")]
    pub extensions: HashMap<String, Vec<u8>>,
}

#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, Enumeration)]
pub enum AnimalType {
    Dog = 0,
    Cat = 1,
    Bird = 2,
}

fn main() {
    let mut animal = Animal::default();
    animal.extensions.insert("color".to_string(), b"brown".to_vec());
    
    let mut buf = Vec::new();
    animal.encode(&mut buf).unwrap();
    
    let decoded_animal = Animal::decode(&buf[..]).unwrap();
    assert_eq!(animal.extensions, decoded_animal.extensions);
}

在这个示例代码中,我们创建了一个Animal对象,并向其添加了一个名为color的扩展字段。然后,我们将该对象序列化为字节数组,并将其反序列化为另一个Animal对象。最后,我们使用assert_eq宏比较这两个对象的扩展字段是否相等。

Proto Oneof

有时,我们可能需要在protobuf消息中使用oneof语法,以表示字段中的多个可能类型。在这种情况下,我们可以使用prost宏的oneof属性来定义oneof字段。

以下是一个示例代码:

syntax = "proto3";

message Animal {
    string name = 1;
    uint32 age = 2;
    oneof animal_type {
        Dog dog = 3;
        Cat cat = 4;
        Bird bird = 5;
    }
}
message Dog {
    string breed = 1;
}
message Cat {
    bool has_tail = 1;
}
message Bird {
    uint32 wingspan = 1;
}

在这个示例代码中,我们定义了一个名为Animal的消息类型,它包含一个名为animal_typeoneof字段。oneof字段中包含三个可能的类型:DogCatBird。每个类型都包含与其相关联的字段。

在Rust代码中,我们可以使用prost::Oneof trait来访问oneof字段。以下是一个示例代码:

use prost::{Enumeration, Message, Oneof};
use std::collections::HashMap;
use core::option::Option;

#[derive(Clone, PartialEq, Message)]
pub struct Animal {
    #[prost(string, tag="1")]
    pub name: String,
    #[prost(uint32, tag="2")]
    pub age: u32,
    #[prost(oneof="AnimalType", tag="3,4,5")]
    pub animal_type: Option<AnimalType>,
}
#[derive(Clone, Debug, PartialEq, Enumeration)]
pub enum AnimalType {
    #[prost(message, tag = "3", name = "Dog")]
    Dog(Dog),
    #[prost(message, tag = "4", name = "Cat")]
    Cat(Cat),
    #[prost(message, tag = "5", name = "Bird")]
    Bird(Bird),
}
#[derive(Clone, PartialEq, Message)]
pub struct Dog {
    #[prost(string, tag="1")]
    pub breed: String
}
#[derive(Clone, PartialEq, Message)]
pub struct Cat {
    #[prost(bool, tag="1")]
    pub has_tail: bool
}
#[derive(Clone, PartialEq, Message)]
pub struct Bird {
    #[prost(uint32, tag="1")]
    pub wingspan: u32
}
fn main() {
    let mut animal = Animal::default();
    animal.name = "Tom".to_string();
    animal.age = 3;
    animal.animal_type = Some(AnimalType::Cat(Cat { has_tail: true }));
    
    let mut buf = Vec::new();
    animal.encode(&mut buf).unwrap();
    
    let decoded_animal = Animal::decode(&buf[..]).unwrap();
    assert_eq!(animal, decoded_animal);
}

在这个示例代码中,我们创建了一个Animal对象,并将其cat字段设置为一个包含has_tail字段的Cat对象。然后,我们将该对象序列化为字节数组,并将其反序列化为另一个Animal对象。最后,我们使用assert_eq宏比较这两个对象是否相等。


prost最佳实践

以下是一些使用Prost的最佳实践经验:

  • ? 在protobuf定义中使用简单的数据类型。Prost支持许多protobuf功能,例如嵌套消息、默认值、枚举类型以及变长编码。但是,使用这些功能可能会导致生成的Rust代码变得复杂。因此,为了使代码保持简单和易于维护,请尽可能使用简单的数据类型。
  • ? 在Rust代码中使用结构体。Prost生成的Rust代码可以是一个模块或一个trait。但是,使用结构体可以使代码更易于使用和维护。因此,建议在Rust代码中使用结构体。
  • ? 使用自定义类型时,请使用标准库或第三方库。Prost支持许多自定义类型,包括日期、时间、UUID等。但是,使用标准库或第三方库可能会使代码更加通用和可移植。因此,建议在使用自定义类型时使用标准库或第三方库。
  • ? 在使用扩展字段时,请注意字段标签。扩展字段的标签必须大于1000。因此,请确保您为扩展字段选择一个大于1000的标签。
  • ? 在使用oneof语法时,请选择一个好的字段名称。oneof字段包含多个可能的类型,因此请为其选择一个好的字段名称。这将使代码更易于理解和维护。

总结

Prost是一个高性能的Rust语言库,可用于序列化和反序列化协议缓冲区数据。它支持许多protobuf功能,并且可以与其他Rust语言库和框架无缝集成。在本教程中,我们介绍了Prost的基础用法和一些高级特性,并提供了一些最佳实践经验。我们希望这个教程能够帮助您更好地使用Prost。

#头条创作挑战赛# #大有学问##每天学点Rust##Rust##程序员##从今天起记录我的2023?#

相关推荐

“版本末期”了?下周平衡补丁!国服最强5套牌!上分首选

明天,酒馆战棋就将迎来大更新,也聊了很多天战棋相关的内容了,趁此机会,给兄弟们穿插一篇构筑模式的卡组推荐!老规矩,我们先来看10职业胜率。目前10职业胜率排名与一周前基本类似,没有太多的变化。平衡补丁...

VS2017 C++ 程序报错“error C2065:“M_PI”: 未声明的标识符&quot;

首先,程序中头文件的选择,要选择头文件,在文件中是没有对M_PI的定义的。选择:项目——>”XXX属性"——>配置属性——>C/C++——>预处理器——>预处理器定义,...

东营交警实名曝光一批酒驾人员名单 88人受处罚

齐鲁网·闪电新闻5月24日讯酒后驾驶是对自己和他人生命安全极不负责的行为,为守护大家的平安出行路,东营交警一直将酒驾作为重点打击对象。5月23日,东营交警公布最新一批饮酒、醉酒名单。对以下驾驶人醉酒...

Qt界面——搭配QCustomPlot(qt platform)

这是我第一个使用QCustomPlot控件的上位机,通过串口精确的5ms发送一次数据,再将读取的数据绘制到图表中。界面方面,尝试卡片式设计,外加QSS简单的配了个色。QCustomPlot官网:Qt...

大话西游2分享赢取种族坐骑手办!PK趣闻录由你书写

老友相聚,仗剑江湖!《大话西游2》2021全民PK季4月激燃打响,各PK玩法鏖战齐开,零门槛参与热情高涨。PK季期间,不仅各种玩法奖励丰厚,参与PK趣闻录活动,投稿自己在PK季遇到的趣事,还有机会带走...

测试谷歌VS Code AI 编程插件 Gemini Code Assist

用ClaudeSonnet3.7的天气测试编码,让谷歌VSCodeAI编程插件GeminiCodeAssist自动编程。生成的文件在浏览器中的效果如下:(附源代码)VSCode...

顾爷想知道第4.5期 国服便利性到底需优化啥?

前段时间DNF国服推出了名为“阿拉德B计划”的系列改版计划,截至目前我们已经看到了两项实装。不过关于便利性上,国服似乎还有很多路要走。自从顾爷回归DNF以来,几乎每天都在跟我抱怨关于DNF里面各种各样...

掌握Visual Studio项目配置【基础篇】

1.前言VisualStudio是Windows上最常用的C++集成开发环境之一,简称VS。VS功能十分强大,对应的,其配置系统较为复杂。不管是对于初学者还是有一定开发经验的开发者来说,捋清楚VS...

还嫌LED驱动设计套路深?那就来看看这篇文章吧

随着LED在各个领域的不同应用需求,LED驱动电路也在不断进步和发展。本文从LED的特性入手,推导出适合LED的电源驱动类型,再进一步介绍各类LED驱动设计。设计必读:LED四个关键特性特性一:非线...

Visual Studio Community 2022(VS2022)安装图文方法

直接上步骤:1,首先可以下载安装一个VisualStudio安装器,叫做VisualStudioinstaller。这个安装文件很小,很快就安装完成了。2,打开VisualStudioins...

Qt添加MSVC构建套件的方法(qt添加c++11)

前言有些时候,在Windows下因为某些需求需要使用MSVC编译器对程序进行编译,假设我们安装Qt的时候又只是安装了MingW构建套件,那么此时我们该如何给现有的Qt添加一个MSVC构建套件呢?本文以...

Qt为什么站稳c++GUI的top1(qt c)

为什么现在QT越来越成为c++界面编程的第一选择,从事QT编程多年,在这之前做C++界面都是基于MFC。当时为什么会从MFC转到QT?主要原因是MFC开发界面想做得好看一些十分困难,引用第三方基于MF...

qt开发IDE应该选择VS还是qt creator

如果一个公司选择了qt来开发自己的产品,在面临IDE的选择时会出现vs或者qtcreator,选择qt的IDE需要结合产品需求、部署平台、项目定位、程序猿本身和公司战略,因为大的软件产品需要明确IDE...

Qt 5.14.2超详细安装教程,不会来打我

Qt简介Qt(官方发音[kju:t],音同cute)是一个跨平台的C++开库,主要用来开发图形用户界面(GraphicalUserInterface,GUI)程序。Qt是纯C++开...

Cygwin配置与使用(四)——VI字体和颜色的配置

简介:VI的操作模式,基本上VI可以分为三种状态,分别是命令模式(commandmode)、插入模式(Insertmode)和底行模式(lastlinemode),各模式的功能区分如下:1)...

取消回复欢迎 发表评论: