Rust语言从入门到精通系列 - 零基础掌握prost库和Protobuf
liebian365 2024-10-24 14:38 14 浏览 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")];
}
在这个示例代码中,我们定义了两个消息类型:Date和Time。它们都包含一个名为value的字节数组字段,并使用prost宏的bytes_type属性将其与chrono库中的NaiveDate和NaiveTime类型相关联。
自定义编解码
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_type的oneof字段。oneof字段中包含三个可能的类型:Dog、Cat和Bird。每个类型都包含与其相关联的字段。
在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。
相关推荐
- 快递查询教程,批量查询物流,一键管理快递
-
作为商家,每天需要查询许许多多的快递单号,面对不同的快递公司,有没有简单一点的物流查询方法呢?小编的回答当然是有的,下面随小编一起来试试这个新技巧。需要哪些工具?安装一个快递批量查询高手快递单号怎么快...
- 一键自动查询所有快递的物流信息 支持圆通、韵达等多家快递
-
对于各位商家来说拥有一个好的快递软件,能够有效的提高自己的工作效率,在管理快递单号的时候都需要对单号进行表格整理,那怎么样能够快速的查询所有单号信息,并自动生成表格呢?1、其实方法很简单,我们不需要一...
- 快递查询单号查询,怎么查物流到哪了
-
输入单号怎么查快递到哪里去了呢?今天小编给大家分享一个新的技巧,它支持多家快递,一次能查询多个单号物流,还可对查询到的物流进行分析、筛选以及导出,下面一起来试试。需要哪些工具?安装一个快递批量查询高手...
- 3分钟查询物流,教你一键批量查询全部物流信息
-
很多朋友在问,如何在短时间内把单号的物流信息查询出来,查询完成后筛选已签收件、筛选未签收件,今天小编就分享一款物流查询神器,感兴趣的朋友接着往下看。第一步,运行【快递批量查询高手】在主界面中点击【添...
- 快递单号查询,一次性查询全部物流信息
-
现在各种快递的查询方式,各有各的好,各有各的劣,总的来说,还是有比较方便的。今天小编就给大家分享一个新的技巧,支持多家快递,一次能查询多个单号的物流,还能对查询到的物流进行分析、筛选以及导出,下面一起...
- 快递查询工具,批量查询多个快递快递单号的物流状态、签收时间
-
最近有朋友在问,怎么快速查询单号的物流信息呢?除了官网,还有没有更简单的方法呢?小编的回答当然是有的,下面一起来看看。需要哪些工具?安装一个快递批量查询高手多个京东的快递单号怎么快速查询?进入快递批量...
- 快递查询软件,自动识别查询快递单号查询方法
-
当你拥有多个快递单号的时候,该如何快速查询物流信息?比如单号没有快递公司时,又该如何自动识别再去查询呢?不知道如何操作的宝贝们,下面随小编一起来试试。需要哪些工具?安装一个快递批量查询高手快递单号若干...
- 教你怎样查询快递查询单号并保存物流信息
-
商家发货,快递揽收后,一般会直接手动复制到官网上一个个查询物流,那么久而久之,就会觉得查询变得特别繁琐,今天小编给大家分享一个新的技巧,下面一起来试试。教程之前,我们来预览一下用快递批量查询高手...
- 简单几步骤查询所有快递物流信息
-
在高峰期订单量大的时候,可能需要一双手当十双手去查询快递物流,但是由于逐一去查询,效率极低,追踪困难。那么今天小编给大家分享一个新的技巧,一次能查询多个快递单号的物流,下面一起来学习一下,希望能给大家...
- 物流单号查询,如何查询快递信息,按最后更新时间搜索需要的单号
-
最近有很多朋友在问,如何通过快递单号查询物流信息,并按最后更新时间搜索出需要的单号呢?下面随小编一起来试试吧。需要哪些工具?安装一个快递批量查询高手快递单号若干怎么快速查询?运行【快递批量查询高手】...
- 连续保存新单号功能解析,导入单号查询并自动识别批量查快递信息
-
快递查询已经成为我们日常生活中不可或缺的一部分。然而,面对海量的快递单号,如何高效、准确地查询每一个快递的物流信息,成为了许多人头疼的问题。幸运的是,随着科技的进步,一款名为“快递批量查询高手”的软件...
- 快递查询教程,快递单号查询,筛选更新量为1的单号
-
最近有很多朋友在问,怎么快速查询快递单号的物流,并筛选出更新量为1的单号呢?今天小编给大家分享一个新方法,一起来试试吧。需要哪些工具?安装一个快递批量查询高手多个快递单号怎么快速查询?运行【快递批量查...
- 掌握批量查询快递动态的技巧,一键查找无信息记录的两种方法解析
-
在快节奏的商业环境中,高效的物流查询是确保业务顺畅运行的关键。作为快递查询达人,我深知时间的宝贵,因此,今天我将向大家介绍一款强大的工具——快递批量查询高手软件。这款软件能够帮助你批量查询快递动态,一...
- 从复杂到简单的单号查询,一键清除单号中的符号并批量查快递信息
-
在繁忙的商务与日常生活中,快递查询已成为不可或缺的一环。然而,面对海量的单号,逐一查询不仅耗时费力,还容易出错。现在,有了快递批量查询高手软件,一切变得简单明了。只需一键,即可搞定单号查询,一键处理单...
- 物流单号查询,在哪里查询快递
-
如果在快递单号多的情况,你还在一个个复制粘贴到官网上手动查询,是一件非常麻烦的事情。于是乎今天小编给大家分享一个新的技巧,下面一起来试试。需要哪些工具?安装一个快递批量查询高手快递单号怎么快速查询?...
你 发表评论:
欢迎- 一周热门
- 最近发表
- 标签列表
-
- wireshark怎么抓包 (75)
- qt sleep (64)
- cs1.6指令代码大全 (55)
- factory-method (60)
- sqlite3_bind_blob (52)
- hibernate update (63)
- c++ base64 (70)
- nc 命令 (52)
- wm_close (51)
- epollin (51)
- sqlca.sqlcode (57)
- lua ipairs (60)
- tv_usec (64)
- 命令行进入文件夹 (53)
- postgresql array (57)
- statfs函数 (57)
- .project文件 (54)
- lua require (56)
- for_each (67)
- c#工厂模式 (57)
- wxsqlite3 (66)
- dmesg -c (58)
- fopen参数 (53)
- tar -zxvf -c (55)
- 速递查询 (52)