Rust从入门到精通之进阶篇:12.高级类型系统
高级类型系统
Rust 的类型系统是其最强大的特性之一,它不仅提供了内存安全保证,还允许开发者创建灵活、可复用的代码。在本章中,我们将深入探索 Rust 的高级类型特性,包括泛型、特质(trait)和高级类型模式。
泛型
泛型允许我们编写适用于多种类型的代码,而不是为每种类型编写重复的代码。
泛型函数
fn largest<T: PartialOrd>(list: &[T]) -> &T {
let mut largest = &list[0];
for item in list.iter() {
if item > largest {
largest = item;
}
}
largest
}
fn main() {
let numbers = vec![34, 50, 25, 100, 65];
let result = largest(&numbers);
println!("最大的数字是 {}", result);
let chars = vec!['y', 'm', 'a', 'q'];
let result = largest(&chars);
println!("最大的字符是 {}", result);
}
在这个例子中,<T: PartialOrd>
表示泛型类型 T
必须实现 PartialOrd
特质,这样我们才能使用 >
运算符比较两个值。
泛型结构体
struct Point<T> {
x: T,
y: T,
}
struct MixedPoint<T, U> {
x: T,
y: U,
}
fn main() {
let integer_point = Point { x: 5, y: 10 };
let float_point = Point { x: 1.0, y: 4.0 };
let mixed_point = MixedPoint { x: 5, y: 4.0 };
}
泛型枚举
标准库中的 Option
和 Result
就是泛型枚举的例子:
enum Option<T> {
Some(T),
None,
}
enum Result<T, E> {
Ok(T),
Err(E),
}
泛型方法
struct Point<T> {
x: T,
y: T,
}
impl<T> Point<T> {
fn x(&self) -> &T {
&self.x
}
}
// 只为特定类型实现方法
impl Point<f32> {
fn distance_from_origin(&self) -> f32 {
(self.x.powi(2) + self.y.powi(2)).sqrt()
}
}
泛型性能
Rust 的泛型实现使用了单态化(monomorphization):编译器会为每种使用的具体类型生成专门的代码。这意味着使用泛型不会有运行时开销,但可能会增加编译后的代码大小。
特质(Trait)
特质定义了类型可以实现的功能,类似于其他语言中的接口。
定义特质
trait Summary {
fn summarize(&self) -> String;
// 带有默认实现的方法
fn default_summary(&self) -> String {
String::from("(...)")
}
}
实现特质
struct NewsArticle {
headline: String,
location: String,
author: String,
content: String,
}
impl Summary for NewsArticle {
fn summarize(&self) -> String {
format!("{}, by {} ({})", self.headline, self.author, self.location)
}
}
struct Tweet {
username: String,
content: String,
reply: bool,
retweet: bool,
}
impl Summary for Tweet {
fn summarize(&self) -> String {
format!("{}: {}", self.username, self.content)
}
}
特质作为参数
// 使用特质约束
fn notify<T: Summary>(item: &T) {
println!("突发新闻!{}", item.summarize());
}
// 使用 impl Trait 语法(语法糖)
fn notify_alt(item: &impl Summary) {
println!("突发新闻!{}", item.summarize());
}
特质约束
// 多个特质约束
fn notify<T: Summary + Display>(item: &T) {
println!("突发新闻!{}", item.summarize());
println!("显示:{}", item);
}
// 使用 where 子句简化复杂约束
fn some_function<T, U>(t: &T, u: &U) -> i32
where T: Display + Clone,
U: Clone + Debug
{
// 函数体
42
}
返回实现特质的类型
fn returns_summarizable() -> impl Summary {
Tweet {
username: String::from("horse_ebooks"),
content: String::from("当然,你知道的,这是一个很好的例子"),
reply: false,
retweet: false,
}
}
注意:impl Trait
语法只允许返回单一类型,不能在不同条件下返回不同类型。
特质对象
当我们需要在运行时处理不同类型的值时,可以使用特质对象:
fn main() {
let articles = vec![
NewsArticle {
headline: String::from("Rust 1.50 发布"),
location: String::from("全球"),
author: String::from("Rust 团队"),
content: String::from("Rust 1.50 带来了许多新特性..."),
},
NewsArticle {
headline: String::from("Rust 在企业中的应用"),
location: String::from("中国"),
author: String::from("张三"),
content: String::from("越来越多的企业开始使用 Rust..."),
},
];
let tweets = vec![
Tweet {
username: String::from("rust_lang"),
content: String::from("Rust 1.50 已发布!"),
reply: false,
retweet: false,
},
Tweet {
username: String::from("alice"),
content: String::from("我爱 Rust!"),
reply: false,
retweet: false,
},
];
// 创建一个包含不同类型的向量,所有类型都实现了 Summary 特质
let mut summary_objects: Vec<Box<dyn Summary>> = Vec::new();
// 添加 NewsArticle 实例
for article in articles {
summary_objects.push(Box::new(article));
}
// 添加 Tweet 实例
for tweet in tweets {
summary_objects.push(Box::new(tweet));
}
// 遍历并调用 summarize 方法
for obj in summary_objects {
println!("{}", obj.summarize());
}
}
特质对象使用 dyn
关键字表示,如 Box<dyn Summary>
。特质对象使用动态分发,这意味着编译器会在运行时查找要调用的方法,这会带来一些性能开销。
对象安全性
只有对象安全的特质才能用作特质对象。特质对象安全需要满足以下条件:
- 返回值不是
Self
- 没有泛型类型参数
关联类型
关联类型是特质中的类型占位符,实现特质时指定具体类型:
trait Iterator {
type Item;
fn next(&mut self) -> Option<Self::Item>;
}
struct Counter {
count: u32,
max: u32,
}
impl Iterator for Counter {
type Item = u32;
fn next(&mut self) -> Option<Self::Item> {
if self.count < self.max {
self.count += 1;
Some(self.count)
} else {
None
}
}
}
关联类型与泛型的区别在于,使用关联类型时,一个类型只能实现特质一次,而使用泛型可以多次实现同一特质。
高级特质特性
默认泛型类型参数
trait Add<RHS=Self> {
type Output;
fn add(self, rhs: RHS) -> Self::Output;
}
struct Point {
x: i32,
y: i32,
}
impl Add for Point {
type Output = Point;
fn add(self, other: Point) -> Point {
Point {
x: self.x + other.x,
y: self.y + other.y,
}
}
}
在这个例子中,RHS=Self
是一个默认类型参数,表示如果不指定 RHS
,则默认使用 Self
类型。
完全限定语法
当存在同名方法时,可以使用完全限定语法明确指定要调用的方法:
trait Pilot {
fn fly(&self);
}
trait Wizard {
fn fly(&self);
}
struct Human;
impl Pilot for Human {
fn fly(&self) {
println!("这里是机长说话...");
}
}
impl Wizard for Human {
fn fly(&self) {
println!("起飞!");
}
}
impl Human {
fn fly(&self) {
println!("*挥动双臂*");
}
}
fn main() {
let person = Human;
person.fly(); // 调用 Human 的方法
Pilot::fly(&person); // 调用 Pilot 特质的方法
Wizard::fly(&person); // 调用 Wizard 特质的方法
}
对于关联函数(没有 &self
参数的函数),语法略有不同:
trait Animal {
fn baby_name() -> String;
}
struct Dog;
impl Dog {
fn baby_name() -> String {
String::from("小狗")
}
}
impl Animal for Dog {
fn baby_name() -> String {
String::from("幼犬")
}
}
fn main() {
println!("狗宝宝的名字是:{}", Dog::baby_name());
println!("动物宝宝的名字是:{}", <Dog as Animal>::baby_name());
}
超特质(Supertrait)
一个特质可以要求实现者也必须实现另一个特质:
trait Display {
fn display(&self) -> String;
}
trait DisplayWithDebug: Display + std::fmt::Debug {
fn display_with_debug(&self) {
println!("DisplayWithDebug: {}", self.display());
println!("Debug: {:?}", self);
}
}
在这个例子中,要实现 DisplayWithDebug
特质,类型必须同时实现 Display
和 std::fmt::Debug
特质。
新类型模式
新类型模式允许我们为外部类型实现外部特质,绕过孤儿规则(不能为外部类型实现外部特质):
struct Wrapper(Vec<String>);
impl std::fmt::Display for Wrapper {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
write!(f, "[{}]", self.0.join(", "))
}
}
fn main() {
let w = Wrapper(vec![String::from("hello"), String::from("world")]);
println!("{}", w);
}
类型别名
类型别名可以为现有类型创建新名称,简化复杂类型签名:
type Kilometers = i32;
type Thunk = Box<dyn Fn() + Send + 'static>;
type Result<T> = std::result::Result<T, std::io::Error>;
fn take_long_type(f: Thunk) {
// --snip--
}
fn returns_long_type() -> Thunk {
// --snip--
Box::new(|| println!("hi"))
}
Never 类型
!
类型,也称为 never 类型,表示永远不会返回的计算:
fn bar() -> ! {
panic!("永远不会返回!");
}
fn main() {
let guess: u32 = match "42".parse() {
Ok(num) => num,
Err(_) => continue, // continue 的类型是 !
};
}
动态大小类型
动态大小类型(DST)是编译时无法知道大小的类型,如 str
和特质对象。我们通常通过指针使用它们,如 &str
或 Box<dyn Trait>
。
fn main() {
let s1: &str = "Hello";
let s2: &str = "World!";
// 不能直接使用 str 类型的变量
// let s3: str = *s1; // 错误
// 但可以通过引用或 Box 使用
let s4: Box<str> = "Boxed string".into();
println!("{} {}", s1, s2);
println!("{}", s4);
}
高级类型系统应用
类型状态模式
类型状态模式使用类型系统确保对象只能在有效状态下执行特定操作:
// 定义状态类型
struct Draft {};
struct PendingReview {};
struct Published {};
// 博客文章结构体,使用泛型参数表示状态
struct Post<State> {
content: String,
state: State,
}
// Draft 状态的实现
impl Post<Draft> {
fn new() -> Self {
Post {
content: String::new(),
state: Draft {},
}
}
fn add_text(&mut self, text: &str) {
self.content.push_str(text);
}
fn request_review(self) -> Post<PendingReview> {
Post {
content: self.content,
state: PendingReview {},
}
}
}
// PendingReview 状态的实现
impl Post<PendingReview> {
fn approve(self) -> Post<Published> {
Post {
content: self.content,
state: Published {},
}
}
fn reject(self) -> Post<Draft> {
Post {
content: self.content,
state: Draft {},
}
}
}
// Published 状态的实现
impl Post<Published> {
fn content(&self) -> &str {
&self.content
}
}
fn main() {
let mut post = Post::new();
post.add_text("今天我学习了 Rust 的高级类型系统");
let post = post.request_review();
// 不能在 PendingReview 状态下添加文本
// post.add_text("更多内容"); // 错误
let post = post.approve();
// 只能在 Published 状态下访问内容
println!("{}", post.content());
}
幽灵类型
幽灵类型是一种在类型级别使用但在值级别不存在的类型参数:
use std::marker::PhantomData;
use std::ops::Add;
// 定义单位类型
struct Meters;
struct Feet;
// 带有幽灵类型参数的距离类型
struct Distance<Unit> {
value: f64,
_unit: PhantomData<Unit>,
}
impl<Unit> Distance<Unit> {
fn new(value: f64) -> Self {
Distance {
value,
_unit: PhantomData,
}
}
}
// 只允许相同单位的距离相加
impl<Unit> Add for Distance<Unit> {
type Output = Distance<Unit>;
fn add(self, other: Distance<Unit>) -> Distance<Unit> {
Distance::new(self.value + other.value)
}
}
// 转换函数
fn meters_to_feet(meters: Distance<Meters>) -> Distance<Feet> {
Distance::new(meters.value * 3.28084)
}
fn main() {
let d1 = Distance::<Meters>::new(5.0);
let d2 = Distance::<Meters>::new(10.0);
let d3 = d1 + d2; // 正确:相同单位
let d4 = Distance::<Feet>::new(15.0);
// let d5 = d3 + d4; // 错误:不同单位
let d6 = meters_to_feet(d3);
println!("{}米 = {}英尺", d3.value, d6.value);
}
最佳实践
使用泛型还是特质对象
- 泛型:当性能至关重要,且在编译时知道所有可能的类型时使用
- 特质对象:当需要在运行时处理多种不同类型,且性能不是主要考虑因素时使用
特质设计原则
- 单一职责:每个特质应该表示一个清晰、独立的功能
- 组合优于继承:使用多个小特质组合,而不是一个大特质
- 提供默认实现:为常见功能提供默认实现,减少重复代码
- 考虑对象安全性:如果特质将用作特质对象,确保它是对象安全的
类型转换最佳实践
- 使用
From
和Into
特质:为类型转换实现这些特质 - 使用
TryFrom
和TryInto
:当转换可能失败时使用 - 使用
AsRef
和AsMut
:当需要临时借用不同类型的引用时使用
struct Person {
name: String,
age: u32,
}
struct PersonDto {
full_name: String,
years: u32,
}
impl From<Person> for PersonDto {
fn from(person: Person) -> Self {
PersonDto {
full_name: person.name,
years: person.age,
}
}
}
fn main() {
let person = Person {
name: String::from("张三"),
age: 30,
};
// 使用 From 特质
let dto = PersonDto::from(person);
// 或使用 Into 特质(自动实现)
// let person = Person { name: String::from("李四"), age: 25 };
// let dto: PersonDto = person.into();
println!("{}, {}", dto.full_name, dto.years);
}
练习题
-
实现一个泛型
Stack<T>
结构体,包含push
、pop
和peek
方法。确保它可以处理任何类型的数据。 -
定义一个
Drawable
特质,包含draw
方法。然后实现多种形状(圆形、矩形、三角形),并创建一个函数,接受一个Vec<Box<dyn Drawable>>
并绘制所有形状。 -
创建一个类型状态模式的例子,模拟一个文件的生命周期:
Closed
->Open
->Reading
/Writing
->Closed
。确保只有在适当的状态下才能执行特定操作。 -
实现一个
Convert
特质,包含to_json
和from_json
方法。为至少两种自定义类型实现这个特质,并演示如何使用它进行 JSON 序列化和反序列化。 -
创建一个使用幽灵类型的例子,确保不同货币(如美元、欧元、人民币)的金额不能直接相加,但可以通过适当的转换函数进行转换。
总结
在本章中,我们深入探讨了 Rust 的高级类型系统:
- 泛型允许我们编写适用于多种类型的代码
- 特质定义了类型可以实现的功能
- 特质对象使我们可以在运行时处理不同类型
- 关联类型提供了一种在特质中定义类型占位符的方法
- 高级特质特性,如默认类型参数和超特质
- 类型别名简化了复杂类型签名
- Never 类型表示永远不会返回的计算
- 动态大小类型处理编译时大小未知的类型
- 高级类型系统应用,如类型状态模式和幽灵类型
Rust 的类型系统是其最强大的特性之一,掌握这些高级特性将使你能够编写更安全、更灵活、更可维护的代码。在下一章中,我们将探索 Rust 的错误处理进阶技术。