0x00 开篇
我们用 24 篇文章结束了 Rust 的基础教程,接下来进入 Rust 进阶的学习。相信你通过阅读前面的文章可以写一些小的应用了。从这篇文章开始,我们将开始深入了解 Rust,我会陆续介绍 Rust 的类型系统,所有权系统以及指针。如果你想练习 Rust ,我推荐大家可以到力扣(LeetCode)找一些简单的题目来练习。中级课程的第一篇文章,我将先介绍 Rust 的泛型。Rust 是一门强类型并且类型安全的静态语言。 在 Rust 中,所有的一切都是类型,包括我们常见的原生类型以及复合数据类型。
0x01 泛型的定义
其实泛型我们已经见过它了,前面介绍的 Vec<T> 就是泛型的应用,T 可以代指任何类型。在 Rust 中,<T>代表泛型类型了。泛型是指在运行时指定数据类型的机制,使用泛型可以编写通用的代码,减少重复的工作量。
0x02 泛型与向量
通过【RUST 学习日记 第 11 课 向量】的文章,我们了解了向量。在向量声明时我们需要指定类型,代码示例如下:
fn main() { // 1、 泛型与向量 let mut vec :Vec<&str> = vec![]; vec.push("Hello"); vec.push("Rust"); println!("{:?}", vec); // 下面的语句会产生错误 // vec.push(1); }
如果我们声明向量时标明的泛型类型为 &str ,那我们插入其它类型的元素就会报错。
因此,在向量中使用泛型类型,不仅提高了可读性,而且还会及时提醒咱们在写代码时的误操作。
0x03 泛型函数
在函数中,参数和返回值都可以是泛型类型,我们将带有泛型类型参数的或者返回值的函数叫做泛型函数。函数签名的形式如下:
声明泛型函数
fn func_name(a : T, {other parameters}) -> T { // code... }
使用泛型函数
同普通函数不同的是,我们需要指定类型
func_name::(a, {other parameters})
Demo
下面通过一个 Demo 来讲解泛型函数。
Question 请写一个函数,输入两个整数数,返回第一个参数。
当然,我们可以直接上手写。示例代码如下:
fn main() { let a = print_generic::<i32>(3, 5); println!("a = {}", a); // 输出结果 // a = 3 } fn print<T>(a: T, b: T) -> T { return a; }
如果我们要求类型可以是 u8,u16,u32,i8,i16,i32,f32 等等呢?我们可以通过泛型类型辅助我们减少工作量。示例代码如下:
fn print_generic<T>(a: T, b: T) -> T { return a; } fn main() { let b = print_generic::<f32>(6.7, 4.5); println!("b = {}", b); // 输出结果 // b = 6.7 let c = print_generic::<&str>("hello", "rust"); println!("c = {}", c); // 输出结果 // c = hello }
通过上面的代码,不仅完成了题目,而且还扩展成了字符串......哈哈~~
0x04 泛型枚举
枚举类型也可以泛型化。在 Rust 中常见的两个泛型枚举就是 Option<T> 和 Result<T, E> 了。这里主要先看下 Option<T> 官方的定义。
/// The `Option` type. See [the module level documentation](self) for more. #[derive(Copy, PartialEq, PartialOrd, Eq, Ord, Debug, Hash)] #[rustc_diagnostic_item = "Option"] #[stable(feature = "rust1", since = "1.0.0")] pub enum Option<T> { /// No value. None, /// Some value of type `T`. Some(T), }
Option<T> 表示一个可选的值。每个 Option 要么是 Some并包含一个值,要么是 None。通常我们使用它来赋初始值或者遇到错误的时候返回 None。示例代码如下:
fn main() { let op = Some(5); println!("{:?}", op); // 除法 let x1 = divider(5.0, 2.0); let x2 = divider(5.0, 0.0); println!("{:?}", x1); println!("{:?}", x2); // 输出结果 // Some(5) // Some(2.5) // None } /// 计算除法 fn divider(a: f64, b: f64) -> Option<f64> { if b == 0.0 { None } else { Some(a / b) } }
在写代码时,我们可以直接使用 Some(T) 和 None 来表示 Option<T>。另外,泛型枚举这里有个注意点:如果类型 T 是 Box 或其它智能指针类型, Rust 在内存中会省掉 Option<T>的标签字段。例如:Option<Box<f64>>在内存中只占一个字节存储(我还没介绍指针,暂做了解即可)。
0x05 小结
泛型是 Rust 类型系统中重要的概念,它可以减少我们的代码重复率,编写出干净简洁更为抽象和通用的代码。泛型不仅可以用在向量、函数、枚举中,还可以用于结构体,集合,trait,方法中,后面我都会一一介绍。